def test_with_exception(self, initialize_mock): initialize_mock.side_effect = ValueError with tempfile.NamedTemporaryFile() as f: f.write('{}\n'.encode('utf-8')) f.flush() sys.argv.append(f.name) return_code = setup_neutron.main() self.assertEqual(1, return_code)
def test_with_arguments(self, initialize_mock): network_desc = {'physical': {'metadata_server': 'foo.bar'}} with tempfile.NamedTemporaryFile() as f: f.write(json.dumps(network_desc).encode('utf-8')) f.flush() sys.argv.append(f.name) return_code = setup_neutron.main() initialize_mock.assert_called_once_with(mock.ANY) self.assertEqual(0, return_code)
def test_with_arguments(self, initialize_mock, get_neutron_client_mock, get_keystone_client_mock): network_desc = {'physical': {'metadata_server': 'foo.bar'}} with tempfile.NamedTemporaryFile() as f: f.write(json.dumps(network_desc).encode('utf-8')) f.flush() sys.argv.append(f.name) return_code = setup_neutron.main() get_keystone_client_mock.assert_called_once_with() get_neutron_client_mock.assert_called_once_with() initialize_mock.assert_called_once_with( network_desc, neutron_client='neutron_client_mock', keystone_client='keystone_client_mock') self.assertEqual(0, return_code)