def test_MPEngine_dropProducers(self): try: print "Starting test" maxCores = 6 mpe = MPEngineProdCons(maxCores, WkrTestProd, WkrTestCons) # Add tasks task_list = [i for i in xrange(1, 10)] mpe.addTaskList(task_list) mpe.startProducers(4) mpe.addProducer() mpe.addProducer() mpe.removeProducer() mpe.removeProducer() mpe.startConsumers(1) while mpe.working(): print("Prod: %d / Cons: %d | %s -> %s -> %s" % mpe.getProgress()) time.sleep(1) mpe.rebalance() print "Test ended" except Exception: traceback.print_exc(file=sys.stdout) self.fail("Exception triggered") # Pass self.assertEquals(1, 1, "test_MPEngine")
def test_MPEngine_brute_both(self): try: print "Starting test" mpe = MPEngineProdCons(99, WkrTestProd, WkrTestCons) max_procs = 5 # Add tasks task_list = [i for i in xrange(1, 5)] mpe.addTaskList(task_list) for i in xrange(0, max_procs): mpe.addProducer() for i in xrange(0, max_procs): mpe.removeProducer() for i in xrange(0, max_procs): mpe.addConsumer() for i in xrange(0, max_procs): mpe.removeConsumer() for i in xrange(0, max_procs): mpe.addProducer() for i in xrange(0, max_procs): mpe.addConsumer() for i in xrange(0, max_procs): mpe.removeProducer() for i in xrange(0, max_procs): mpe.removeConsumer() del mpe print "Test ended" except Exception: traceback.print_exc(file=sys.stdout) self.fail("Exception triggered") # Pass self.assertEquals(1, 1, "test_MPEngine")
def test_MPEngine_ProducerSimple(self): # Get temp db name for the test tempdb = tempfile.NamedTemporaryFile(suffix='.db', prefix='testCase', dir=tempfile.gettempdir()) tempdb.close() dbfilenameFullPath = tempdb.name try: with appDB.DBClass(dbfilenameFullPath, settings.__version__) as DB: DB.appInitDB() print "Starting test" mpe = MPEngineProdCons(6, WkrTestProd, WkrTestCons) # Add tasks task_list = [i for i in xrange(1, 5)] mpe.addTaskList(task_list) mpe.addProducer() mpe.removeProducer() time.sleep(2) del mpe print "Test ended" except Exception: traceback.print_exc(file=sys.stdout) self.fail("Exception triggered") # Pass self.assertEquals(1, 1, "test_MPEngine")
def test_MPEngine4(self): try: print "Starting test" mpe = MPEngineProdCons(6, WkrTestProd, WkrTestCons) mpe.removeProducer() mpe.removeProducer() print "Test ended" except Exception: traceback.print_exc(file=sys.stdout) self.fail("Exception triggered") # Pass self.assertEquals(1, 1, "test_MPEngine")
def test_MPEngine_brute_both_badmix_single(self): try: print "Starting test" mpe = MPEngineProdCons(99, WkrTestProd, WkrTestCons) max_procs = 10 # Add tasks task_list = [i for i in xrange(1, 5)] mpe.addTaskList(task_list) mpe.addProducer() mpe.removeProducer() mpe.addConsumer() mpe.removeConsumer() mpe.addProducer() mpe.addConsumer() mpe.removeProducer() mpe.removeConsumer() mpe.addProducer() mpe.addConsumer() mpe.removeConsumer() mpe.removeProducer() if (mpe.check_mpEngineStatus()): print("++Status looks ok!") else: print("++Status looks wrong!") del mpe print "Test ended" except Exception: traceback.print_exc(file=sys.stdout) self.fail("Exception triggered") # Pass self.assertEquals(1, 1, "test_MPEngine")