def test_order_submitted_event(self): """Test to verify the mechanics of the order submitted event""" # Constants initial_time = hour_to_sec(15) placement_time = time(15, 0, 0) preparation_time = time(15, 1, 0) ready_time = time(15, 15, 0) # Services env = Environment(initial_time=initial_time) dispatcher = Dispatcher(env=env, matching_policy=DummyMatchingPolicy()) # Creates an order and submits it to the dispatcher order = Order(order_id=32, placement_time=placement_time) dispatcher.order_submitted_event(order, preparation_time, ready_time) env.run(until=initial_time + 121) # Verify order properties are set and it is correctly allocated self.assertEqual(order.preparation_time, preparation_time) self.assertEqual(order.ready_time, ready_time) self.assertIn(order.order_id, dispatcher.unassigned_orders.keys())
def test_buffer_event(self): """Test to verify how the mechanics of the dispatcher buffering orders work""" # Constants initial_time = hour_to_sec(16) placement_time = time(16, 0, 0) time_delta = min_to_sec(10) # Verifies two test cases for how the dispatcher buffers orders # For each test, assert the correct number of orders are buffered # Test 1: schedules the submission of three orders and assert that only two are buffered env = Environment(initial_time=initial_time) dispatcher = Dispatcher( env=env, buffering_policy=RollingBufferingPolicy(), matching_policy=DummyMatchingPolicy() ) order_1 = Order(order_id=1, placement_time=placement_time) order_2 = Order(order_id=2, placement_time=placement_time) order_3 = Order(order_id=3, placement_time=placement_time) dispatcher.order_submitted_event(order_1, preparation_time=time(16, 0, 27), ready_time=time(16, 10, 0)) dispatcher.order_submitted_event(order_2, preparation_time=time(16, 0, 43), ready_time=time(16, 10, 0)) dispatcher.order_submitted_event(order_3, preparation_time=time(18, 0, 0), ready_time=time(18, 10, 0)) env.run(until=initial_time + time_delta) self.assertEqual(len(dispatcher.unassigned_orders), 2) # Test 2: schedules the submission of three orders and assert that all three orders are buffered env = Environment(initial_time=initial_time) dispatcher = Dispatcher( env=env, buffering_policy=RollingBufferingPolicy(), matching_policy=DummyMatchingPolicy() ) order_1 = Order(order_id=1, placement_time=placement_time) order_2 = Order(order_id=2, placement_time=placement_time) order_3 = Order(order_id=3, placement_time=placement_time) dispatcher.order_submitted_event(order_1, preparation_time=time(16, 0, 27), ready_time=time(16, 10, 0)) dispatcher.order_submitted_event(order_2, preparation_time=time(16, 0, 43), ready_time=time(16, 10, 0)) dispatcher.order_submitted_event(order_3, preparation_time=time(16, 4, 1), ready_time=time(16, 14, 0)) env.run(until=initial_time + time_delta) self.assertEqual(len(dispatcher.unassigned_orders), 3)