summaryrefslogtreecommitdiff
path: root/tests/unittests/sources/vmware/test_guestcust_util.py
diff options
context:
space:
mode:
authorJames Falcon <james.falcon@canonical.com>2021-12-15 20:16:38 -0600
committerGitHub <noreply@github.com>2021-12-15 19:16:38 -0700
commitbae9b11da9ed7dd0b16fe5adeaf4774b7cc628cf (patch)
tree1fbb3269fc87e39832e3286ef42eefd2b23fcd44 /tests/unittests/sources/vmware/test_guestcust_util.py
parent2bcf4fa972fde686c2e3141c58e640640b44dd00 (diff)
downloadvyos-cloud-init-bae9b11da9ed7dd0b16fe5adeaf4774b7cc628cf.tar.gz
vyos-cloud-init-bae9b11da9ed7dd0b16fe5adeaf4774b7cc628cf.zip
Adopt Black and isort (SC-700) (#1157)
Applied Black and isort, fixed any linting issues, updated tox.ini and CI.
Diffstat (limited to 'tests/unittests/sources/vmware/test_guestcust_util.py')
-rw-r--r--tests/unittests/sources/vmware/test_guestcust_util.py79
1 files changed, 45 insertions, 34 deletions
diff --git a/tests/unittests/sources/vmware/test_guestcust_util.py b/tests/unittests/sources/vmware/test_guestcust_util.py
index 9114f0b9..fc63bcae 100644
--- a/tests/unittests/sources/vmware/test_guestcust_util.py
+++ b/tests/unittests/sources/vmware/test_guestcust_util.py
@@ -21,78 +21,89 @@ class TestGuestCustUtil(CiTestCase):
This test is designed to verify the behavior if vmware-toolbox-cmd
is not installed.
"""
- with mock.patch.object(subp, 'which', return_value=None):
+ with mock.patch.object(subp, "which", return_value=None):
self.assertEqual(
- get_tools_config('section', 'key', 'defaultVal'), 'defaultVal')
+ get_tools_config("section", "key", "defaultVal"), "defaultVal"
+ )
def test_get_tools_config_internal_exception(self):
"""
This test is designed to verify the behavior if internal exception
is raised.
"""
- with mock.patch.object(subp, 'which', return_value='/dummy/path'):
- with mock.patch.object(subp, 'subp',
- return_value=('key=value', b''),
- side_effect=subp.ProcessExecutionError(
- "subp failed", exit_code=99)):
+ with mock.patch.object(subp, "which", return_value="/dummy/path"):
+ with mock.patch.object(
+ subp,
+ "subp",
+ return_value=("key=value", b""),
+ side_effect=subp.ProcessExecutionError(
+ "subp failed", exit_code=99
+ ),
+ ):
# verify return value is 'defaultVal', not 'value'.
self.assertEqual(
- get_tools_config('section', 'key', 'defaultVal'),
- 'defaultVal')
+ get_tools_config("section", "key", "defaultVal"),
+ "defaultVal",
+ )
def test_get_tools_config_normal(self):
"""
This test is designed to verify the value could be parsed from
key = value of the given [section]
"""
- with mock.patch.object(subp, 'which', return_value='/dummy/path'):
+ with mock.patch.object(subp, "which", return_value="/dummy/path"):
# value is not blank
- with mock.patch.object(subp, 'subp',
- return_value=('key = value ', b'')):
+ with mock.patch.object(
+ subp, "subp", return_value=("key = value ", b"")
+ ):
self.assertEqual(
- get_tools_config('section', 'key', 'defaultVal'),
- 'value')
+ get_tools_config("section", "key", "defaultVal"), "value"
+ )
# value is blank
- with mock.patch.object(subp, 'subp',
- return_value=('key = ', b'')):
+ with mock.patch.object(subp, "subp", return_value=("key = ", b"")):
self.assertEqual(
- get_tools_config('section', 'key', 'defaultVal'),
- '')
+ get_tools_config("section", "key", "defaultVal"), ""
+ )
# value contains =
- with mock.patch.object(subp, 'subp',
- return_value=('key=Bar=Wark', b'')):
+ with mock.patch.object(
+ subp, "subp", return_value=("key=Bar=Wark", b"")
+ ):
self.assertEqual(
- get_tools_config('section', 'key', 'defaultVal'),
- 'Bar=Wark')
+ get_tools_config("section", "key", "defaultVal"),
+ "Bar=Wark",
+ )
# value contains specific characters
- with mock.patch.object(subp, 'subp',
- return_value=('[a] b.c_d=e-f', b'')):
+ with mock.patch.object(
+ subp, "subp", return_value=("[a] b.c_d=e-f", b"")
+ ):
self.assertEqual(
- get_tools_config('section', 'key', 'defaultVal'),
- 'e-f')
+ get_tools_config("section", "key", "defaultVal"), "e-f"
+ )
def test_set_gc_status(self):
"""
This test is designed to verify the behavior of set_gc_status
"""
# config is None, return None
- self.assertEqual(set_gc_status(None, 'Successful'), None)
+ self.assertEqual(set_gc_status(None, "Successful"), None)
# post gc status is NO, return None
cf = ConfigFile("tests/data/vmware/cust-dhcp-2nic.cfg")
conf = Config(cf)
- self.assertEqual(set_gc_status(conf, 'Successful'), None)
+ self.assertEqual(set_gc_status(conf, "Successful"), None)
# post gc status is YES, subp is called to execute command
cf._insertKey("MISC|POST-GC-STATUS", "YES")
conf = Config(cf)
- with mock.patch.object(subp, 'subp',
- return_value=('ok', b'')) as mockobj:
- self.assertEqual(
- set_gc_status(conf, 'Successful'), ('ok', b''))
+ with mock.patch.object(
+ subp, "subp", return_value=("ok", b"")
+ ) as mockobj:
+ self.assertEqual(set_gc_status(conf, "Successful"), ("ok", b""))
mockobj.assert_called_once_with(
- ['vmware-rpctool', 'info-set guestinfo.gc.status Successful'],
- rcs=[0])
+ ["vmware-rpctool", "info-set guestinfo.gc.status Successful"],
+ rcs=[0],
+ )
+
# vi: ts=4 expandtab