Fix tests

Author: miaow2

netbox_config_diff/migrations/0002_add_script.py

@@ -1,18 +1,23 @@
+import logging
 import os
 
 from core.choices import ManagedFileRootPathChoices
 from django.conf import settings
 from django.db import migrations
 from pathlib import Path
 
+logger = logging.getLogger(__name__)
 SCRIPT_NAME = "config_diff.py"
 
 
 def forward_func(apps, schema_editor):
     script_path = Path(__file__).parent.parent.joinpath("scripts", SCRIPT_NAME)
     new_path = os.path.join(settings.SCRIPTS_ROOT, SCRIPT_NAME)
-    with open(script_path, "r") as script_file, open(new_path, "w") as new_file:
-        new_file.write(script_file.read())
+    try:
+        with open(script_path, "r") as script_file, open(new_path, "w") as new_file:
+            new_file.write(script_file.read())
+    except FileNotFoundError:
+        logger.warning(f"Failed to open: {script_path} or {new_path}")
 
     ManagedFile = apps.get_model("core", "ManagedFile")
     db_alias = schema_editor.connection.alias

netbox_config_diff/migrations/0004_update_script.py

@@ -1,17 +1,23 @@
+import logging
 import os
 
 from django.conf import settings
 from django.db import migrations
 from pathlib import Path
 
+logger = logging.getLogger(__name__)
+
 SCRIPT_NAME = "config_diff.py"
 
 
 def forward_func(apps, schema_editor):
     script_path = Path(__file__).parent.parent.joinpath("scripts", SCRIPT_NAME)
     new_path = os.path.join(settings.SCRIPTS_ROOT, SCRIPT_NAME)
-    with open(script_path, "r") as script_file, open(new_path, "w") as new_file:
-        new_file.write(script_file.read())
+    try:
+        with open(script_path, "r") as script_file, open(new_path, "w") as new_file:
+            new_file.write(script_file.read())
+    except FileNotFoundError:
+        logger.warning(f"Failed to open: {script_path} or {new_path}")
 
 
 def reverse_func(apps, schema_editor):

tests/conftest.py

@@ -2,6 +2,8 @@
 from unittest.mock import Mock
 
 import pytest
+from django.db import connection
+from django.test.utils import setup_databases
 from faker import Faker
 from typing_extensions import Unpack
 
@@ -10,6 +12,27 @@
 from tests.factories import DataSourceFactory
 
 
+@pytest.fixture(scope="session")
+def django_db_setup(request, django_test_environment, django_db_blocker):
+    with django_db_blocker.unblock():
+        with connection.cursor() as cursor:
+            cursor.execute(
+                """
+                DO $$
+                BEGIN
+                    CREATE COLLATION IF NOT EXISTS natural_sort (
+                        provider = icu,
+                        locale = 'und-u-kn-true',
+                        deterministic = true
+                    );
+                END $$;
+            """
+            )
+        setup_databases(verbosity=request.config.option.verbose, interactive=False, serialized_aliases=[])
+
+    yield
+
+
 class ScriptData(TypedDict, total=False):
     site: str | None
     devices: list[str] | None