Review: a few fixes

[pyi2ncommon] / test / cnfvar / test_store.py

diff --git a/test/cnfvar/test_store.py b/test/cnfvar/test_store.py
index e234c4d..b5612d3 100644 (file)
--- a/test/cnfvar/test_store.py
+++ b/test/cnfvar/test_store.py
@@ -57,7 +57,8 @@ class TestStore(unittest.TestCase):
              patch.object(store, "_wait_for_generate"):
             self._query_and_update(store)
 
-        cnf_api_input = set_cnf_mock.call_args.kwargs["vars"]
+        args, kwargs = set_cnf_mock.call_args
+        cnf_api_input = kwargs["vars"]
         cnf_input = CnfList.from_api_structure(cnf_api_input)
 
         expected_input = dedent("""\
@@ -100,7 +101,8 @@ class TestStore(unittest.TestCase):
              patch.object(store, "_wait_for_generate"):
             store.delete(cnfvars)
 
-        cnf_api_input = set_cnf_mock.call_args.kwargs["vars"]
+        args, kwargs = set_cnf_mock.call_args
+        cnf_api_input = kwargs["vars"]
         cnf_input = CnfList.from_api_structure(cnf_api_input)
 
         expected_input = dedent("""\