def test_on_request_with_process_name(self, _, _2):
ws = WebsocketServerComms(self.p, dict(port=1))
request = MagicMock(fields=dict(endpoint="anything"),
endpoint=[".", "blocks"])
ws.on_request(request)
self.p.q.put.assert_called_once_with(request)
self.assertEqual(request.endpoint, [self.p.name, "blocks"])
def test_send_to_client_error(self, _, _2):
ws = WebsocketServerComms(self.p, dict(port=1))
response = Error(11, MagicMock(), "bad")
ws._send_to_client(response)
response.context.set_status.assert_called_once_with(500, "bad")
response.context.write_error.assert_called_once_with(500)
def setUp(self):
self.sf = SyncFactory("sync")
self.process = Process("proc", self.sf)
part = HelloPart(self.process, None)
DefaultController("hello", self.process, parts={"hello": part})
WebsocketServerComms(self.process, dict(port=self.socket))
self.process.start()
def test_send_to_client_unknown(self, _, _2):
ws = WebsocketServerComms(self.p, dict(port=1))
response = MagicMock()
ws._send_to_client(response)
response.context.set_status.assert_called_once_with(
500, "Unknown response %s" % type(response))
response.context.write_error.assert_called_once_with(500)
def test_start(self, _, _2):
self.p.spawn = MagicMock()
self.WS = WebsocketServerComms(self.p, dict(port=1))
self.WS.start()
self.assertEqual([call(self.WS.send_loop),
call(self.WS.loop.start)],
self.p.spawn.call_args_list)
def test_send_to_client_return(self, _, _2, json_mock):
ws = WebsocketServerComms(self.p, dict(port=1))
response = MagicMock(spec=Return)
response.value = MagicMock()
response.context = MagicMock()
ws._send_to_client(response)
json_mock.dumps.assert_called_once_with(response.value.to_dict())
response.context.finish.assert_called_once_with(
json_mock.dumps.return_value + "\n")
def test_send_to_client_error(self, _, _2, json_mock):
ws = WebsocketServerComms(self.p, dict(port=1))
response = MagicMock(spec=Error)
response.context = MagicMock()
response.message = MagicMock()
ws._send_to_client(response)
response.context.set_status.assert_called_once_with(
500, response.message)
response.context.write_error.assert_called_once_with(500)
def test_send_to_client(self, _, _2, json_mock):
self.WS = WebsocketServerComms(self.p, dict(port=1))
response_mock = MagicMock()
response_mock.context = MagicMock(spec=MalcWebSocketHandler)
self.WS._send_to_client(response_mock)
json_mock.dumps.assert_called_once_with(response_mock.to_dict())
response_mock.context.write_message.assert_called_once_with(
json_mock.dumps())
def test_wait(self, ioloop_mock, server_mock):
spawnable_mocks = [MagicMock(), MagicMock()]
timeout = MagicMock()
self.p.spawn = MagicMock(side_effect=spawnable_mocks)
self.WS = WebsocketServerComms(self.p, dict(port=1))
self.WS.start()
self.WS.wait(timeout)
spawnable_mocks[0].wait.assert_called_once_with(timeout=timeout)
spawnable_mocks[1].wait.assert_called_once_with(timeout=timeout)
def test_stop(self, ioloop_mock, server_mock):
loop_mock = MagicMock()
ioloop_mock.current.return_value = loop_mock
self.p.spawn = MagicMock()
self.WS = WebsocketServerComms(self.p, dict(port=1))
self.WS.start()
self.WS.stop()
self.assertEqual([call(self.WS.server.stop),
call(self.WS.loop.stop)],
loop_mock.add_callback.call_args_list)
self.p.spawn.return_value.wait.assert_not_called()
def setUp(self):
self.sf = SyncFactory("sync")
self.process = Process("proc", self.sf)
DefaultController("hello",
self.process,
parts=dict(hello=HelloPart(self.process, None)))
DefaultController("counter",
self.process,
parts=dict(counter=CounterPart(self.process, None)))
WebsocketServerComms(self.process, dict(port=self.socket))
self.process.start()
self.process2 = Process("proc2", self.sf)
WebsocketClientComms(self.process2,
dict(hostname="localhost", port=self.socket))
self.process2.start()
def setUp(self):
sf = SyncFactory("sync")
self.process = Process("proc", sf)
Hello(self.process, dict(mri="hello"))
Counter(self.process, dict(mri="counter"))
self.process.add_comms(
WebsocketServerComms(self.process, dict(port=self.socket)))
self.process.start()
# If we don't wait long enough, sometimes the websocket_connect()
# in process2 will hang...
time.sleep(0.1)
self.process2 = Process("proc2", sf)
self.process2.add_comms(
WebsocketClientComms(self.process2,
dict(hostname="localhost", port=self.socket)))
self.process2.start()
def test_MWSH_on_message(self, _, _1, json_mock, deserialize_mock):
self.WS = WebsocketServerComms(self.p, dict(port=1))
message_dict = dict(name="TestMessage")
json_mock.loads.return_value = message_dict
request = MagicMock()
request.context = self.WS.server.request_callback.handlers[0][1][
1].handler_class
deserialize_mock.return_value = request
m = MagicMock()
MWSH = MalcWebSocketHandler(m, m)
self.WS.server.request_callback.handlers[0][1][
1].handler_class.on_message(MWSH, "TestMessage")
json_mock.loads.assert_called_once_with("TestMessage",
object_pairs_hook=OrderedDict)
deserialize_mock.assert_called_once_with(message_dict, Request)
self.p.q.put.assert_called_once_with(request)
def test_listen_called(self, ioloop_mock, listen_mock):
self.WS = WebsocketServerComms(self.p, dict(port=1))
self.assertEqual(ioloop_mock.current(), self.WS.loop)
def setUp(self):
self.process = Process("proc", SyncFactory("sync"))
Hello(self.process, dict(mri="hello"))
self.process.add_comms(
WebsocketServerComms(self.process, dict(port=self.socket)))
self.process.start()
def test_send_to_client(self, _, _2):
ws = WebsocketServerComms(self.p, dict(port=1))
response = Return(11, MagicMock(spec=MalcWebSocketHandler), "me")
ws._send_to_client(response)
response.context.write_message.assert_called_once_with(
'{"typeid": "malcolm:core/Return:1.0", "id": 11, "value": "me"}')
def test_send_to_client_return(self, _, _2):
ws = WebsocketServerComms(self.p, dict(port=1))
response = Return(11, MagicMock(), "me")
ws._send_to_client(response)
response.context.finish.assert_called_once_with('"me"\n')
def test_init(self, ioloop_mock, server_mock):
self.WS = WebsocketServerComms(self.p, dict(port=1))
self.assertEqual(self.p, self.WS.process)
self.assertEqual(server_mock(), self.WS.server)
self.assertEqual(ioloop_mock.current(), self.WS.loop)