def testC_MultiPool(self): """ _testMultiPool_ Run a test with multiple workers """ raise nose.SkipTest config = self.testInit.getConfiguration() config.Agent.useHeartbeat = False self.testInit.generateWorkDir(config) processPool = ProcessPool("ProcessPool_t.ProcessPoolTestWorker", totalSlaves = 3, componentDir = config.General.workDir, namespace = "WMCore_t", config = config) for i in range(100): input = [] while i > 0: input.append("COMMAND%s" % i) i -= 1 processPool.enqueue(input) result = processPool.dequeue(len(input)) self.assertEqual(len(result), len(input), "Error: Wrong number of results returned.")
def testA_ProcessPool(self): """ _testProcessPool_ """ raise nose.SkipTest config = self.testInit.getConfiguration() config.Agent.useHeartbeat = False self.testInit.generateWorkDir(config) processPool = ProcessPool("ProcessPool_t.ProcessPoolTestWorker", totalSlaves = 1, componentDir = config.General.workDir, config = config, namespace = "WMCore_t") processPool.enqueue(["One", "Two", "Three"]) result = processPool.dequeue(3) self.assertEqual(len(result), 3, "Error: Expected three items back.") self.assertTrue( "One" in result) self.assertTrue( "Two" in result) self.assertTrue( "Three" in result) return
def testB_ProcessPoolStress(self): """ _testProcessPoolStress_ """ config = self.testInit.getConfiguration() config.Agent.useHeartbeat = False self.testInit.generateWorkDir(config) processPool = ProcessPool( "ProcessPool_t.ProcessPoolTestWorker", totalSlaves=1, componentDir=config.General.workDir, namespace="WMCore_t", config=config, slaveInit={}, ) for i in range(1000): input = [] while i > 0: input.append("COMMAND%s" % i) i -= 1 processPool.enqueue(input) result = processPool.dequeue(len(input)) assert len(result) == len(input), "Error: Wrong number of results returned." return
def testA_ProcessPool(self): """ _testProcessPool_ """ config = self.testInit.getConfiguration() config.Agent.useHeartbeat = False self.testInit.generateWorkDir(config) processPool = ProcessPool( "ProcessPool_t.ProcessPoolTestWorker", totalSlaves=1, componentDir=config.General.workDir, config=config, namespace="WMCore_t", slaveInit={}, ) processPool.enqueue(["One", "Two", "Three"]) result = processPool.dequeue(3) assert len(result) == 3, "Error: Expected three items back." assert "One" in result, "Error: Missing data from dequeue()" assert "Two" in result, "Error: Missing data from dequeue()" assert "Three" in result, "Error: Missing data from dequeue()" return