refactor(api): fix pylint errors

This commit is contained in:
jo 2022-04-01 17:29:11 +02:00 committed by Kyle Robbertze
parent bf8db3e6f0
commit 0bbd46c33f
14 changed files with 159 additions and 169 deletions

View file

@ -1,3 +1,6 @@
from typing import List, Type
from django.db.models import Model
from django.test.runner import DiscoverRunner
@ -8,16 +11,21 @@ class ManagedModelTestRunner(DiscoverRunner):
to execute the SQL manually to create them.
"""
unmanaged_models: List[Type[Model]] = []
def setup_test_environment(self, *args, **kwargs):
from django.apps import apps
self.unmanaged_models = [m for m in apps.get_models() if not m._meta.managed]
for m in self.unmanaged_models:
m._meta.managed = True
for model in apps.get_models():
if not model._meta.managed:
model._meta.managed = True
self.unmanaged_models.append(model)
super().setup_test_environment(*args, **kwargs)
def teardown_test_environment(self, *args, **kwargs):
super().teardown_test_environment(*args, **kwargs)
# reset unmanaged models
for m in self.unmanaged_models:
m._meta.managed = False
for model in self.unmanaged_models:
model._meta.managed = False