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