1 files changed, 147 insertions, 0 deletions
diff --git a/tests/test_push.py b/tests/test_push.py
new file mode 100644
index 0000000..e6d732a
--- /dev/null
+++ b/tests/test_push.py
@@ -0,0 +1,147 @@
+import pytest
+
+from api import *
+
+def test_account_destroy(account, push_server):
+    dev = Device(account, "dev", pcb=push_server.good("09d0114e-3b23-4ba3-8474-efac7433e3ba"))
+    account.destroy_account(account.email, "")
+    p = push_server.wait()
+    assert p[0] == "/09d0114e-3b23-4ba3-8474-efac7433e3ba"
+    assert dev.decrypt(p[2]) == {
+        'command': 'fxaccounts:account_destroyed',
+        'data': {'uid': account.props['uid']},
+        'version': 1,
+    }
+    assert push_server.done()
+
+def test_account_verify(client, push_server, mail_server):
+    account = client.create_account("test@test", "")
+    try:
+        (to, body) = mail_server.wait()
+        assert to == ["test@test"]
+        data = json.loads(base64.urlsafe_b64decode(body.split("#/verify/", maxsplit=1)[1]).decode('utf8'))
+        dev = Device(account, "dev", pcb=push_server.good("8accbe08-4040-44c2-8fd9-cf2669b56cb1"))
+        account.post_a('/recovery_email/verify_code', { 'uid': data['uid'], 'code': data['code'] })
+        p = push_server.wait()
+        assert p[0] == "/8accbe08-4040-44c2-8fd9-cf2669b56cb1"
+        assert p[2] == b''
+        assert push_server.done()
+    finally:
+        account.destroy_account(account.email, "")
+
+def test_session_destroy(client, account, push_server):
+    dev1 = Device(account, "dev1")
+    session = client.login(account.email, "")
+    dev2 = Device(session, "dev2")
+    dev1.update_pcb(push_server.good("e6d21a00-9e5e-4d21-92bc-90860cba836e"))
+    session.destroy_session()
+    p = push_server.wait()
+    assert p[0] == "/e6d21a00-9e5e-4d21-92bc-90860cba836e"
+    assert dev1.decrypt(p[2]) == {
+        'command': 'fxaccounts:device_disconnected',
+        'data': {'id': dev2.id},
+        'version': 1,
+    }
+    assert push_server.done()
+
+def test_device_connected(client, account, push_server):
+    dev1 = Device(account, "dev1", pcb=push_server.good("236b8205-daee-4879-b911-64b1f4fd8fd7"))
+    session = client.login(account.email, "")
+    dev2 = Device(session, "dev2")
+    p = push_server.wait()
+    assert p[0] == "/236b8205-daee-4879-b911-64b1f4fd8fd7"
+    assert dev1.decrypt(p[2]) == {
+        'command': 'fxaccounts:device_connected',
+        'data': {'deviceName': 'dev2'},
+        'version': 1,
+    }
+    assert push_server.done()
+
+def test_device_invoke(account, login, push_server):
+    dev = Device(account, "dev1", commands={'a':'a'}, pcb=push_server.good("3610b071-e2ef-4daa-a4e3-eaa74e50f2a0"))
+    account.post_a("/account/devices/invoke_command", {
+        "target": dev.id,
+        "command": "a",
+        "payload": {"data": "foo"},
+        "ttl": 10,
+    })
+    p = push_server.wait()
+    assert p[0] == "/3610b071-e2ef-4daa-a4e3-eaa74e50f2a0"
+    msg = dev.decrypt(p[2])
+    # NOTE needed because index is unpredictable due to pg sequence use
+    del msg['data']['index']
+    del msg['data']['url']
+    assert msg == {
+        'command': 'fxaccounts:command_received',
+        'data': {'command': 'a', 'sender': dev.id},
+        'version': 1,
+    }
+    assert push_server.done()
+
+def test_expiry(account, login, push_server):
+    dev = Device(account, "dev1", commands={'a':'a'}, pcb=push_server.bad("59ba8fcc-f3b0-4b1f-ac27-36d66e022d1e"))
+    account.post_a("/account/devices/invoke_command", {
+        "target": dev.id,
+        "command": "a",
+        "payload": {"data": "foo"},
+        "ttl": 10,
+    })
+    p = push_server.wait()
+    assert p[0] == "/err/59ba8fcc-f3b0-4b1f-ac27-36d66e022d1e"
+    account.post_a("/account/devices/invoke_command", {
+        "target": dev.id,
+        "command": "a",
+        "payload": {"data": "foo"},
+        "ttl": 10,
+    })
+    with pytest.raises(queue.Empty):
+        push_server.wait()
+    dev.update_pcb(push_server.good("59ba8fcc-f3b0-4b1f-ac27-36d66e022d1e"))
+    account.post_a("/account/devices/invoke_command", {
+        "target": dev.id,
+        "command": "a",
+        "payload": {"data": "foo"},
+        "ttl": 10,
+    })
+    p = push_server.wait()
+    assert p[0] == "/59ba8fcc-f3b0-4b1f-ac27-36d66e022d1e"
+    assert push_server.done()
+
+def test_device_notify(account, login, push_server):
+    dev1 = Device(account, "dev1")
+    dev2 = Device(login, "dev2")
+    dev1.update_pcb(push_server.good("738ac7e3-96ef-461c-880c-0af20e311354"))
+    dev2.update_pcb(push_server.good("85a98191-9486-46e2-877a-1152e3d4af4e"))
+    account.post_a("/account/devices/notify", {
+        "to": "all",
+        "_endpointAction": "accountVerify",
+        "excluded": [dev2.id],
+        "payload": {'a':1},
+        "TTL": 0,
+    })
+    p = push_server.wait()
+    assert p[0] == "/738ac7e3-96ef-461c-880c-0af20e311354"
+    assert dev1.decrypt(p[2]) == {'a':1}
+    account.post_a("/account/devices/notify", {
+        "to": [dev2.id],
+        "_endpointAction": "accountVerify",
+        "payload": {'a':2},
+        "TTL": 0,
+    })
+    p = push_server.wait()
+    assert p[0] == "/85a98191-9486-46e2-877a-1152e3d4af4e"
+    assert dev2.decrypt(p[2]) == {'a':2}
+    assert push_server.done()
+
+def test_profile(account, push_server):
+    dev = Device(account, "dev", pcb=push_server.good("12608154-8942-4f1c-9de2-a56465d27d6e"))
+    profile = account.profile()
+    profile.post_a("/display_name", {"displayName": "foo"})
+    p = push_server.wait()
+    assert p[0] == "/12608154-8942-4f1c-9de2-a56465d27d6e"
+    assert dev.decrypt(p[2]) == {'command': 'fxaccounts:profile_updated', 'version': 1}
+    profile.post_a("/avatar/upload", "doesn't parse the image", headers={'content-type': 'image/png'})
+    p = push_server.wait()
+    assert p[0] == "/12608154-8942-4f1c-9de2-a56465d27d6e"
+    assert dev.decrypt(p[2]) == {'command': 'fxaccounts:profile_updated', 'version': 1}
+    assert push_server.done()