Added more tests

Author: syrusakbary

.travis.yml

@@ -2,9 +2,9 @@ language: python
 sudo: false
 python:
 - 2.7
-- 3.4
-- 3.5
-- 3.6
+# - 3.4
+# - 3.5
+# - 3.6
 - "pypy-5.3.1"
 before_install:
 - |
@@ -33,6 +33,11 @@ matrix:
     - pip install pytest-asyncio
     script:
     - py.test --cov=graphql graphql tests tests_py35
+  - python: '3.6'
+    after_install:
+    - pip install pytest-asyncio
+    script:
+    - py.test --cov=graphql graphql tests tests_py35
   - python: '2.7'
     install: pip install flake8
     script:

graphql/execution/tests/test_subscribe.py

@@ -264,3 +264,119 @@ def test_produces_a_payload_for_multiple_subscribe_in_same_subscription():
 
     assert payload1[0].data == expected_payload
     assert payload2[0].data == expected_payload
+
+
+# Subscription Publish Phase
+def test_produces_a_payload_per_subscription_event():
+    stream = Subject()
+    send_important_email, subscription = create_subscription(stream)
+
+    payload = []
+
+    subscription.subscribe(payload.append)
+    send_important_email(Email(
+        from_='[email protected]',
+        subject='Alright',
+        message='Tests are good',
+        unread=True,
+    ))
+    expected_payload = {
+        'importantEmail': {
+            'email': {
+                'from': '[email protected]',
+                'subject': 'Alright',
+            },
+            'inbox': {
+                'unread': 1,
+                'total': 2,
+            },
+        }
+    }
+
+    assert len(payload) == 1
+    assert payload[0].data == expected_payload
+
+    send_important_email(Email(
+        from_='[email protected]',
+        subject='Tools',
+        message='I <3 making things',
+        unread=True,
+    ))
+    expected_payload = {
+        'importantEmail': {
+            'email': {
+                'from': '[email protected]',
+                'subject': 'Tools',
+            },
+            'inbox': {
+                'unread': 2,
+                'total': 3,
+            },
+        }
+    }
+
+    assert len(payload) == 2
+    assert payload[-1].data == expected_payload
+
+    # The client decides to disconnect
+    stream.on_completed()
+
+    send_important_email(Email(
+        from_='[email protected]',
+        subject='Important',
+        message='Read me please',
+        unread=True,
+    ))
+
+    assert len(payload) == 2
+
+
+def test_event_order_is_correct_for_multiple_publishes():
+    stream = Subject()
+    send_important_email, subscription = create_subscription(stream)
+
+    payload = []
+
+    subscription.subscribe(payload.append)
+    send_important_email(Email(
+        from_='[email protected]',
+        subject='Message',
+        message='Tests are good',
+        unread=True,
+    ))
+    send_important_email(Email(
+        from_='[email protected]',
+        subject='Message 2',
+        message='Tests are good 2',
+        unread=True,
+    ))
+
+    expected_payload1 = {
+        'importantEmail': {
+            'email': {
+                'from': '[email protected]',
+                'subject': 'Message',
+            },
+            'inbox': {
+                'unread': 1,
+                'total': 2,
+            },
+        }
+    }
+
+    expected_payload2 = {
+        'importantEmail': {
+            'email': {
+                'from': '[email protected]',
+                'subject': 'Message 2',
+            },
+            'inbox': {
+                'unread': 2,
+                'total': 3,
+            },
+        }
+    }
+
+    assert len(payload) == 2
+    assert payload[0].data == expected_payload1
+    assert payload[1].data == expected_payload2