class TestGet(unittest.TestCase): def setUp(self): self.callback = MagicMock() self.path = ["BL18I:XSPRESS3", "state", "value"] self.o = Get(32, self.path) self.o.set_callback(self.callback) def test_init(self): assert self.o.typeid == "malcolm:core/Get:1.0" assert self.o.id == 32 assert self.o.callback == self.callback assert self.path == self.o.path def test_setters(self): self.o.path = ["BL18I:XSPRESS3"] assert get_doc_json("get_xspress3") == self.o.to_dict() def test_doc_state(self): assert get_doc_json("get_xspress3_state_value") == self.o.to_dict()
def test_MWSH_on_message(self, ioloop_mock, server_mock): MWSH = MalcWebSocketHandler(MagicMock(), MagicMock()) MWSH.servercomms = MagicMock() request = Get(None, None, ["block", "attr"]) request.set_id(54) message = """{ "typeid": "malcolm:core/Get:1.0", "id": 54, "endpoint": ["block", "attr"] }""" MWSH.on_message(message) self.assertEquals(MWSH.servercomms.on_request.call_count, 1) actual = MWSH.servercomms.on_request.call_args[0][0] self.assertEquals(actual.to_dict(), request.to_dict())