diff options
Diffstat (limited to 'tests')
| -rw-r--r-- | tests/migrations/test_state.py | 24 |
1 files changed, 24 insertions, 0 deletions
diff --git a/tests/migrations/test_state.py b/tests/migrations/test_state.py index ef198eb0b6..44d4cdf268 100644 --- a/tests/migrations/test_state.py +++ b/tests/migrations/test_state.py @@ -532,6 +532,30 @@ class StateTests(SimpleTestCase): project_state.add_model(ModelState.from_model(B)) self.assertEqual(len(project_state.apps.get_models()), 2) + def test_reload_model_relationship_consistency(self): + project_state = ProjectState() + project_state.add_model(ModelState('migrations', 'A', [])) + project_state.add_model(ModelState('migrations', 'B', [ + ('a', models.ForeignKey('A', models.CASCADE)), + ])) + project_state.add_model(ModelState('migrations', 'C', [ + ('b', models.ForeignKey('B', models.CASCADE)), + ])) + A = project_state.apps.get_model('migrations.A') + B = project_state.apps.get_model('migrations.B') + C = project_state.apps.get_model('migrations.C') + self.assertEqual([r.related_model for r in A._meta.related_objects], [B]) + self.assertEqual([r.related_model for r in B._meta.related_objects], [C]) + self.assertEqual([r.related_model for r in C._meta.related_objects], []) + + project_state.reload_model('migrations', 'a', delay=True) + A = project_state.apps.get_model('migrations.A') + B = project_state.apps.get_model('migrations.B') + C = project_state.apps.get_model('migrations.C') + self.assertEqual([r.related_model for r in A._meta.related_objects], [B]) + self.assertEqual([r.related_model for r in B._meta.related_objects], [C]) + self.assertEqual([r.related_model for r in C._meta.related_objects], []) + def test_add_relations(self): """ #24573 - Adding relations to existing models should reload the |