def setUp(self): super(TestNodeCommand, self).setUp() self.m_client.get_all.return_value = [ fake_node.get_fake_node() for i in range(10) ] self.m_client.get_by_id.return_value = fake_node.get_fake_node()
def test_node_list_ansible_inventory(self): self.m_client.get_all.return_value = [ fake_node.get_fake_node(hostname='node-1', ip='10.20.0.2', roles=['compute']), fake_node.get_fake_node(hostname='node-2', ip='10.20.0.3', roles=['compute', 'ceph-osd']), fake_node.get_fake_node(hostname='node-3', ip='10.20.0.4', roles=['controller']), fake_node.get_fake_node(hostname='node-4', ip='10.20.0.5', roles=['controller', 'mongo']), fake_node.get_fake_node(hostname='node-5', ip='10.20.0.6', roles=['controller', 'ceph-osd']), ] expected_output = '''\ [ceph-osd] node-2 ansible_host=10.20.0.3 node-5 ansible_host=10.20.0.6 [compute] node-1 ansible_host=10.20.0.2 node-2 ansible_host=10.20.0.3 [controller] node-3 ansible_host=10.20.0.4 node-4 ansible_host=10.20.0.5 node-5 ansible_host=10.20.0.6 [mongo] node-4 ansible_host=10.20.0.5 ''' args = 'node ansible-inventory --env 1' with mock.patch('sys.stdout', new=io.StringIO()) as mstdout: rv = self.exec_command(args) actual_output = mstdout.getvalue() self.assertFalse(rv) self.assertEqual(expected_output, actual_output) self.m_get_client.assert_called_once_with('node', mock.ANY) self.m_client.get_all.assert_called_once_with( environment_id=1, labels=None)
def test_node_set_name(self, m_emit_one): self.m_client._updatable_attributes = \ node.NodeClient._updatable_attributes node_id = 37 expected_field_data = cmd_node.NodeShow.columns test_cases = ('new-name', 'New Name', 'śćż∑ Pó', '你一定是无聊') for name in test_cases: self.m_client.update.return_value = fake_node.get_fake_node( node_id=node_id, node_name=name) cmd = ['node', 'update', str(node_id), '--name', name] # NOTE(sbrzeczkowski): due to shlex inability to accept # unicode arguments prior to python 2.7.3 main_mod.main(argv=cmd) if six.PY2: name = name.decode('utf-8') m_emit_one.assert_called_with(expected_field_data, mock.ANY, mock.ANY, mock.ANY) self.m_get_client.assert_called_once_with('node', mock.ANY) self.m_client.update.assert_called_once_with( node_id, name=name) self.m_get_client.reset_mock() self.m_client.reset_mock()
def setUp(self): super(TestNodeExecuteTasksAction, self).setUp() self.tasks = ['netconfig', 'hiera', 'install'] node_patch = patch('fuelclient.objects.node.Node.get_fresh_data') m_fresh_data = node_patch.start() m_fresh_data.return_value = fake_node.get_fake_node() self.addCleanup(node_patch.stop)
def test_node_set_hostname(self): self.m_client._updatable_attributes = \ node.NodeClient._updatable_attributes node_id = 42 hostname = 'test-name' self.m_client.update.return_value = \ fake_node.get_fake_node(node_id=node_id, hostname=hostname) args = 'node update {node_id} --hostname {hostname}'\ .format(node_id=node_id, hostname=hostname) self.exec_command(args) self.m_get_client.assert_called_once_with('node', mock.ANY) self.m_client.update.assert_called_once_with(node_id, hostname=hostname)
def test_node_set_hostname(self): self.m_client._updatable_attributes = \ node.NodeClient._updatable_attributes node_id = 42 hostname = 'test-name' self.m_client.update.return_value = \ fake_node.get_fake_node(node_id=node_id, hostname=hostname) args = 'node update {node_id} --hostname {hostname}'\ .format(node_id=node_id, hostname=hostname) self.exec_command(args) self.m_get_client.assert_called_once_with('node', mock.ANY) self.m_client.update.assert_called_once_with( node_id, hostname=hostname)
def test_node_set_hostname(self, m_emit_one): self.m_client._updatable_attributes = \ node.NodeClient._updatable_attributes node_id = 42 hostname = 'test-name' expected_field_data = cmd_node.NodeShow.columns self.m_client.update.return_value = \ fake_node.get_fake_node(node_id=node_id, hostname=hostname) args = 'node update {node_id} --hostname {hostname}'\ .format(node_id=node_id, hostname=hostname) self.exec_command(args) m_emit_one.assert_called_once_with(expected_field_data, mock.ANY, mock.ANY, mock.ANY) self.m_get_client.assert_called_once_with('node', mock.ANY) self.m_client.update.assert_called_once_with(node_id, hostname=hostname)
def test_node_set_hostname(self, m_emit_one): self.m_client._updatable_attributes = \ node.NodeClient._updatable_attributes node_id = 42 hostname = 'test-name' expected_field_data = cmd_node.NodeShow.columns self.m_client.update.return_value = \ fake_node.get_fake_node(node_id=node_id, hostname=hostname) args = 'node update {node_id} --hostname {hostname}'\ .format(node_id=node_id, hostname=hostname) self.exec_command(args) m_emit_one.assert_called_once_with(expected_field_data, mock.ANY, mock.ANY, mock.ANY) self.m_get_client.assert_called_once_with('node', mock.ANY) self.m_client.update.assert_called_once_with( node_id, hostname=hostname)
def setUp(self): super(TestNodeCommand, self).setUp() self.m_client.get_all.return_value = [fake_node.get_fake_node() for i in range(10)] self.m_client.get_by_id.return_value = fake_node.get_fake_node()