def test_context_operations():
ws = mock.Mock()
context = base.BaseConnectionContext(ws)
assert not context.has_operation(1)
context.register_operation(1, None)
assert context.has_operation(1)
context.remove_operation(1)
assert not context.has_operation(1)
# Removing a non-existant operation fails silently.
context.remove_operation(999)
def test_observer_data():
ws = mock.Mock()
context = base.BaseConnectionContext(ws)
send_result, send_error, send_message = mock.Mock(), mock.Mock(
), mock.Mock()
observer = SubscriptionObserver(
connection_context=context,
op_id=1,
send_execution_result=send_result,
send_error=send_error,
send_message=send_message,
)
observer.on_next("data")
assert send_result.called
assert not send_error.called
def test_close(self):
with pytest.raises(NotImplementedError):
base.BaseConnectionContext(ws=None).close(123)
def test_send(self):
with pytest.raises(NotImplementedError):
base.BaseConnectionContext(ws=None).send("TEST")
def test_receive(self):
with pytest.raises(NotImplementedError):
base.BaseConnectionContext(ws=None).receive()
def test_no_operations_initially(self):
cc = base.BaseConnectionContext(ws=None)
assert not cc.operations
def cc():
cc = base.BaseConnectionContext(ws=None)
cc.operations = {"yes": "1"}
return cc
