def test_NodeInfoStoresNodeData(self):
"""Checks if node info data are stored in messageBroker instance."""
message = {"messagetype" : "nodeinfo",
"data" : [{"nodeid" : "testnodeid", "description" : "testdescr"}]}
mbroker = MessBro(None, None, "")
mbroker.nodeInfo(message)
self.assertEqual(message["data"], mbroker._hardware)
def test_NodeInfoMessage(self):
"""MessageBroker handles messageType nodeinfo."""
mock = MessMock()
message = {"messagetype" : "nodeinfo",
"data" : {}}
mbroker = MessBro(None, None, "")
mbroker.nodeInfo = mock.mockNodeInfo
mbroker.parseJsonMessage(message)
self.assertTrue(mock.mockNodeInfoCalled)
self.assertEqual(message, mock.message)