def test_node():
message = message_factory.create("NodeCreateVMReq")
# server
server = Node(HOST, PORT)
server.createVMReqHandler = MagicMock(return_value="hello world")
# test start threading
server.sayHello = MagicMock(return_value="hello")
server.start_thread(target=server.sayHello)
def server_thread():
server.run()
t = threading.Thread(target = server_thread)
t.start()
# client
try:
sleep(3)
ret = client.send_message((HOST, PORT), message)
assert ret == "hello world"
client.sendonly_message((HOST, PORT), message)
finally:
server.shutdown()
server.sayHello.assert_called_with()
server.createVMReqHandler.assert_called_with(message, ('127.0.0.1', ANY))
def test_node():
message = message_factory.create("NodeCreateVMReq")
# server
server = Node(HOST, PORT)
server.createVMReqHandler = MagicMock(return_value="hello world")
# test start threading
server.sayHello = MagicMock(return_value="hello")
server.start_thread(target=server.sayHello)
def server_thread():
server.run()
t = threading.Thread(target=server_thread)
t.start()
# client
try:
sleep(3)
ret = client.send_message((HOST, PORT), message)
assert ret == "hello world"
client.sendonly_message((HOST, PORT), message)
finally:
server.shutdown()
server.sayHello.assert_called_with()
server.createVMReqHandler.assert_called_with(message, ('127.0.0.1', ANY))
