From 9add0b1dc1db9725de183d8ae8840994558ce6da Mon Sep 17 00:00:00 2001
From: Houkime <>
Date: Wed, 15 May 2024 18:15:16 +0000
Subject: [PATCH] test(websocket) test connection init

---
 tests/test_graphql/test_websocket.py | 48 ++++++++++++++++++++++++++--
 1 file changed, 46 insertions(+), 2 deletions(-)

diff --git a/tests/test_graphql/test_websocket.py b/tests/test_graphql/test_websocket.py
index fb2ac33..2431285 100644
--- a/tests/test_graphql/test_websocket.py
+++ b/tests/test_graphql/test_websocket.py
@@ -1,6 +1,50 @@
+from tests.common import generate_jobs_subscription
+from selfprivacy_api.graphql.queries.jobs import Job as _Job
+from selfprivacy_api.jobs import Jobs
+
+# JOBS_SUBSCRIPTION = """
+# jobUpdates {
+#     uid
+#     typeId
+#     name
+#     description
+#     status
+#     statusText
+#     progress
+#     createdAt
+#     updatedAt
+#     finishedAt
+#     error
+#     result
+# }
+# """
+
 
 def test_websocket_connection_bare(authorized_client):
-    client =authorized_client
-    with client.websocket_connect('/graphql', subprotocols=[ "graphql-transport-ws","graphql-ws"] ) as websocket:
+    client = authorized_client
+    with client.websocket_connect(
+        "/graphql", subprotocols=["graphql-transport-ws", "graphql-ws"]
+    ) as websocket:
         assert websocket is not None
         assert websocket.scope is not None
+
+
+def test_websocket_graphql_init(authorized_client):
+    client = authorized_client
+    with client.websocket_connect(
+        "/graphql", subprotocols=["graphql-transport-ws"]
+    ) as websocket:
+        websocket.send_json({"type": "connection_init", "payload": {}})
+        ack = websocket.receive_json()
+        assert ack == {"type": "connection_ack"}
+
+
+# def test_websocket_subscription(authorized_client):
+#     client = authorized_client
+#     with client.websocket_connect(
+#         "/graphql", subprotocols=["graphql-transport-ws", "graphql-ws"]
+#     ) as websocket:
+#         websocket.send(generate_jobs_subscription([JOBS_SUBSCRIPTION]))
+#         Jobs.add("bogus","bogus.bogus", "yyyaaaaayy")
+#         joblist = websocket.receive_json()
+#         raise NotImplementedError(joblist)