Avoid blacklisting app migrations failing in projects with multiple d…

…atabases (#429)

* add a second database for tests as an indicator that migrations for the blacklisting app will run with no issues in projects with multiple databases

* set the db_alias when migrating the jti claim to avoid migrations failing in projects with multiple databases

rest_framework_simplejwt/token_blacklist/migrations/0003_auto_20171017_2007.py

@@ -6,15 +6,17 @@
 def populate_jti_hex(apps, schema_editor):
     OutstandingToken = apps.get_model('token_blacklist', 'OutstandingToken')
 
-    for token in OutstandingToken.objects.all():
+    db_alias = schema_editor.connection.alias
+    for token in OutstandingToken.objects.using(db_alias).all():
         token.jti_hex = token.jti.hex
         token.save()
 
 
 def reverse_populate_jti_hex(apps, schema_editor):  # pragma: no cover
     OutstandingToken = apps.get_model('token_blacklist', 'OutstandingToken')
 
-    for token in OutstandingToken.objects.all():
+    db_alias = schema_editor.connection.alias
+    for token in OutstandingToken.objects.using(db_alias).all():
         token.jti = UUID(hex=token.jti_hex)
         token.save()
 

tests/conftest.py

@@ -14,6 +14,10 @@ def pytest_configure():
             'default': {
                 'ENGINE': 'django.db.backends.sqlite3',
                 'NAME': ':memory:'
+            },
+            'other': {
+                'ENGINE': 'django.db.backends.sqlite3',
+                'NAME': 'other'
             }
         },
         SITE_ID=1,