Allow running tests with --nomigrations

Author: AlanCoding

ansible_base/resource_registry/models/service_identifier.py

@@ -1,5 +1,7 @@
+import sys
 import uuid
 
+from django.conf import settings
 from django.db import models
 
 
@@ -23,5 +25,11 @@ def save(self, *args, **kwargs):
 def service_id():
     global _service_id
     if not _service_id:
-        _service_id = str(ServiceID.objects.first().pk)
+        obj = ServiceID.objects.first()
+        if obj is None:
+            if settings.DEBUG or any("pytest" in arg for arg in sys.argv):
+                obj = ServiceID.objects.create()
+            else:
+                raise RuntimeError('Expected ServiceID to be created in data migrations but was not found')
+        _service_id = str(obj.pk)
     return _service_id

test_app/tests/conftest.py

@@ -13,6 +13,7 @@
 from django.contrib.contenttypes.models import ContentType
 from django.db.migrations.recorder import MigrationRecorder
 from django.db.models.signals import post_migrate
+from django.db.utils import ProgrammingError
 from django.test.client import RequestFactory
 from django.test.utils import override_settings
 from drf_spectacular.generators import SchemaGenerator
@@ -47,7 +48,13 @@ def test_migrations_okay(apps=None, app_config=None, **kwargs):
         return  # so that it is only ran once
     disk_steps = defaultdict(set)
     app_exceptions = {'default': 'auth', 'social_auth': 'social_django'}
-    for app in MigrationRecorder.Migration.objects.values_list('app', flat=True).distinct():
+
+    try:
+        app_list = list(MigrationRecorder.Migration.objects.values_list('app', flat=True).distinct())
+    except ProgrammingError:
+        return
+
+    for app in app_list:
         if app in app_exceptions:
             continue
         try: