|
33 | 33 |
|
34 | 34 |
|
35 | 35 | @pytest.fixture(scope="session")
|
36 |
| -def django_db_modify_db_settings_xdist_suffix(request): |
| 36 | +def django_db_modify_db_settings_tox_suffix(request): |
37 | 37 | skip_if_no_django()
|
38 | 38 |
|
39 |
| - from django.conf import settings |
| 39 | + is_tox_parallel = os.environ.get("TOX_PARALLEL_ENV") |
| 40 | + tox_environment = os.environ.get("TOXENV") |
| 41 | + if is_tox_parallel and tox_environment: |
| 42 | + # Put a suffix like _py27-django21 on tox workers |
| 43 | + _set_suffix_to_test_databases(suffix=tox_environment) |
40 | 44 |
|
41 |
| - for db_settings in settings.DATABASES.values(): |
42 |
| - |
43 |
| - try: |
44 |
| - test_name = db_settings["TEST"]["NAME"] |
45 |
| - except KeyError: |
46 |
| - test_name = None |
47 | 45 |
|
48 |
| - if not test_name: |
49 |
| - if db_settings["ENGINE"] == "django.db.backends.sqlite3": |
50 |
| - return ":memory:" |
51 |
| - else: |
52 |
| - test_name = "test_{}".format(db_settings["NAME"]) |
| 46 | +@pytest.fixture(scope="session") |
| 47 | +def django_db_modify_db_settings_xdist_suffix(request): |
| 48 | + skip_if_no_django() |
53 | 49 |
|
| 50 | + xdist_suffix = getattr(request.config, "slaveinput", {}).get("slaveid") |
| 51 | + if xdist_suffix: |
54 | 52 | # Put a suffix like _gw0, _gw1 etc on xdist processes
|
55 |
| - xdist_suffix = getattr(request.config, "slaveinput", {}).get("slaveid") |
56 |
| - if test_name != ":memory:" and xdist_suffix is not None: |
57 |
| - test_name = "{}_{}".format(test_name, xdist_suffix) |
| 53 | + _set_suffix_to_test_databases(suffix=xdist_suffix) |
58 | 54 |
|
59 |
| - db_settings.setdefault("TEST", {}) |
60 |
| - db_settings["TEST"]["NAME"] = test_name |
| 55 | + |
| 56 | +@pytest.fixture(scope="session") |
| 57 | +def django_db_modify_db_settings_parallel_suffix( |
| 58 | + django_db_modify_db_settings_tox_suffix, |
| 59 | + django_db_modify_db_settings_xdist_suffix, |
| 60 | +): |
| 61 | + skip_if_no_django() |
61 | 62 |
|
62 | 63 |
|
63 | 64 | @pytest.fixture(scope="session")
|
64 |
| -def django_db_modify_db_settings(django_db_modify_db_settings_xdist_suffix): |
| 65 | +def django_db_modify_db_settings(django_db_modify_db_settings_parallel_suffix): |
65 | 66 | skip_if_no_django()
|
66 | 67 |
|
67 | 68 |
|
@@ -153,6 +154,31 @@ def _disable_native_migrations():
|
153 | 154 | settings.MIGRATION_MODULES = DisableMigrations()
|
154 | 155 |
|
155 | 156 |
|
| 157 | +def _set_suffix_to_test_databases(suffix): |
| 158 | + if not suffix: |
| 159 | + return |
| 160 | + |
| 161 | + from django.conf import settings |
| 162 | + |
| 163 | + for db_settings in settings.DATABASES.values(): |
| 164 | + |
| 165 | + try: |
| 166 | + test_name = db_settings["TEST"]["NAME"] |
| 167 | + except KeyError: |
| 168 | + test_name = None |
| 169 | + |
| 170 | + if not test_name: |
| 171 | + if db_settings["ENGINE"] == "django.db.backends.sqlite3": |
| 172 | + continue |
| 173 | + test_name = "test_{}".format(db_settings["NAME"]) |
| 174 | + |
| 175 | + if test_name == ":memory:": |
| 176 | + continue |
| 177 | + |
| 178 | + db_settings.setdefault("TEST", {}) |
| 179 | + db_settings["TEST"]["NAME"] = "{}_{}".format(test_name, suffix) |
| 180 | + |
| 181 | + |
156 | 182 | # ############### User visible fixtures ################
|
157 | 183 |
|
158 | 184 |
|
|
0 commit comments