mirror of
https://git.selfprivacy.org/SelfPrivacy/selfprivacy-rest-api.git
synced 2024-11-10 04:53:11 +00:00
62 lines
1.8 KiB
Python
62 lines
1.8 KiB
Python
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:
|
|
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_graphql_ping(authorized_client):
|
|
client = authorized_client
|
|
with client.websocket_connect(
|
|
"/graphql", subprotocols=["graphql-transport-ws"]
|
|
) as websocket:
|
|
# https://github.com/enisdenjo/graphql-ws/blob/master/PROTOCOL.md#ping
|
|
websocket.send_json({"type": "ping", "payload": {}})
|
|
pong = websocket.receive_json()
|
|
assert pong == {"type": "pong"}
|
|
|
|
|
|
# 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)
|