update tests to assert deterministic db name suffix logic

Author: SalmonMode

tests/test_db_setup.py

@@ -159,32 +159,32 @@ def test_xdist_with_reuse(django_testdir):
 
         from .app.models import Item
 
-        def _check(settings):
+        def _check(settings, worker_id):
             # Make sure that the database name looks correct
             db_name = settings.DATABASES['default']['NAME']
-            assert db_name.endswith('_gw0') or db_name.endswith('_gw1')
+            assert db_name == 'test_custom_db_name_{}'.format(worker_id)
 
             assert Item.objects.count() == 0
             Item.objects.create(name='foo')
             assert Item.objects.count() == 1
 
 
         @pytest.mark.django_db
-        def test_a(settings):
-            _check(settings)
+        def test_a(settings, worker_id):
+            _check(settings, worker_id)
 
 
         @pytest.mark.django_db
-        def test_b(settings):
-            _check(settings)
+        def test_b(settings, worker_id):
+            _check(settings, worker_id)
 
         @pytest.mark.django_db
-        def test_c(settings):
-            _check(settings)
+        def test_c(settings, worker_id):
+            _check(settings, worker_id)
 
         @pytest.mark.django_db
-        def test_d(settings):
-            _check(settings)
+        def test_d(settings, worker_id):
+            _check(settings, worker_id)
     """
     )
 
@@ -270,7 +270,7 @@ def test_sqlite_database_renamed(self, django_testdir):
             from django.db import connections
 
             @pytest.mark.django_db
-            def test_a():
+            def test_a(worker_id):
                 (conn_db2, conn_default) = sorted(
                     connections.all(),
                     key=lambda conn: conn.alias,
@@ -288,7 +288,7 @@ def test_a():
 
                 assert conn_db2.vendor == 'sqlite'
                 db_name = conn_db2.creation._get_test_db_name()
-                assert db_name.startswith('test_custom_db_name_gw')
+                assert db_name == 'test_custom_db_name_{}'.format(worker_id)
         """
         )
 
@@ -377,13 +377,15 @@ def test_db_with_tox_suffix(self, django_testdir, monkeypatch):
             from django.db import connections
 
             @pytest.mark.django_db
-            def test_inner():
+            def test_inner(worker_id):
 
                 (conn, ) = connections.all()
 
                 assert conn.vendor == 'sqlite'
                 db_name = conn.creation._get_test_db_name()
-                assert db_name.startswith('test_custom_db_name_py37-django22_gw')
+                assert db_name == 'test_custom_db_name_py37-django22_{}'.format(
+                    worker_id,
+                )
         """
         )