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_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)
