[fix] Load Organization model with swappable in tests

Author: nemesifier

openwisp_network_topology/tests/test_utils.py

@@ -10,11 +10,10 @@
 from django.test import TestCase
 from django.utils.timezone import now
 
-from openwisp_users.models import Organization
-
 from .. import settings as app_settings
 from .utils import CreateGraphObjectsMixin, CreateOrgMixin, LoadMixin
 
+Organization = swapper.load_model('openwisp_users', 'Organization')
 Link = swapper.load_model('topology', 'Link')
 Node = swapper.load_model('topology', 'Node')
 Snapshot = swapper.load_model('topology', 'Snapshot')

openwisp_network_topology/tests/utils.py

@@ -1,9 +1,12 @@
 import logging
 import os
 
-from openwisp_users.models import Organization
+import swapper
+
 from openwisp_utils.tests import TimeLoggingTestRunner
 
+Organization = swapper.load_model('openwisp_users', 'Organization')
+
 
 class CreateOrgMixin(object):
     def _create_org(self, **kwargs):