Test for our external events [redis_e2e_test]

hjwp · hjwp · commit 2f707da6fa82 · 2019-07-26T15:31:11.000+01:00
diff --git a/tests/e2e/test_external_events.py b/tests/e2e/test_external_events.py
@@ -0,0 +1,38 @@
+import json
+import pytest
+from .wait_for import wait_for
+from . import api_client, redis_client
+from ..random_refs import random_batchref, random_orderid, random_sku
+
+
+
+@pytest.mark.usefixtures('postgres_db')
+@pytest.mark.usefixtures('restart_api')
+@pytest.mark.usefixtures('restart_redis_pubsub')
+def test_change_batch_quantity_leading_to_reallocation():
+    # start with two batches and an order allocated to one of them
+    orderid, sku = random_orderid(), random_sku()
+    earlier_batch, later_batch = random_batchref('old'), random_batchref('newer')
+    api_client.post_to_add_batch(earlier_batch, sku, qty=10, eta='2011-01-02')
+    api_client.post_to_add_batch(later_batch, sku, qty=10, eta='2011-01-02')
+    response = api_client.post_to_allocate(orderid, sku, 10)
+    assert response.json()['batchref'] == earlier_batch
+
+    subscription = redis_client.subscribe_to('line_allocated')
+
+    # change quantity on allocated batch so it's less than our order
+    redis_client.publish_message('change_batch_quantity', {
+        'batchref': earlier_batch, 'qty': 5
+    })
+
+    # wait until we see a message saying the order has been reallocated
+    messages = []
+    def assert_new_allocation_published():
+        messages.append(wait_for(subscription.get_message))
+        print(messages)
+        data = json.loads(messages[-1]['data'])
+        assert data['orderid'] == orderid
+        assert data['batchref'] == later_batch
+        return True
+
+    wait_for(assert_new_allocation_published)
