test_cmd.py

#!/usr/local/bin/python
#
#                               Copyright 2002
#                                     by
#                        The Board of Trustees of the
#                     Leland Stanford Junior University.
#                            All rights reserved.
#

"""Test script for commanding
Contents:
Sample script for testing write command throughput
"""
__facility__ = "Online"
__abstract__ = "Test script for testing write command throughput"
__author__   = "Selim Tuvi <stuvi@slac.stanford.edu> SLAC - GLAST I&T/Online"
__date__     = ("$Date: 2005/03/21 18:37:28 $").split(' ')[1]
__version__  = "$Revision: 2.3 $"
__release__  = "$Name: R04-12-00 $"
__credits__  = "SLAC"

import LATTE.copyright_SLAC

from LATTE.client.gOptions import Options
from LATTE.client.gCmdCli import CmdCli
from LATTE.database.gLAT import GLAT

import os
import time


def test(cmd=None, disconnectOnExit=True):
  global tem, glt

  options = Options(['server', 'schema'])
  try:
    options.parse()
  except Exception, msg:
    options.usage(str(msg))
    return

  if cmd == None:
    cmd = CmdCli()
    cmd.connect(options.server)
  passed = True

  try:
    lat = GLAT(cmd)

    lat.readSchema(options.schema)

    # Get a TEM instance
    if lat.TEMcnt() == 1:
      tem = lat.TEM[lat.TEM.keys()[0]]
    elif lat.TEMcnt() > 1:
      tem = lat.TEM.all()

    cnt = 10000
    start = time.time()
    for i in xrange(cnt):
      tem.COMMAND_RESPONSE = 1234            # clear TEM stats reg
      #cmd.gTEMload(0, 3, 1234)  # ~18500 writes/second
      #if i % 1000 == 0: print i
    end = time.time()
    et = end-start
    if et == 0: et = 0.000001
    print "Finished in %f seconds (%.1f write commands/second)" %(et, (cnt/et))

  except Exception, e:
    print "Exception occurred: ", e
    passed = False

  if passed:
    print "test_cmd:  PASSED"
  else:
    print "test_cmd:  FAILED"

  if disconnectOnExit:
    cmd.disconnect()

if __name__ == '__main__':
  test()

# RiC 6/16/04: Measured 10298 +/- ~150 writes/sec on D600 laptop (1.7 GHz,
#              600 MHz bus, 1 GB) in local mode with prints commented out,
#              without cmdRelayer.
# RiC 6/17/04: Measured 5841 +- ~50 writes/sec on same setup as above but with
#              SocketServer based cmdRelayer.

---