def setUp(self):
server_port = 9999
self.controller = ServerController(server_port)
self.controller.start_server()
self.username = "******"
self.node_address = "some_address"
self.node_port = 22
class Test(base.CQSTest):
def setUp(self):
server_port = 9999
self.controller = ServerController(server_port)
self.controller.start_server()
self.username = "******"
self.node_address = "some_address"
self.node_port = 22
def tearDown(self):
self.controller.stop_server()
def test_client_sent_invalid_json(self):
invalid_json = "bad command bro"
SendCommandFromClient(invalid_json)
retreived_command = self.controller.cluster.get_json()
self.assertTrue(retreived_command is None)
@mock.patch.object(ClusterNode, 'send_command')
def test_json_fits_schema_request_with_node_available(self, mock_send_command):
self.controller.cluster.add_node(self.username, self.node_address, self.node_port)
valid_json = '{"name": "any", "command":"Rossum"}'
SendCommandFromClient(valid_json)
time.sleep(2)
self.assertEquals(None, self.controller.cluster.get_json())
@mock.patch.object(ClusterNode, 'send_command')
def test_json_fits_schema_request_without_node_available(self, mock_send_command):
#self.controller.cluster.add_node(self.username, self.node_address, self.node_port)
valid_json = '{"name": "any", "command":"Rossum"}'
SendCommandFromClient(valid_json)
time.sleep(2)
self.assertEquals(json.loads(valid_json), self.controller.cluster.get_json())
@mock.patch.object(ClusterNode, 'send_command')
def test_handoff_command_to_node(self, mock_send_command):
valid_json = '{"name": "any", "command":"Rossum"}'
parsed_json = json.loads(valid_json)
command_to_send = parsed_json['command']
self.controller.cluster.add_node(self.username, self.node_address, self.node_port)
self.assertEquals(1, self.controller.cluster.number_of_nodes_available())
self.controller.cluster.send_command_to_node(command_to_send)
mock_send_command.assert_called_with(command_to_send)
class Test(base.CQSTest):
def setUp(self):
server_port = 9999
self.controller = ServerController(server_port)
def assert_thread(self, thread):
time.sleep(0.1)
self.assertTrue(thread is not None)
self.assertTrue(thread.isAlive())
def test_start_and_stop_webserver(self):
self.controller.start_server()
self.assert_thread(self.controller.server_thread)
self.controller.stop_server()
self.assertFalse(self.controller.server_thread.isAlive())
def test_start_poll_for_commands(self):
self.controller.start_server()
self.assert_thread(self.controller.cluster_thread)
self.controller.stop_server()
self.assertFalse(self.controller.cluster_thread.isAlive())
def setUp(self):
server_port = 9999
self.controller = ServerController(server_port)
