Merge pull request #97 from openimis/develop

fix test params
openimis · Sep 19, 2024 · b0fae47 · b0fae47
2 parents 4a01f66 + 0a89a91
commit b0fae47
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/contract/tests/services/services_tests.py b/contract/tests/services/services_tests.py
@@ -343,11 +343,11 @@ def test_get_rule_details(self):
         class_name2 = "ContributionPlan"
         result = get_rule_details(class_name=class_name)
         result2 = get_rule_details(class_name=class_name2)
-        result_param = [param["name"] for param in result["parameters"]]
-        result2_param = [param["name"] for param in result2["parameters"]]
+        result_param = [param['name'] for param in result[class_name]]
+        result2_param = [param['name'] for param in result2[class_name2]]
         self.assertEqual(
             (class_name, class_name2, ["income"], ["rate", "includeFamily"]),
-            (result["class"], result2["class"], result_param, result2_param)
+            (list(result.keys())[0], list(result2.keys())[0], result_param, result2_param)
         )
 
     def test_get_rule_details_not_existing(self):