summaryrefslogtreecommitdiff
path: root/tests/unittests/cmd/test_cloud_id.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/unittests/cmd/test_cloud_id.py')
-rw-r--r--tests/unittests/cmd/test_cloud_id.py99
1 files changed, 58 insertions, 41 deletions
diff --git a/tests/unittests/cmd/test_cloud_id.py b/tests/unittests/cmd/test_cloud_id.py
index 9a010402..42941d4f 100644
--- a/tests/unittests/cmd/test_cloud_id.py
+++ b/tests/unittests/cmd/test_cloud_id.py
@@ -2,41 +2,45 @@
"""Tests for cloud-id command line utility."""
-from cloudinit import util
from collections import namedtuple
from io import StringIO
+from cloudinit import util
from cloudinit.cmd import cloud_id
-
from tests.unittests.helpers import CiTestCase, mock
class TestCloudId(CiTestCase):
- args = namedtuple('cloudidargs', ('instance_data json long'))
+ args = namedtuple("cloudidargs", "instance_data json long")
def setUp(self):
super(TestCloudId, self).setUp()
self.tmp = self.tmp_dir()
- self.instance_data = self.tmp_path('instance-data.json', dir=self.tmp)
+ self.instance_data = self.tmp_path("instance-data.json", dir=self.tmp)
def test_cloud_id_arg_parser_defaults(self):
"""Validate the argument defaults when not provided by the end-user."""
- cmd = ['cloud-id']
- with mock.patch('sys.argv', cmd):
+ cmd = ["cloud-id"]
+ with mock.patch("sys.argv", cmd):
args = cloud_id.get_parser().parse_args()
self.assertEqual(
- '/run/cloud-init/instance-data.json',
- args.instance_data)
+ "/run/cloud-init/instance-data.json", args.instance_data
+ )
self.assertEqual(False, args.long)
self.assertEqual(False, args.json)
def test_cloud_id_arg_parse_overrides(self):
"""Override argument defaults by specifying values for each param."""
- util.write_file(self.instance_data, '{}')
- cmd = ['cloud-id', '--instance-data', self.instance_data, '--long',
- '--json']
- with mock.patch('sys.argv', cmd):
+ util.write_file(self.instance_data, "{}")
+ cmd = [
+ "cloud-id",
+ "--instance-data",
+ self.instance_data,
+ "--long",
+ "--json",
+ ]
+ with mock.patch("sys.argv", cmd):
args = cloud_id.get_parser().parse_args()
self.assertEqual(self.instance_data, args.instance_data)
self.assertEqual(True, args.long)
@@ -44,37 +48,40 @@ class TestCloudId(CiTestCase):
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:
+ 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:\nFile not found '%s'" % self.instance_data,
- m_stderr.getvalue())
+ m_stderr.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:
+ 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:\nFile '%s' is not valid json." % self.instance_data,
- m_stderr.getvalue())
+ m_stderr.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:
+ '{"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)
@@ -84,10 +91,11 @@ class TestCloudId(CiTestCase):
"""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:
+ '{"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)
@@ -98,10 +106,11 @@ class TestCloudId(CiTestCase):
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:
+ ' "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)
@@ -112,16 +121,24 @@ class TestCloudId(CiTestCase):
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:
+ ' "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())
+ self.assertEqual(expected + "\n", m_stdout.getvalue())
+
# vi: ts=4 expandtab
generated by cgit v1.2.3 (git 2.39.1) at 2024-11-13 01:00:20 +0000