Fix migration 0004 when run against a non default database (updated) (#215)

* Fix migrations when run against a non default database.

* Change test DB to SQLite as per PR review.

Co-authored-by: Florimond Manca <florimond.manca@protonmail.com>

* Format

Co-authored-by: Niall McAndrew <niall@mediasuite.co.nz>

Author: florimondmanca

src/rest_framework_api_key/migrations/0004_prefix_hashed_key.py

@@ -10,7 +10,7 @@
 def populate_prefix_hashed_key(apps, schema_editor) -> None:  # type: ignore
     model = apps.get_model(APP_NAME, MODEL_NAME)
 
-    for api_key in model.objects.all():
+    for api_key in model.objects.using(schema_editor.connection.alias).all():
         prefix, _, hashed_key = api_key.id.partition(".")
         api_key.prefix = prefix
         api_key.hashed_key = hashed_key

test_project/heroes/migrations/0002_prefix_hashed_key.py

@@ -10,7 +10,7 @@
 def populate_prefix_hashed_key(apps, schema_editor):  # type: ignore
     model = apps.get_model(APP_NAME, MODEL_NAME)
 
-    for api_key in model.objects.all():
+    for api_key in model.objects.using(schema_editor.connection.alias).all():
         prefix, _, hashed_key = api_key.id.partition(".")
         api_key.prefix = prefix
         api_key.hashed_key = hashed_key

test_project/project/settings.py

@@ -55,7 +55,12 @@
 
 # Database
 
-DATABASES = {"default": dj_database_url.config(default="sqlite:///db.sqlite3")}
+DATABASES = {
+    "default": dj_database_url.config(default="sqlite:///db.sqlite3"),
+    "test": dj_database_url.config(
+        "TEST_DATABASE_URL", default="sqlite://test.sqlite3"
+    ),
+}
 
 
 # Password validation

tests/conftest.py

@@ -45,7 +45,8 @@ def pytest_configure() -> None:
             ],
             "ROOT_URLCONF": "test_project.project.urls",
             "DATABASES": {
-                "default": dj_database_url.config(default="sqlite://:memory:")
+                "default": dj_database_url.config(default="sqlite://:memory:"),
+                "test": dj_database_url.config(default="sqlite://:memory:"),
             },
         }
     )