check for attribute 'model_class' for compatibility ydaniv#29

lukeaus · Oct 4, 2017 · 6ace69b · 6ace69b
1 parent b79f507
commit 6ace69b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/rest_assured/testcases.py b/rest_assured/testcases.py
@@ -263,7 +263,7 @@ def get_lookup_from_response(self, data):
         return data.get(self.response_lookup_field)
 
     def get_model(self):
-        return self.model_class or self.object.__class__
+        return self.model_class if hasattr(self, 'model_class') else self.object.__class__
 
     def test_create(self, data=None, **kwargs):
         """Send request to the create view endpoint, verify and return the response.