Sort out some tests

Signed-off-by: mulhern <[email protected]>
stratis-storage · Dec 4, 2023 · 044e083 · 044e083
1 parent 89eeb31
commit 044e083
Showing 1 changed file with 11 additions and 9 deletions.
diff --git a/tests/whitebox/integration/test_stratis.py b/tests/whitebox/integration/test_stratis.py
@@ -23,6 +23,7 @@
 
 # isort: LOCAL
 from stratis_cli import StratisCliErrorCodes, run
+from stratis_cli._actions import _dynamic
 from stratis_cli._errors import StratisCliStratisdVersionError
 
 from ._misc import RUNNER, TEST_RUNNER, RunTestCase, SimTestCase
@@ -76,15 +77,14 @@ def test_outdated_stratisd_version(self):
         Verify that an outdated version of stratisd will produce a
         StratisCliStratisdVersionError.
         """
-        # pylint: disable=import-outside-toplevel
-        # isort: LOCAL
-        from stratis_cli._actions import _data
+        _dynamic.make_dyn_class("Manager0")
 
         command_line = ["--propagate", "daemon", "version"]
 
-        # pylint: disable=protected-access
         with patch.object(
-            _data.Manager0.Properties.Version,
+            _dynamic._LOOKUP["Manager0"][  # pylint: disable=protected-access
+                2
+            ].Properties.Version,
             "Get",
             return_value="1.0.0",
         ):
@@ -107,12 +107,14 @@ def test_catch_keyboard_exception(self):
         at the calling method generated by dbus-python-client-gen.
         """
 
-        # pylint: disable=import-outside-toplevel
-        # isort: LOCAL
-        from stratis_cli._actions import _data
+        _dynamic.make_dyn_class("Manager0")
 
         with patch.object(
-            _data.Manager0.Properties.Version, "Get", side_effect=KeyboardInterrupt()
+            _dynamic._LOOKUP["Manager0"][  # pylint: disable=protected-access
+                2
+            ].Properties.Version,
+            "Get",
+            side_effect=KeyboardInterrupt(),
         ):
             with self.assertRaises(KeyboardInterrupt):
                 run()(["daemon", "version"])