def test_cloud_id_missing_instance_data_json(self):
"""Exit error when the provided instance-data.json does not exist."""
cmd = ['cloud-id', '--instance-data', self.instance_data]
with mock.patch('sys.argv', cmd):
with mock.patch('sys.stderr', new_callable=StringIO) as m_stderr:
with self.assertRaises(SystemExit) as context_manager:
cloud_id.main()
self.assertEqual(1, context_manager.exception.code)
self.assertIn("ERROR: File not found '%s'" % self.instance_data,
m_stderr.getvalue())
def test_cloud_id_missing_instance_data_json(self, get_status_details,
tmpdir, capsys):
"""Exit error when the provided instance-data.json does not exist."""
get_status_details.return_value = cloud_id.UXAppStatus.DONE, "n/a", ""
instance_data = tmpdir.join("instance-data.json")
cmd = ["cloud-id", "--instance-data", instance_data.strpath]
with mock.patch("sys.argv", cmd):
with pytest.raises(SystemExit) as context_manager:
cloud_id.main()
assert 1 == context_manager.value.code
_out, err = capsys.readouterr()
assert "Error:\nFile not found '%s'" % instance_data.strpath in err
def test_cloud_id_long_name_from_instance_data(self):
"""Report long cloud-id format from cloud_name and region."""
util.write_file(
self.instance_data,
'{"v1": {"cloud_name": "mycloud", "region": "somereg"}}')
cmd = ['cloud-id', '--instance-data', self.instance_data, '--long']
with mock.patch('sys.argv', cmd):
with mock.patch('sys.stdout', new_callable=StringIO) as m_stdout:
with self.assertRaises(SystemExit) as context_manager:
cloud_id.main()
self.assertEqual(0, context_manager.exception.code)
self.assertEqual("mycloud\tsomereg\n", m_stdout.getvalue())
def test_cloud_id_non_json_instance_data(self):
"""Exit error when the provided instance-data.json is not json."""
cmd = ['cloud-id', '--instance-data', self.instance_data]
util.write_file(self.instance_data, '{')
with mock.patch('sys.argv', cmd):
with mock.patch('sys.stderr', new_callable=StringIO) as m_stderr:
with self.assertRaises(SystemExit) as context_manager:
cloud_id.main()
self.assertEqual(1, context_manager.exception.code)
self.assertIn(
"ERROR: File '%s' is not valid json." % self.instance_data,
m_stderr.getvalue())
def test_cloud_id_unique_exit_codes_for_status(self, get_status_details,
status, exit_code, tmpdir,
capsys):
"""cloud-id returns unique exit codes for status."""
get_status_details.return_value = status, "n/a", ""
instance_data = tmpdir.join("instance-data.json")
if status == cloud_id.UXAppStatus.RUNNING:
instance_data.write("{}")
cmd = ["cloud-id", "--instance-data", instance_data.strpath, "--json"]
with mock.patch("sys.argv", cmd):
with pytest.raises(SystemExit) as context_manager:
cloud_id.main()
assert exit_code == context_manager.value.code
def test_cloud_id_lookup_from_instance_data_region(self):
"""Report discovered canonical cloud_id when region lookup matches."""
util.write_file(
self.instance_data,
'{"v1": {"cloud_name": "aws", "region": "cn-north-1",'
' "platform": "ec2"}}')
cmd = ['cloud-id', '--instance-data', self.instance_data, '--long']
with mock.patch('sys.argv', cmd):
with mock.patch('sys.stdout', new_callable=StringIO) as m_stdout:
with self.assertRaises(SystemExit) as context_manager:
cloud_id.main()
self.assertEqual(0, context_manager.exception.code)
self.assertEqual("aws-china\tcn-north-1\n", m_stdout.getvalue())
def test_cloud_id_from_cloud_name_in_instance_data(self):
"""Report canonical cloud-id from cloud_name in instance-data."""
util.write_file(
self.instance_data,
'{"v1": {"cloud_name": "mycloud", "region": "somereg"}}',
)
cmd = ["cloud-id", "--instance-data", self.instance_data]
with mock.patch("sys.argv", cmd):
with mock.patch("sys.stdout", new_callable=StringIO) as m_stdout:
with self.assertRaises(SystemExit) as context_manager:
cloud_id.main()
self.assertEqual(0, context_manager.exception.code)
self.assertEqual("mycloud\n", m_stdout.getvalue())
def test_cloud_id_long_name_from_instance_data(self, get_status_details,
tmpdir, capsys):
"""Report long cloud-id format from cloud_name and region."""
get_status_details.return_value = cloud_id.UXAppStatus.DONE, "n/a", ""
instance_data = tmpdir.join("instance-data.json")
instance_data.write(
'{"v1": {"cloud_name": "mycloud", "region": "somereg"}}', )
cmd = ["cloud-id", "--instance-data", instance_data.strpath, "--long"]
with mock.patch("sys.argv", cmd):
with pytest.raises(SystemExit) as context_manager:
cloud_id.main()
out, _err = capsys.readouterr()
assert 0 == context_manager.value.code
assert "mycloud\tsomereg\n" == out
def test_cloud_id_non_json_instance_data(self, get_status_details, tmpdir,
capsys):
"""Exit error when the provided instance-data.json is not json."""
get_status_details.return_value = cloud_id.UXAppStatus.DONE, "n/a", ""
instance_data = tmpdir.join("instance-data.json")
cmd = ["cloud-id", "--instance-data", instance_data.strpath]
instance_data.write("{")
with mock.patch("sys.argv", cmd):
with pytest.raises(SystemExit) as context_manager:
cloud_id.main()
assert 1 == context_manager.value.code
_out, err = capsys.readouterr()
assert ("Error:\nFile '%s' is not valid json." % instance_data.strpath
in err)
def test_cloud_id_lookup_from_instance_data_region(self,
get_status_details,
tmpdir, capsys):
"""Report discovered canonical cloud_id when region lookup matches."""
get_status_details.return_value = cloud_id.UXAppStatus.DONE, "n/a", ""
instance_data = tmpdir.join("instance-data.json")
instance_data.write(
'{"v1": {"cloud_name": "aws", "region": "cn-north-1",'
' "platform": "ec2"}}', )
cmd = ["cloud-id", "--instance-data", instance_data.strpath, "--long"]
with mock.patch("sys.argv", cmd):
with pytest.raises(SystemExit) as context_manager:
cloud_id.main()
assert 0 == context_manager.value.code
out, _err = capsys.readouterr()
assert "aws-china\tcn-north-1\n" == out
def test_cloud_id_lookup_json_instance_data_adds_cloud_id_to_json(self):
"""Report v1 instance-data content with cloud_id when --json set."""
util.write_file(
self.instance_data,
'{"v1": {"cloud_name": "unknown", "region": "dfw",'
' "platform": "openstack", "public_ssh_keys": []}}')
expected = util.json_dumps({
'cloud_id': 'openstack',
'cloud_name': 'unknown',
'platform': 'openstack',
'public_ssh_keys': [],
'region': 'dfw'
})
cmd = ['cloud-id', '--instance-data', self.instance_data, '--json']
with mock.patch('sys.argv', cmd):
with mock.patch('sys.stdout', new_callable=StringIO) as m_stdout:
with self.assertRaises(SystemExit) as context_manager:
cloud_id.main()
self.assertEqual(0, context_manager.exception.code)
self.assertEqual(expected + '\n', m_stdout.getvalue())
def test_cloud_id_lookup_json_instance_data_adds_cloud_id_to_json(
self, get_status_details, tmpdir, capsys):
"""Report v1 instance-data content with cloud_id when --json set."""
get_status_details.return_value = cloud_id.UXAppStatus.DONE, "n/a", ""
instance_data = tmpdir.join("instance-data.json")
instance_data.write(
'{"v1": {"cloud_name": "unknown", "region": "dfw",'
' "platform": "openstack", "public_ssh_keys": []}}', )
expected = util.json_dumps({
"cloud_id": "openstack",
"cloud_name": "unknown",
"platform": "openstack",
"public_ssh_keys": [],
"region": "dfw",
})
cmd = ["cloud-id", "--instance-data", instance_data.strpath, "--json"]
with mock.patch("sys.argv", cmd):
with pytest.raises(SystemExit) as context_manager:
cloud_id.main()
out, _err = capsys.readouterr()
assert 0 == context_manager.value.code
assert expected + "\n" == out
