Loading tests/migrations/test_operations.py +42 −40 Original line number Diff line number Diff line Loading @@ -226,46 +226,48 @@ class OperationTests(MigrationTestBase): self.assertTableExists("test_dlmo_pony") self.assertTableNotExists("test_rnmo_horse") def test_rename_model_with_related(self): """ Tests the real-world combo of a RenameModel operation with AlterField for a related field. """ project_state = self.set_up_test_model( "test_rnmowr", related_model=True) # Test the state alterations model_operation = migrations.RenameModel("Pony", "Horse") new_state = project_state.clone() model_operation.state_forwards("test_rnmowr", new_state) self.assertNotIn(("test_rnmowr", "pony"), new_state.models) self.assertIn(("test_rnmowr", "horse"), new_state.models) self.assertEqual( "Pony", project_state.render().get_model("test_rnmowr", "rider") ._meta.get_field_by_name("pony")[0].rel.to._meta.object_name) field_operation = migrations.AlterField( "Rider", "pony", models.ForeignKey("Horse")) field_operation.state_forwards("test_rnmowr", new_state) self.assertEqual( "Horse", new_state.render().get_model("test_rnmowr", "rider") ._meta.get_field_by_name("pony")[0].rel.to._meta.object_name) # Test the database alterations self.assertTableExists("test_rnmowr_pony") self.assertTableNotExists("test_rnmowr_horse") with connection.schema_editor() as editor: model_operation.database_forwards("test_rnmowr", editor, project_state, new_state) field_operation.database_forwards("test_rnmowr", editor, project_state, new_state) self.assertTableNotExists("test_rnmowr_pony") self.assertTableExists("test_rnmowr_horse") # And test reversal with connection.schema_editor() as editor: field_operation.database_backwards("test_rnmowr", editor, new_state, project_state) model_operation.database_backwards("test_rnmowr", editor, new_state, project_state) self.assertTableExists("test_rnmowr_pony") self.assertTableNotExists("test_rnmowr_horse") # See #22248 - this will fail until that's fixed. # # def test_rename_model_with_related(self): # """ # Tests the real-world combo of a RenameModel operation with AlterField # for a related field. # """ # project_state = self.set_up_test_model( # "test_rnmowr", related_model=True) # # Test the state alterations # model_operation = migrations.RenameModel("Pony", "Horse") # new_state = project_state.clone() # model_operation.state_forwards("test_rnmowr", new_state) # self.assertNotIn(("test_rnmowr", "pony"), new_state.models) # self.assertIn(("test_rnmowr", "horse"), new_state.models) # self.assertEqual( # "Pony", # project_state.render().get_model("test_rnmowr", "rider") # ._meta.get_field_by_name("pony")[0].rel.to._meta.object_name) # field_operation = migrations.AlterField( # "Rider", "pony", models.ForeignKey("Horse")) # field_operation.state_forwards("test_rnmowr", new_state) # self.assertEqual( # "Horse", # new_state.render().get_model("test_rnmowr", "rider") # ._meta.get_field_by_name("pony")[0].rel.to._meta.object_name) # # Test the database alterations # self.assertTableExists("test_rnmowr_pony") # self.assertTableNotExists("test_rnmowr_horse") # with connection.schema_editor() as editor: # model_operation.database_forwards("test_rnmowr", editor, project_state, new_state) # field_operation.database_forwards("test_rnmowr", editor, project_state, new_state) # self.assertTableNotExists("test_rnmowr_pony") # self.assertTableExists("test_rnmowr_horse") # # And test reversal # with connection.schema_editor() as editor: # field_operation.database_backwards("test_rnmowr", editor, new_state, project_state) # model_operation.database_backwards("test_rnmowr", editor, new_state, project_state) # self.assertTableExists("test_rnmowr_pony") # self.assertTableNotExists("test_rnmowr_horse") def test_add_field(self): """ Loading Loading
tests/migrations/test_operations.py +42 −40 Original line number Diff line number Diff line Loading @@ -226,46 +226,48 @@ class OperationTests(MigrationTestBase): self.assertTableExists("test_dlmo_pony") self.assertTableNotExists("test_rnmo_horse") def test_rename_model_with_related(self): """ Tests the real-world combo of a RenameModel operation with AlterField for a related field. """ project_state = self.set_up_test_model( "test_rnmowr", related_model=True) # Test the state alterations model_operation = migrations.RenameModel("Pony", "Horse") new_state = project_state.clone() model_operation.state_forwards("test_rnmowr", new_state) self.assertNotIn(("test_rnmowr", "pony"), new_state.models) self.assertIn(("test_rnmowr", "horse"), new_state.models) self.assertEqual( "Pony", project_state.render().get_model("test_rnmowr", "rider") ._meta.get_field_by_name("pony")[0].rel.to._meta.object_name) field_operation = migrations.AlterField( "Rider", "pony", models.ForeignKey("Horse")) field_operation.state_forwards("test_rnmowr", new_state) self.assertEqual( "Horse", new_state.render().get_model("test_rnmowr", "rider") ._meta.get_field_by_name("pony")[0].rel.to._meta.object_name) # Test the database alterations self.assertTableExists("test_rnmowr_pony") self.assertTableNotExists("test_rnmowr_horse") with connection.schema_editor() as editor: model_operation.database_forwards("test_rnmowr", editor, project_state, new_state) field_operation.database_forwards("test_rnmowr", editor, project_state, new_state) self.assertTableNotExists("test_rnmowr_pony") self.assertTableExists("test_rnmowr_horse") # And test reversal with connection.schema_editor() as editor: field_operation.database_backwards("test_rnmowr", editor, new_state, project_state) model_operation.database_backwards("test_rnmowr", editor, new_state, project_state) self.assertTableExists("test_rnmowr_pony") self.assertTableNotExists("test_rnmowr_horse") # See #22248 - this will fail until that's fixed. # # def test_rename_model_with_related(self): # """ # Tests the real-world combo of a RenameModel operation with AlterField # for a related field. # """ # project_state = self.set_up_test_model( # "test_rnmowr", related_model=True) # # Test the state alterations # model_operation = migrations.RenameModel("Pony", "Horse") # new_state = project_state.clone() # model_operation.state_forwards("test_rnmowr", new_state) # self.assertNotIn(("test_rnmowr", "pony"), new_state.models) # self.assertIn(("test_rnmowr", "horse"), new_state.models) # self.assertEqual( # "Pony", # project_state.render().get_model("test_rnmowr", "rider") # ._meta.get_field_by_name("pony")[0].rel.to._meta.object_name) # field_operation = migrations.AlterField( # "Rider", "pony", models.ForeignKey("Horse")) # field_operation.state_forwards("test_rnmowr", new_state) # self.assertEqual( # "Horse", # new_state.render().get_model("test_rnmowr", "rider") # ._meta.get_field_by_name("pony")[0].rel.to._meta.object_name) # # Test the database alterations # self.assertTableExists("test_rnmowr_pony") # self.assertTableNotExists("test_rnmowr_horse") # with connection.schema_editor() as editor: # model_operation.database_forwards("test_rnmowr", editor, project_state, new_state) # field_operation.database_forwards("test_rnmowr", editor, project_state, new_state) # self.assertTableNotExists("test_rnmowr_pony") # self.assertTableExists("test_rnmowr_horse") # # And test reversal # with connection.schema_editor() as editor: # field_operation.database_backwards("test_rnmowr", editor, new_state, project_state) # model_operation.database_backwards("test_rnmowr", editor, new_state, project_state) # self.assertTableExists("test_rnmowr_pony") # self.assertTableNotExists("test_rnmowr_horse") def test_add_field(self): """ Loading
