def test_send_and_recv(self, mocker):
msg = BatsimMessage(0)
mocker.patch.object(protocol.zmq.Socket, 'recv_json', return_value=msg)
p = NetworkHandler("tcp://localhost:28000")
p.bind()
m = p.send_and_recv(msg)
assert m == msg
protocol.zmq.Socket.recv_json.assert_called_once()
protocol.zmq.Socket.send_json.assert_called_once_with(msg.to_json())
def test_to_json_must_include_requests_only(self):
events = [
RequestedCallBatsimEvent(15),
KillJobBatsimRequest(10, "w!1", "w!2")
]
m = BatsimMessage(15, events)
events_json = [
BatsimRequestAPI.get_kill_job(10, ["w!1", "w!2"]),
]
m_jsn = BatsimAPI.get_message(15, events_json)
assert m.to_json() == m_jsn
def test_send(self):
msg = BatsimMessage(0)
p = NetworkHandler("tcp://localhost:28000")
p.bind()
p.send(msg)
protocol.zmq.Socket.send_json.assert_called_once_with(msg.to_json())
