diff --git a/app/centurion/tests/unit_models.py b/app/centurion/tests/unit_models.py index 6d5c2e5a..1f86812a 100644 --- a/app/centurion/tests/unit_models.py +++ b/app/centurion/tests/unit_models.py @@ -116,8 +116,10 @@ class ModelTestCases( create_instance = True, ) + instance = instance['instance'] - yield instance['instance'] + + yield instance if 'mockmodel' in apps.all_models['core']: diff --git a/app/tests/fixtures/model_instance.py b/app/tests/fixtures/model_instance.py index 9a8a510c..8d68e611 100644 --- a/app/tests/fixtures/model_instance.py +++ b/app/tests/fixtures/model_instance.py @@ -58,6 +58,8 @@ def model_instance(django_db_blocker, model_kwarg_data, model, model_kwargs): create_instance = True, ) + obj = obj['instance'] + model_objs += [ obj ]