mirrors/django
1
0
Fork 0
mirror of https://github.com/django/django.git synced 2025-03-31 11:37:06 +00:00
Code Issues 32 Releases Wiki Activity

Fixed #25251 -- Restored serialized test data at the end of the test suite

Browse Source 
This allows subsequent test calls with --keepdb to rely on test data,
when it was preserved during database setup.
This commit is contained in:
Markus Holtermann 2021-03-18 21:07:17 +01:00 committed by David Sanders
parent 6f7c0a4d66
commit eb958c52f6
1 changed files with 23 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
django/test/utils.py
View File

@ -368,6 +368,29 @@ def teardown_databases(old_config, verbosity, parallel=0, keepdb=False):
)
connection.creation.destroy_test_db(old_name, verbosity, keepdb)
# When we keep the database and have serialized content, we need to roll back the data
# at the end of the test suite. When using a parallelized test, the data must be restored
# per clone.
if keepdb and hasattr(connection, "_test_serialized_contents"):
serialized_data = connection._test_serialized_contents
old_settings = dict(connection.settings_dict)
test_database_name = connection.creation._get_test_db_name()
if parallel > 1:
for index in range(parallel):
connection.settings_dict["NAME"] = test_database_name
settings_dict = connection.creation.get_test_db_clone_settings(str(index + 1))
connection.settings_dict.update(settings_dict)
connection.close()
connection.creation.deserialize_db_from_string(serialized_data)
else:
connection.settings_dict["NAME"] = test_database_name
connection.close()
connection.creation.deserialize_db_from_string(serialized_data)
# Reset the connection settings
connection.settings_dict.update(old_settings)
connection.close()
def get_runner(settings, test_runner_class=None):
test_runner_class = test_runner_class or settings.TEST_RUNNER