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)