fix(tests): assert global to_dict rather than model-level one

Author: shortcuts

templates/python/tests/e2e/e2e.mustache

@@ -40,7 +40,7 @@ class Test{{#lambda.pascalcase}}{{{client}}}{{/lambda.pascalcase}}{{#isSyncClien
         {{#body}}
         resp = {{^isSyncClient}}await {{/isSyncClient}}{{#lambda.pascalcase}}{{{client}}}{{#isSyncClient}}Sync{{/isSyncClient}}{{/lambda.pascalcase}}(self._e2e_app_id, self._e2e_api_key{{#hasRegionalHost}}, "{{{defaultRegion}}}"{{/hasRegionalHost}}).{{#lambda.snakecase}}{{method}}{{/lambda.snakecase}}({{#parametersWithDataType}}{{> tests/generateParams}}{{/parametersWithDataType}}{{#hasRequestOptions}} request_options={ {{#requestOptions.headers.parameters}}"headers":loads("""{{{.}}}"""),{{/requestOptions.headers.parameters}}{{#requestOptions.queryParameters.parameters}}"query_parameters":loads("""{{{.}}}"""),{{/requestOptions.queryParameters.parameters}} }{{/hasRequestOptions}})
         _expected_body = loads("""{{{.}}}""")
-        assert self._helpers.union(_expected_body, resp) == _expected_body
+        assert self._helpers.union(_expected_body, self._helpers.unwrap(resp)) == _expected_body
         {{/body}}
         {{/response}}
 

tests/output/python/tests/helpers.py

@@ -1,4 +1,13 @@
 class Helpers:
+    def unwrap(self, resp):
+        """ converts the object response to its dictionary form, if it's a list, every items are iterated on and we call to_dict on it, otherwise we just call to_dict """
+        if isinstance(resp, list):
+            _res = []
+            for _, r in enumerate(resp):
+                _res.append(r.to_dict())
+            return _res
+        return resp.to_dict()
+
     def union(self, expected, received):
         """
         creates a new result based on the keys of 'expected' that are also in 'received'
@@ -9,8 +18,6 @@ def union(self, expected, received):
                 _res.append(self.union(_v, received[_iv]))
             return _res
         if isinstance(expected, dict):
-            if hasattr(received, "to_dict"):
-                received = received.to_dict()
             _res = {}
             for k, v in expected.items():
                 if k not in received or received[k] is None: