From 8a28e983df091d94eaba77cb82fbe3ef60a80799 Mon Sep 17 00:00:00 2001 From: Simon Charette Date: Tue, 7 Nov 2023 05:32:34 -0500 Subject: Fixed #34946 -- Preserved db_default on combined default field addition. Regression in 7414704e88d73dafbcfbb85f9bc54cb6111439d3. --- tests/migrations/test_operations.py | 13 ++++++++++--- 1 file changed, 10 insertions(+), 3 deletions(-) (limited to 'tests/migrations') diff --git a/tests/migrations/test_operations.py b/tests/migrations/test_operations.py index d58da8b7ac..57a9086c19 100644 --- a/tests/migrations/test_operations.py +++ b/tests/migrations/test_operations.py @@ -1692,15 +1692,22 @@ class OperationTests(OperationTestBase): field = new_state.models[app_label, "pony"].fields["height"] self.assertEqual(field.default, 3) self.assertEqual(field.db_default, Value(4)) - project_state.apps.get_model(app_label, "pony").objects.create(weight=4) + pre_pony_pk = ( + project_state.apps.get_model(app_label, "pony").objects.create(weight=4).pk + ) self.assertColumnNotExists(table_name, "height") # Add field. with connection.schema_editor() as editor: operation.database_forwards(app_label, editor, project_state, new_state) self.assertColumnExists(table_name, "height") + post_pony_pk = ( + project_state.apps.get_model(app_label, "pony").objects.create(weight=10).pk + ) new_model = new_state.apps.get_model(app_label, "pony") - old_pony = new_model.objects.get() - self.assertEqual(old_pony.height, 4) + pre_pony = new_model.objects.get(pk=pre_pony_pk) + self.assertEqual(pre_pony.height, 4) + post_pony = new_model.objects.get(pk=post_pony_pk) + self.assertEqual(post_pony.height, 4) new_pony = new_model.objects.create(weight=5) if not connection.features.can_return_columns_from_insert: new_pony.refresh_from_db() -- cgit v1.3