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_base_server():
# server
server = BaseServer(HOST, PORT)
server.register_handle_function("CmdHello", hello)
holder = {'hello': ""}
def sayHello():
holder['hello'] = "world"
server.register_start_function(sayHello)
def server_thread():
server.run()
t = threading.Thread(target = server_thread)
t.start()
sleep(4)
# client
try:
ret = client.send_message((HOST, PORT), CmdHello("world"))
assert ret == "hello world"
client.sendonly_message((HOST, PORT), CmdHello("world"))
finally:
server.shutdown()
assert holder["hello"] == "world"
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_base_server():
# server
server = BaseServer(HOST, PORT)
server.register_handle_function("CmdHello", hello)
holder = {'hello': ""}
def sayHello():
holder['hello'] = "world"
server.register_start_function(sayHello)
def server_thread():
server.run()
t = threading.Thread(target=server_thread)
t.start()
sleep(4)
# client
try:
ret = client.send_message((HOST, PORT), CmdHello("world"))
assert ret == "hello world"
client.sendonly_message((HOST, PORT), CmdHello("world"))
finally:
server.shutdown()
assert holder["hello"] == "world"
