def __init__(self): self.dataPort = 9988 self.dataStoreClientList = [] self.uniqueId = GUID() # Start transfer service self.dataTransferServer = GSIHTTPTransferServer( ('', int(self.dataPort)), numThreads=4, sslCompat=0) self.dataTransferServer.run() self.CreateDataStore() # Basic test d = DataStoreClient(self.dataStore, 0) d.StartDataOperations() # Thread test #self.StartDataStoreClients() self.ShutDown()
class StartTest: ''' Creates a transfer engine and a datastore. Also, creates DataStoreClient that uses datastore method to test datastore. ''' def __init__(self): self.dataPort = 9988 self.dataStoreClientList = [] self.uniqueId = GUID() # Start transfer service self.dataTransferServer = GSIHTTPTransferServer( ('', int(self.dataPort)), numThreads=4, sslCompat=0) self.dataTransferServer.run() self.CreateDataStore() # Basic test d = DataStoreClient(self.dataStore, 0) d.StartDataOperations() # Thread test #self.StartDataStoreClients() self.ShutDown() def CreateDataStore(self): self.dataStorePath = 'DataStoreTestDir' #self.dataStorePath = os.path.join(self.dataStoreLocation, str(self.uniqueId)) if not os.path.exists(self.dataStorePath): try: os.mkdir(self.dataStorePath) except OSError: self.dataStorePath = None self.dataStore = DataStore.DataStore(self, self.dataStorePath, str(self.uniqueId), self.dataTransferServer) #def StartDataStoreClients(self): # '''Creates several data store clients, and starts methods in threads to # stress test datastore''' # # Create data store clients # index = 1 # for client in range(nrOfClients): # d = DataStoreClient(self.dataStore, index) # self.dataStoreClientList.append(d) # index = index + 1 # # self.threadList = [] # # Start data operations in threads # for d in self.dataStoreClientList: # thread = threading.Thread(target = d.StartDataStoreOperations) # self.threadList.append(thread) # # for thread in self.threadList: # thread.start() # # # Join threads # for thread in self.threadList: # thread.join() def ShutDown(self): '''Stop server and datastore''' self.dataTransferServer.stop() self.dataStore.Shutdown() def AddData(self, data): '''Callback for datastore''' pass def UpdateData(self, data): '''Callback for datastore''' pass def GetData(self): '''Callback for datastore''' return none