CTOOLS-373: Testing key

finbourne · Oct 1, 2024 · 225089e · 225089e
1 parent a51369d
commit 225089e
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/generate/templates/api_test.mustache b/generate/templates/api_test.mustache
@@ -99,6 +99,16 @@ class {{#operations}}Test{{classname}}(unittest.IsolatedAsyncioTestCase):
             # {{complexType}}
             if "example" in jsonContent:
                 {{paramName}}: {{dataType}} = jsonContent["example"]
+
+                if isinstance(jsonContent, object):
+                    is_key = list(jsonContent.keys())[0] if len(jsonContent.keys()) > 0 else None
+
+                    if is_key is not None:
+                        if list(jsonContent.keys())[0] == 'data-access-page-evaluation':
+                            request_body: Dict[str, EvaluationRequest] = jsonContent["example"]["data-access-page-evaluation"]
+                    else:
+                        request_body: Dict[str, EvaluationRequest] = jsonContent["example"]
+
             elif "examples" in jsonContent:
                 examples = jsonContent["examples"]
                 examples = [examples[key] for key in examples]