def test_multiprocessing_multiapp_system(self):
system = System(
Orders | Reservations | Orders,
Orders | Payments | Orders,
setup_tables=True,
infrastructure_class=self.infrastructure_class,
)
self.set_db_uri()
with system.construct_app(Orders) as app:
# Create a new order.
order_id = create_new_order()
# Check new order exists in the repository.
assert order_id in app.repository
self.close_connections_before_forking()
with MultiprocessRunner(system):
with system.construct_app(Orders) as app:
retries = 50
while not app.repository[order_id].is_reserved:
sleep(0.1)
retries -= 1
assert retries, "Failed set order.is_reserved"
while retries and not app.repository[order_id].is_paid:
sleep(0.1)
retries -= 1
assert retries, "Failed set order.is_paid"
def test_not_infrastructure_class_exception(self):
system = System(
Orders | Reservations | Orders,
Orders | Payments | Orders,
setup_tables=True,
infrastructure_class=TestCase,
)
with self.assertRaises(ProgrammingError):
system.construct_app(Orders)
def test_construct_app(self):
system = System(
Orders | Reservations | Orders,
Orders | Payments | Orders,
setup_tables=True,
infrastructure_class=self.infrastructure_class,
)
with system.construct_app(Orders) as app:
self.assertIsInstance(app, Orders)
with system.construct_app(Payments) as app:
self.assertIsInstance(app, Payments)
with system.construct_app(Reservations) as app:
self.assertIsInstance(app, Reservations)
class TestActors(unittest.TestCase):
infrastructure_class = SQLAlchemyApplication
def setUp(self):
# Set environment.
set_db_uri()
# Define system.
self.system = System(
Orders | Reservations | Orders | Payments | Orders,
infrastructure_class=self.infrastructure_class,
)
def test_simple_system_base(self):
start_actor_system()
self.check_actors()
@skip("Having trouble running Thespian's 'multiproc tcp base'")
def test_multiproc_tcp_base(self):
start_multiproc_tcp_base_system()
self.check_actors()
def close_connections_before_forking(self):
# Used for closing Django connection before multiprocessing module forks the OS process.
pass
def check_actors(self, num_pipelines=3, num_orders_per_pipeline=5):
pipeline_ids = list(range(num_pipelines))
self.close_connections_before_forking()
actors = ActorModelRunner(self.system,
pipeline_ids=pipeline_ids,
shutdown_on_close=True)
# Todo: Use wakeupAfter() to poll for new notifications (see Timer Messages).
order_ids = []
with self.system.construct_app(Orders,
setup_table=True) as app, actors:
# Create some new orders.
for _ in range(num_orders_per_pipeline):
for pipeline_id in pipeline_ids:
app.change_pipeline(pipeline_id)
order_id = create_new_order()
order_ids.append(order_id)
# Wait for orders to be reserved and paid.
retries = 20 + 10 * num_orders_per_pipeline * len(pipeline_ids)
for i, order_id in enumerate(order_ids):
while not app.repository[order_id].is_reserved:
time.sleep(0.1)
retries -= 1
assert retries, "Failed set order.is_reserved {} ({})".format(
order_id, i)
while retries and not app.repository[order_id].is_paid:
time.sleep(0.1)
retries -= 1
assert retries, "Failed set order.is_paid ({})".format(i)
# Calculate timings from event timestamps.
orders = [app.repository[oid] for oid in order_ids]
first_timestamp = min([o.__created_on__ for o in orders])
last_timestamp = max([o.__last_modified__ for o in orders])
duration = last_timestamp - first_timestamp
rate = len(order_ids) / float(duration)
period = 1 / rate
print(
"Orders system processed {} orders in {:.3f}s at rate of {:.1f} "
"orders/s, {:.3f}s each".format(len(order_ids), duration, rate,
period))
# Print min, average, max duration.
durations = [
o.__last_modified__ - o.__created_on__ for o in orders
]
print("Min order processing time: {:.3f}s".format(min(durations)))
print("Mean order processing time: {:.3f}s".format(
sum(durations) / len(durations)))
print("Max order processing time: {:.3f}s".format(max(durations)))
def tearDown(self):
# Unset environment.
try:
del os.environ["DB_URI"]
except KeyError:
pass
try:
# Shutdown base actor system.
shutdown_actor_system()
finally:
# Clear event handlers.
try:
assert_event_handlers_empty()
finally:
clear_event_handlers()
