def test_send_message(): nod = Node('public_key', 'private_key', MagicMock()) host = '127.0.0.1' port = 8888 protocol = MagicMock() nod._contacts[(host, port)] = protocol msg = { 'type': 'message', 'message': 'hello' } nod.send_message(host, port, msg) assert 1 == protocol.sendMessage.call_count assert msg == protocol.sendMessage.call_args[0][0]
def test_send_message(): nod = Node('public_key', 'private_key', MagicMock()) host = '127.0.0.1' port = 8888 nod.send_message = MagicMock() class Screen(object): def __init__(self): self.addLine = MagicMock() self.peer_host = host self.peer_port = port self._node = nod screen = Screen() send_message(screen, 'hello') assert 1 == screen._node.send_message.call_count assert host == screen._node.send_message.call_args[0][0] assert port == screen._node.send_message.call_args[0][1] expected = {'type': 'message', 'message': 'hello'} assert expected == screen._node.send_message.call_args[0][2]