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")
