class Test(base.CQSTest):
def setUp(self):
self.cluster = Cluster(Queue())
self.username = "******"
self.address = "some_address"
self.port = 22
def test_add_node(self):
self.cluster.add_node(self.username, self.address, self.port)
self.assertEquals(1, self.cluster.number_of_nodes_available())
def test_number_of_nodes_available(self):
self.assertEquals(0, self.cluster.number_of_nodes_available())
def test_no_json_available(self):
self.assertEquals(None, self.cluster.get_json())
def test_get_node(self):
node = ClusterNode(self.username, self.address, self.port)
self.assertTrue(node.is_available())
self.cluster.add_node(self.username, self.address, self.port)
node = self.cluster.get_available_node()
self.assertTrue(node is not None)
self.assertEquals(node.address, self.address)
self.assertEquals(node.username, self.username)
self.assertEquals(node.port, self.port)
def test_construct_command(self):
node = ClusterNode(self.username, self.address, self.port)
constructed_command = node.construct_command("ls -l")
self.assertEquals("ssh cooluser@some_address 'ls -l'", constructed_command)
def setUp(self):
self.cluster = Cluster(Queue())
self.username = "******"
self.address = "some_address"
self.port = 22