def test_sendMessage(): proto = MessageProtocol() proto.sendString = MagicMock() msg_dict = { 'type': 'message', 'message': 'hello' } proto.sendMessage(msg_dict) assert 1 == proto.sendString.call_count assert json.dumps(msg_dict) == proto.sendString.call_args[0][0]
def test_error(): proto = MessageProtocol() proto.sendMessage = MagicMock() expected_message = { 'type': 'error', 'description': 'BANG!' } proto.error('BANG!') assert 1 == proto.sendMessage.call_count assert expected_message == proto.sendMessage.call_args[0][0]