12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322 |
- from django.conf import settings
- from django.test import TestCase, override_settings
- from django.db.migrations.autodetector import MigrationAutodetector
- from django.db.migrations.questioner import MigrationQuestioner
- from django.db.migrations.state import ProjectState, ModelState
- from django.db.migrations.graph import MigrationGraph
- from django.db.migrations.loader import MigrationLoader
- from django.db import models, connection
- from django.contrib.auth.models import AbstractBaseUser
- class DeconstructableObject(object):
- """
- A custom deconstructable object.
- """
- def deconstruct(self):
- return self.__module__ + '.' + self.__class__.__name__, [], {}
- class AutodetectorTests(TestCase):
- """
- Tests the migration autodetector.
- """
- author_empty = ModelState("testapp", "Author", [("id", models.AutoField(primary_key=True))])
- author_name = ModelState("testapp", "Author", [("id", models.AutoField(primary_key=True)), ("name", models.CharField(max_length=200))])
- author_name_longer = ModelState("testapp", "Author", [("id", models.AutoField(primary_key=True)), ("name", models.CharField(max_length=400))])
- author_name_renamed = ModelState("testapp", "Author", [("id", models.AutoField(primary_key=True)), ("names", models.CharField(max_length=200))])
- author_name_default = ModelState("testapp", "Author", [("id", models.AutoField(primary_key=True)), ("name", models.CharField(max_length=200, default='Ada Lovelace'))])
- author_name_deconstructable_1 = ModelState("testapp", "Author", [("id", models.AutoField(primary_key=True)), ("name", models.CharField(max_length=200, default=DeconstructableObject()))])
- author_name_deconstructable_2 = ModelState("testapp", "Author", [("id", models.AutoField(primary_key=True)), ("name", models.CharField(max_length=200, default=DeconstructableObject()))])
- author_name_deconstructable_3 = ModelState("testapp", "Author", [("id", models.AutoField(primary_key=True)), ("name", models.CharField(max_length=200, default=models.IntegerField()))])
- author_name_deconstructable_4 = ModelState("testapp", "Author", [("id", models.AutoField(primary_key=True)), ("name", models.CharField(max_length=200, default=models.IntegerField()))])
- author_custom_pk = ModelState("testapp", "Author", [("pk_field", models.IntegerField(primary_key=True))])
- author_with_book = ModelState("testapp", "Author", [("id", models.AutoField(primary_key=True)), ("name", models.CharField(max_length=200)), ("book", models.ForeignKey("otherapp.Book"))])
- author_with_book_order_wrt = ModelState("testapp", "Author", [("id", models.AutoField(primary_key=True)), ("name", models.CharField(max_length=200)), ("book", models.ForeignKey("otherapp.Book"))], options={"order_with_respect_to": "book"})
- author_renamed_with_book = ModelState("testapp", "Writer", [("id", models.AutoField(primary_key=True)), ("name", models.CharField(max_length=200)), ("book", models.ForeignKey("otherapp.Book"))])
- author_with_publisher_string = ModelState("testapp", "Author", [("id", models.AutoField(primary_key=True)), ("name", models.CharField(max_length=200)), ("publisher_name", models.CharField(max_length=200))])
- author_with_publisher = ModelState("testapp", "Author", [("id", models.AutoField(primary_key=True)), ("name", models.CharField(max_length=200)), ("publisher", models.ForeignKey("testapp.Publisher"))])
- author_with_custom_user = ModelState("testapp", "Author", [("id", models.AutoField(primary_key=True)), ("name", models.CharField(max_length=200)), ("user", models.ForeignKey("thirdapp.CustomUser"))])
- author_proxy = ModelState("testapp", "AuthorProxy", [], {"proxy": True}, ("testapp.author", ))
- author_proxy_options = ModelState("testapp", "AuthorProxy", [], {"proxy": True, "verbose_name": "Super Author"}, ("testapp.author", ))
- author_proxy_notproxy = ModelState("testapp", "AuthorProxy", [], {}, ("testapp.author", ))
- author_proxy_third = ModelState("thirdapp", "AuthorProxy", [], {"proxy": True}, ("testapp.author", ))
- author_proxy_proxy = ModelState("testapp", "AAuthorProxyProxy", [], {"proxy": True}, ("testapp.authorproxy", ))
- author_unmanaged = ModelState("testapp", "AuthorUnmanaged", [], {"managed": False}, ("testapp.author", ))
- author_unmanaged_managed = ModelState("testapp", "AuthorUnmanaged", [], {}, ("testapp.author", ))
- author_unmanaged_default_pk = ModelState("testapp", "Author", [("id", models.AutoField(primary_key=True))])
- author_unmanaged_custom_pk = ModelState("testapp", "Author", [
- ("pk_field", models.IntegerField(primary_key=True)),
- ])
- author_with_m2m = ModelState("testapp", "Author", [
- ("id", models.AutoField(primary_key=True)),
- ("publishers", models.ManyToManyField("testapp.Publisher")),
- ])
- author_with_m2m_through = ModelState("testapp", "Author", [("id", models.AutoField(primary_key=True)), ("publishers", models.ManyToManyField("testapp.Publisher", through="testapp.Contract"))])
- author_with_options = ModelState("testapp", "Author", [("id", models.AutoField(primary_key=True))], {"verbose_name": "Authi", "permissions": [('can_hire', 'Can hire')]})
- contract = ModelState("testapp", "Contract", [("id", models.AutoField(primary_key=True)), ("author", models.ForeignKey("testapp.Author")), ("publisher", models.ForeignKey("testapp.Publisher"))])
- publisher = ModelState("testapp", "Publisher", [("id", models.AutoField(primary_key=True)), ("name", models.CharField(max_length=100))])
- publisher_with_author = ModelState("testapp", "Publisher", [("id", models.AutoField(primary_key=True)), ("author", models.ForeignKey("testapp.Author")), ("name", models.CharField(max_length=100))])
- publisher_with_aardvark_author = ModelState("testapp", "Publisher", [("id", models.AutoField(primary_key=True)), ("author", models.ForeignKey("testapp.Aardvark")), ("name", models.CharField(max_length=100))])
- publisher_with_book = ModelState("testapp", "Publisher", [("id", models.AutoField(primary_key=True)), ("author", models.ForeignKey("otherapp.Book")), ("name", models.CharField(max_length=100))])
- other_pony = ModelState("otherapp", "Pony", [("id", models.AutoField(primary_key=True))])
- other_stable = ModelState("otherapp", "Stable", [("id", models.AutoField(primary_key=True))])
- third_thing = ModelState("thirdapp", "Thing", [("id", models.AutoField(primary_key=True))])
- book = ModelState("otherapp", "Book", [("id", models.AutoField(primary_key=True)), ("author", models.ForeignKey("testapp.Author")), ("title", models.CharField(max_length=200))])
- book_proxy_fk = ModelState("otherapp", "Book", [("id", models.AutoField(primary_key=True)), ("author", models.ForeignKey("thirdapp.AuthorProxy")), ("title", models.CharField(max_length=200))])
- book_migrations_fk = ModelState("otherapp", "Book", [("id", models.AutoField(primary_key=True)), ("author", models.ForeignKey("migrations.UnmigratedModel")), ("title", models.CharField(max_length=200))])
- book_with_no_author = ModelState("otherapp", "Book", [("id", models.AutoField(primary_key=True)), ("title", models.CharField(max_length=200))])
- book_with_author_renamed = ModelState("otherapp", "Book", [("id", models.AutoField(primary_key=True)), ("author", models.ForeignKey("testapp.Writer")), ("title", models.CharField(max_length=200))])
- book_with_field_and_author_renamed = ModelState("otherapp", "Book", [("id", models.AutoField(primary_key=True)), ("writer", models.ForeignKey("testapp.Writer")), ("title", models.CharField(max_length=200))])
- book_with_multiple_authors = ModelState("otherapp", "Book", [("id", models.AutoField(primary_key=True)), ("authors", models.ManyToManyField("testapp.Author")), ("title", models.CharField(max_length=200))])
- book_with_multiple_authors_through_attribution = ModelState("otherapp", "Book", [("id", models.AutoField(primary_key=True)), ("authors", models.ManyToManyField("testapp.Author", through="otherapp.Attribution")), ("title", models.CharField(max_length=200))])
- book_unique = ModelState("otherapp", "Book", [("id", models.AutoField(primary_key=True)), ("author", models.ForeignKey("testapp.Author")), ("title", models.CharField(max_length=200))], {"unique_together": {("author", "title")}})
- book_unique_2 = ModelState("otherapp", "Book", [("id", models.AutoField(primary_key=True)), ("author", models.ForeignKey("testapp.Author")), ("title", models.CharField(max_length=200))], {"unique_together": {("title", "author")}})
- book_unique_3 = ModelState("otherapp", "Book", [("id", models.AutoField(primary_key=True)), ("newfield", models.IntegerField()), ("author", models.ForeignKey("testapp.Author")), ("title", models.CharField(max_length=200))], {"unique_together": {("title", "newfield")}})
- attribution = ModelState("otherapp", "Attribution", [("id", models.AutoField(primary_key=True)), ("author", models.ForeignKey("testapp.Author")), ("book", models.ForeignKey("otherapp.Book"))])
- edition = ModelState("thirdapp", "Edition", [("id", models.AutoField(primary_key=True)), ("book", models.ForeignKey("otherapp.Book"))])
- custom_user = ModelState("thirdapp", "CustomUser", [("id", models.AutoField(primary_key=True)), ("username", models.CharField(max_length=255))], bases=(AbstractBaseUser, ))
- custom_user_no_inherit = ModelState("thirdapp", "CustomUser", [("id", models.AutoField(primary_key=True)), ("username", models.CharField(max_length=255))])
- aardvark = ModelState("thirdapp", "Aardvark", [("id", models.AutoField(primary_key=True))])
- aardvark_testapp = ModelState("testapp", "Aardvark", [("id", models.AutoField(primary_key=True))])
- aardvark_based_on_author = ModelState("testapp", "Aardvark", [], bases=("testapp.Author", ))
- aardvark_pk_fk_author = ModelState("testapp", "Aardvark", [("id", models.OneToOneField("testapp.Author", primary_key=True))])
- knight = ModelState("eggs", "Knight", [("id", models.AutoField(primary_key=True))])
- rabbit = ModelState("eggs", "Rabbit", [("id", models.AutoField(primary_key=True)), ("knight", models.ForeignKey("eggs.Knight")), ("parent", models.ForeignKey("eggs.Rabbit"))], {"unique_together": {("parent", "knight")}})
- def repr_changes(self, changes):
- output = ""
- for app_label, migrations in sorted(changes.items()):
- output += " %s:\n" % app_label
- for migration in migrations:
- output += " %s\n" % migration.name
- for operation in migration.operations:
- output += " %s\n" % operation
- return output
- def assertNumberMigrations(self, changes, app_label, number):
- if len(changes.get(app_label, [])) != number:
- self.fail("Incorrect number of migrations (%s) for %s (expected %s)\n%s" % (
- len(changes.get(app_label, [])),
- app_label,
- number,
- self.repr_changes(changes),
- ))
- def assertOperationTypes(self, changes, app_label, index, types):
- if not changes.get(app_label, None):
- self.fail("No migrations found for %s\n%s" % (app_label, self.repr_changes(changes)))
- if len(changes[app_label]) < index + 1:
- self.fail("No migration at index %s for %s\n%s" % (index, app_label, self.repr_changes(changes)))
- migration = changes[app_label][index]
- real_types = [operation.__class__.__name__ for operation in migration.operations]
- if types != real_types:
- self.fail("Operation type mismatch for %s.%s (expected %s):\n%s" % (
- app_label,
- migration.name,
- types,
- self.repr_changes(changes),
- ))
- def assertOperationAttributes(self, changes, app_label, index, operation_index, **attrs):
- if not changes.get(app_label, None):
- self.fail("No migrations found for %s\n%s" % (app_label, self.repr_changes(changes)))
- if len(changes[app_label]) < index + 1:
- self.fail("No migration at index %s for %s\n%s" % (index, app_label, self.repr_changes(changes)))
- migration = changes[app_label][index]
- if len(changes[app_label]) < index + 1:
- self.fail("No operation at index %s for %s.%s\n%s" % (
- operation_index,
- app_label,
- migration.name,
- self.repr_changes(changes),
- ))
- operation = migration.operations[operation_index]
- for attr, value in attrs.items():
- if getattr(operation, attr, None) != value:
- self.fail("Attribute mismatch for %s.%s op #%s, %s (expected %r, got %r):\n%s" % (
- app_label,
- migration.name,
- operation_index,
- attr,
- value,
- getattr(operation, attr, None),
- self.repr_changes(changes),
- ))
- def make_project_state(self, model_states):
- "Shortcut to make ProjectStates from lists of predefined models"
- project_state = ProjectState()
- for model_state in model_states:
- project_state.add_model_state(model_state.clone())
- return project_state
- def test_arrange_for_graph(self):
- "Tests auto-naming of migrations for graph matching."
-
- graph = MigrationGraph()
- graph.add_node(("testapp", "0001_initial"), None)
- graph.add_node(("testapp", "0002_foobar"), None)
- graph.add_node(("otherapp", "0001_initial"), None)
- graph.add_dependency("testapp.0002_foobar", ("testapp", "0002_foobar"), ("testapp", "0001_initial"))
- graph.add_dependency("testapp.0002_foobar", ("testapp", "0002_foobar"), ("otherapp", "0001_initial"))
-
- before = self.make_project_state([])
- after = self.make_project_state([self.author_empty, self.other_pony, self.other_stable])
- autodetector = MigrationAutodetector(before, after)
- changes = autodetector._detect_changes()
-
- changes = autodetector.arrange_for_graph(changes, graph)
-
- self.assertEqual(changes["testapp"][0].name, "0003_author")
- self.assertEqual(changes["testapp"][0].dependencies, [("testapp", "0002_foobar")])
- self.assertEqual(changes["otherapp"][0].name, "0002_pony_stable")
- self.assertEqual(changes["otherapp"][0].dependencies, [("otherapp", "0001_initial")])
- def test_trim_apps(self):
- "Tests that trim does not remove dependencies but does remove unwanted apps"
-
- before = self.make_project_state([])
- after = self.make_project_state([self.author_empty, self.other_pony, self.other_stable, self.third_thing])
- autodetector = MigrationAutodetector(before, after, MigrationQuestioner(defaults={"ask_initial": True}))
- changes = autodetector._detect_changes()
-
- graph = MigrationGraph()
- changes = autodetector.arrange_for_graph(changes, graph)
- changes["testapp"][0].dependencies.append(("otherapp", "0001_initial"))
- changes = autodetector._trim_to_apps(changes, {"testapp"})
-
- self.assertEqual(changes["testapp"][0].name, "0001_initial")
- self.assertEqual(changes["otherapp"][0].name, "0001_initial")
- self.assertNotIn("thirdapp", changes)
- def test_custom_migration_name(self):
- "Tests custom naming of migrations for graph matching."
-
- graph = MigrationGraph()
- graph.add_node(("testapp", "0001_initial"), None)
- graph.add_node(("testapp", "0002_foobar"), None)
- graph.add_node(("otherapp", "0001_initial"), None)
- graph.add_dependency("testapp.0002_foobar", ("testapp", "0002_foobar"), ("testapp", "0001_initial"))
-
- before = self.make_project_state([])
- after = self.make_project_state([self.author_empty, self.other_pony, self.other_stable])
- autodetector = MigrationAutodetector(before, after)
- changes = autodetector._detect_changes()
-
- migration_name = 'custom_name'
- changes = autodetector.arrange_for_graph(changes, graph, migration_name)
-
- self.assertEqual(changes["testapp"][0].name, "0003_%s" % migration_name)
- self.assertEqual(changes["testapp"][0].dependencies, [("testapp", "0002_foobar")])
- self.assertEqual(changes["otherapp"][0].name, "0002_%s" % migration_name)
- self.assertEqual(changes["otherapp"][0].dependencies, [("otherapp", "0001_initial")])
- def test_new_model(self):
- "Tests autodetection of new models"
-
- before = self.make_project_state([])
- after = self.make_project_state([self.author_empty])
- autodetector = MigrationAutodetector(before, after)
- changes = autodetector._detect_changes()
-
- self.assertEqual(len(changes['testapp']), 1)
-
- migration = changes['testapp'][0]
- self.assertEqual(len(migration.operations), 1)
-
- action = migration.operations[0]
- self.assertEqual(action.__class__.__name__, "CreateModel")
- self.assertEqual(action.name, "Author")
- def test_old_model(self):
- "Tests deletion of old models"
-
- before = self.make_project_state([self.author_empty])
- after = self.make_project_state([])
- autodetector = MigrationAutodetector(before, after)
- changes = autodetector._detect_changes()
-
- self.assertEqual(len(changes['testapp']), 1)
-
- migration = changes['testapp'][0]
- self.assertEqual(len(migration.operations), 1)
-
- action = migration.operations[0]
- self.assertEqual(action.__class__.__name__, "DeleteModel")
- self.assertEqual(action.name, "Author")
- def test_add_field(self):
- "Tests autodetection of new fields"
-
- before = self.make_project_state([self.author_empty])
- after = self.make_project_state([self.author_name])
- autodetector = MigrationAutodetector(before, after)
- changes = autodetector._detect_changes()
-
- self.assertEqual(len(changes['testapp']), 1)
-
- migration = changes['testapp'][0]
- self.assertEqual(len(migration.operations), 1)
-
- action = migration.operations[0]
- self.assertEqual(action.__class__.__name__, "AddField")
- self.assertEqual(action.name, "name")
- def test_remove_field(self):
- "Tests autodetection of removed fields"
-
- before = self.make_project_state([self.author_name])
- after = self.make_project_state([self.author_empty])
- autodetector = MigrationAutodetector(before, after)
- changes = autodetector._detect_changes()
-
- self.assertEqual(len(changes['testapp']), 1)
-
- migration = changes['testapp'][0]
- self.assertEqual(len(migration.operations), 1)
-
- action = migration.operations[0]
- self.assertEqual(action.__class__.__name__, "RemoveField")
- self.assertEqual(action.name, "name")
- def test_alter_field(self):
- "Tests autodetection of new fields"
-
- before = self.make_project_state([self.author_name])
- after = self.make_project_state([self.author_name_longer])
- autodetector = MigrationAutodetector(before, after)
- changes = autodetector._detect_changes()
-
- self.assertEqual(len(changes['testapp']), 1)
-
- migration = changes['testapp'][0]
- self.assertEqual(len(migration.operations), 1)
-
- action = migration.operations[0]
- self.assertEqual(action.__class__.__name__, "AlterField")
- self.assertEqual(action.name, "name")
- def test_rename_field(self):
- "Tests autodetection of renamed fields"
-
- before = self.make_project_state([self.author_name])
- after = self.make_project_state([self.author_name_renamed])
- autodetector = MigrationAutodetector(before, after, MigrationQuestioner({"ask_rename": True}))
- changes = autodetector._detect_changes()
-
- self.assertNumberMigrations(changes, 'testapp', 1)
- self.assertOperationTypes(changes, 'testapp', 0, ["RenameField"])
- self.assertOperationAttributes(changes, 'testapp', 0, 0, old_name="name", new_name="names")
- def test_rename_model(self):
- "Tests autodetection of renamed models"
-
- before = self.make_project_state([self.author_with_book, self.book])
- after = self.make_project_state([self.author_renamed_with_book, self.book_with_author_renamed])
- autodetector = MigrationAutodetector(before, after, MigrationQuestioner({"ask_rename_model": True}))
- changes = autodetector._detect_changes()
-
- self.assertNumberMigrations(changes, 'testapp', 1)
-
- migration = changes['testapp'][0]
- self.assertEqual(len(migration.operations), 1)
-
- action = migration.operations[0]
- self.assertEqual(action.__class__.__name__, "RenameModel")
- self.assertEqual(action.old_name, "Author")
- self.assertEqual(action.new_name, "Writer")
-
-
- self.assertNumberMigrations(changes, 'otherapp', 0)
- def test_rename_model_with_renamed_rel_field(self):
- """
- Tests autodetection of renamed models while simultaneously renaming one
- of the fields that relate to the renamed model.
- """
-
- before = self.make_project_state([self.author_with_book, self.book])
- after = self.make_project_state([self.author_renamed_with_book, self.book_with_field_and_author_renamed])
- autodetector = MigrationAutodetector(before, after, MigrationQuestioner({"ask_rename_model": True, "ask_rename": True}))
- changes = autodetector._detect_changes()
-
- self.assertNumberMigrations(changes, 'testapp', 1)
-
- migration = changes['testapp'][0]
- self.assertEqual(len(migration.operations), 1)
-
- action = migration.operations[0]
- self.assertEqual(action.__class__.__name__, "RenameModel")
- self.assertEqual(action.old_name, "Author")
- self.assertEqual(action.new_name, "Writer")
-
-
- self.assertNumberMigrations(changes, 'otherapp', 1)
-
- migration = changes['otherapp'][0]
- self.assertEqual(len(migration.operations), 1)
-
- action = migration.operations[0]
- self.assertEqual(action.__class__.__name__, "RenameField")
- self.assertEqual(action.old_name, "author")
- self.assertEqual(action.new_name, "writer")
- def test_fk_dependency(self):
- "Tests that having a ForeignKey automatically adds a dependency"
-
-
-
-
- before = self.make_project_state([])
- after = self.make_project_state([self.author_name, self.book, self.edition])
- autodetector = MigrationAutodetector(before, after)
- changes = autodetector._detect_changes()
-
- self.assertEqual(len(changes['testapp']), 1)
- self.assertEqual(len(changes['otherapp']), 1)
- self.assertEqual(len(changes['thirdapp']), 1)
-
- migration1 = changes['testapp'][0]
- self.assertEqual(len(migration1.operations), 1)
- migration2 = changes['otherapp'][0]
- self.assertEqual(len(migration2.operations), 1)
- migration3 = changes['thirdapp'][0]
- self.assertEqual(len(migration3.operations), 1)
-
- action = migration1.operations[0]
- self.assertEqual(action.__class__.__name__, "CreateModel")
- action = migration2.operations[0]
- self.assertEqual(action.__class__.__name__, "CreateModel")
- action = migration3.operations[0]
- self.assertEqual(action.__class__.__name__, "CreateModel")
-
- self.assertEqual(migration1.dependencies, [])
- self.assertEqual(migration2.dependencies, [("testapp", "auto_1")])
- self.assertEqual(migration3.dependencies, [("otherapp", "auto_1")])
- def test_proxy_fk_dependency(self):
- "Tests that FK dependencies still work on proxy models"
-
-
-
- before = self.make_project_state([])
- after = self.make_project_state([self.author_empty, self.author_proxy_third, self.book_proxy_fk])
- autodetector = MigrationAutodetector(before, after)
- changes = autodetector._detect_changes()
-
- self.assertNumberMigrations(changes, 'testapp', 1)
- self.assertNumberMigrations(changes, 'otherapp', 1)
- self.assertNumberMigrations(changes, 'thirdapp', 1)
-
-
- self.assertOperationTypes(changes, 'otherapp', 0, ["CreateModel"])
- self.assertOperationTypes(changes, 'testapp', 0, ["CreateModel"])
- self.assertOperationTypes(changes, 'thirdapp', 0, ["CreateModel"])
-
- self.assertEqual(changes['testapp'][0].dependencies, [])
- self.assertEqual(changes['otherapp'][0].dependencies, [("thirdapp", "auto_1")])
- self.assertEqual(changes['thirdapp'][0].dependencies, [("testapp", "auto_1")])
- def test_same_app_no_fk_dependency(self):
- """
- Tests that a migration with a FK between two models of the same app
- does not have a dependency to itself.
- """
-
- before = self.make_project_state([])
- after = self.make_project_state([self.author_with_publisher, self.publisher])
- autodetector = MigrationAutodetector(before, after)
- changes = autodetector._detect_changes()
-
- self.assertNumberMigrations(changes, 'testapp', 1)
- self.assertOperationTypes(changes, 'testapp', 0, ["CreateModel", "CreateModel", "AddField"])
- self.assertOperationAttributes(changes, "testapp", 0, 0, name="Author")
- self.assertOperationAttributes(changes, "testapp", 0, 1, name="Publisher")
- self.assertOperationAttributes(changes, "testapp", 0, 2, name="publisher")
-
- self.assertEqual(changes['testapp'][0].dependencies, [])
- def test_circular_fk_dependency(self):
- """
- Tests that having a circular ForeignKey dependency automatically
- resolves the situation into 2 migrations on one side and 1 on the other.
- """
-
- before = self.make_project_state([])
- after = self.make_project_state([self.author_with_book, self.book, self.publisher_with_book])
- autodetector = MigrationAutodetector(before, after)
- changes = autodetector._detect_changes()
-
- self.assertNumberMigrations(changes, 'testapp', 1)
- self.assertNumberMigrations(changes, 'otherapp', 2)
-
- self.assertOperationTypes(changes, 'testapp', 0, ["CreateModel", "CreateModel"])
- self.assertOperationTypes(changes, 'otherapp', 0, ["CreateModel"])
- self.assertOperationTypes(changes, 'otherapp', 1, ["AddField"])
- self.assertOperationAttributes(changes, "testapp", 0, 0, name="Author")
- self.assertOperationAttributes(changes, "testapp", 0, 1, name="Publisher")
-
- self.assertEqual(changes['testapp'][0].dependencies, [("otherapp", "auto_1")])
- self.assertEqual(changes['otherapp'][0].dependencies, [])
- self.assertEqual(set(changes['otherapp'][1].dependencies), {("otherapp", "auto_1"), ("testapp", "auto_1")})
- def test_same_app_circular_fk_dependency(self):
- """
- Tests that a migration with a FK between two models of the same app
- does not have a dependency to itself.
- """
-
- before = self.make_project_state([])
- after = self.make_project_state([self.author_with_publisher, self.publisher_with_author])
- autodetector = MigrationAutodetector(before, after)
- changes = autodetector._detect_changes()
-
- self.assertNumberMigrations(changes, 'testapp', 1)
- self.assertOperationTypes(changes, 'testapp', 0, ["CreateModel", "CreateModel", "AddField"])
- self.assertOperationAttributes(changes, "testapp", 0, 0, name="Author")
- self.assertOperationAttributes(changes, "testapp", 0, 1, name="Publisher")
- self.assertOperationAttributes(changes, "testapp", 0, 2, name="publisher")
-
- self.assertEqual(changes['testapp'][0].dependencies, [])
- def test_same_app_circular_fk_dependency_and_unique_together(self):
- """
- Tests that a migration with circular FK dependency does not try to
- create unique together constraint before creating all required fields first.
- See ticket #22275.
- """
-
- before = self.make_project_state([])
- after = self.make_project_state([self.knight, self.rabbit])
- autodetector = MigrationAutodetector(before, after)
- changes = autodetector._detect_changes()
-
- self.assertNumberMigrations(changes, 'eggs', 1)
- self.assertOperationTypes(changes, 'eggs', 0, ["CreateModel", "CreateModel", "AlterUniqueTogether"])
- self.assertFalse("unique_together" in changes['eggs'][0].operations[0].options)
- self.assertFalse("unique_together" in changes['eggs'][0].operations[1].options)
-
- self.assertEqual(changes['eggs'][0].dependencies, [])
- def test_unique_together(self):
- "Tests unique_together detection"
-
- before = self.make_project_state([self.author_empty, self.book])
- after = self.make_project_state([self.author_empty, self.book_unique])
- autodetector = MigrationAutodetector(before, after)
- changes = autodetector._detect_changes()
-
- self.assertEqual(len(changes['otherapp']), 1)
-
- migration = changes['otherapp'][0]
- self.assertEqual(len(migration.operations), 1)
-
- action = migration.operations[0]
- self.assertEqual(action.__class__.__name__, "AlterUniqueTogether")
- self.assertEqual(action.name, "book")
- self.assertEqual(action.unique_together, {("author", "title")})
- def test_unique_together_no_changes(self):
- "Tests that unique_togther doesn't generate a migration if no changes have been made"
-
- before = self.make_project_state([self.author_empty, self.book_unique])
- after = self.make_project_state([self.author_empty, self.book_unique])
- autodetector = MigrationAutodetector(before, after)
- changes = autodetector._detect_changes()
-
- self.assertEqual(len(changes), 0)
- def test_empty_foo_together(self):
- "#23452 - Empty unique/index_togther shouldn't generate a migration."
-
-
- model_state_not_secified = ModelState("a", "model",
- [("id", models.AutoField(primary_key=True))]
- )
-
-
- model_state_none = ModelState("a", "model",
- [("id", models.AutoField(primary_key=True))],
- {"unique_together": None, "index_together": None}
- )
-
-
- model_state_empty = ModelState("a", "model",
- [("id", models.AutoField(primary_key=True))],
- {"unique_together": set(), "index_together": set()}
- )
- def test(from_state, to_state, msg):
- before = self.make_project_state([from_state])
- after = self.make_project_state([to_state])
- autodetector = MigrationAutodetector(before, after)
- changes = autodetector._detect_changes()
- if len(changes) > 0:
- ops = ', '.join(o.__class__.__name__ for o in changes['a'][0].operations)
- self.fail('Created operation(s) %s from %s' % (ops, msg))
- tests = (
- (model_state_not_secified, model_state_not_secified, '"not specified" to "not specified"'),
- (model_state_not_secified, model_state_none, '"not specified" to "None"'),
- (model_state_not_secified, model_state_empty, '"not specified" to "empty"'),
- (model_state_none, model_state_not_secified, '"None" to "not specified"'),
- (model_state_none, model_state_none, '"None" to "None"'),
- (model_state_none, model_state_empty, '"None" to "empty"'),
- (model_state_empty, model_state_not_secified, '"empty" to "not specified"'),
- (model_state_empty, model_state_none, '"empty" to "None"'),
- (model_state_empty, model_state_empty, '"empty" to "empty"'),
- )
- for t in tests:
- test(*t)
- def test_unique_together_ordering(self):
- "Tests that unique_together also triggers on ordering changes"
-
- before = self.make_project_state([self.author_empty, self.book_unique])
- after = self.make_project_state([self.author_empty, self.book_unique_2])
- autodetector = MigrationAutodetector(before, after)
- changes = autodetector._detect_changes()
-
- self.assertEqual(len(changes['otherapp']), 1)
-
- migration = changes['otherapp'][0]
- self.assertEqual(len(migration.operations), 1)
-
- action = migration.operations[0]
- self.assertEqual(action.__class__.__name__, "AlterUniqueTogether")
- self.assertEqual(action.name, "book")
- self.assertEqual(action.unique_together, {("title", "author")})
- def test_add_field_and_unique_together(self):
- "Tests that added fields will be created before using them in unique together"
- before = self.make_project_state([self.author_empty, self.book])
- after = self.make_project_state([self.author_empty, self.book_unique_3])
- autodetector = MigrationAutodetector(before, after)
- changes = autodetector._detect_changes()
-
- self.assertEqual(len(changes['otherapp']), 1)
-
- migration = changes['otherapp'][0]
- self.assertEqual(len(migration.operations), 2)
-
- action1 = migration.operations[0]
- action2 = migration.operations[1]
- self.assertEqual(action1.__class__.__name__, "AddField")
- self.assertEqual(action2.__class__.__name__, "AlterUniqueTogether")
- self.assertEqual(action2.unique_together, {("title", "newfield")})
- def test_remove_index_together(self):
- author_index_together = ModelState("testapp", "Author", [
- ("id", models.AutoField(primary_key=True)), ("name", models.CharField(max_length=200))
- ], {"index_together": {("id", "name")}})
- before = self.make_project_state([author_index_together])
- after = self.make_project_state([self.author_name])
- autodetector = MigrationAutodetector(before, after)
- changes = autodetector._detect_changes()
-
- self.assertEqual(len(changes['testapp']), 1)
- migration = changes['testapp'][0]
-
- self.assertEqual(len(migration.operations), 1)
-
- action = migration.operations[0]
- self.assertEqual(action.__class__.__name__, "AlterIndexTogether")
- self.assertEqual(action.index_together, set())
- def test_remove_unique_together(self):
- author_unique_together = ModelState("testapp", "Author", [
- ("id", models.AutoField(primary_key=True)), ("name", models.CharField(max_length=200))
- ], {"unique_together": {("id", "name")}})
- before = self.make_project_state([author_unique_together])
- after = self.make_project_state([self.author_name])
- autodetector = MigrationAutodetector(before, after)
- changes = autodetector._detect_changes()
-
- self.assertEqual(len(changes['testapp']), 1)
- migration = changes['testapp'][0]
-
- self.assertEqual(len(migration.operations), 1)
-
- action = migration.operations[0]
- self.assertEqual(action.__class__.__name__, "AlterUniqueTogether")
- self.assertEqual(action.unique_together, set())
- def test_proxy(self):
- "Tests that the autodetector correctly deals with proxy models"
-
- before = self.make_project_state([self.author_empty])
- after = self.make_project_state([self.author_empty, self.author_proxy])
- autodetector = MigrationAutodetector(before, after)
- changes = autodetector._detect_changes()
-
- self.assertNumberMigrations(changes, "testapp", 1)
- self.assertOperationTypes(changes, "testapp", 0, ["CreateModel"])
- self.assertOperationAttributes(changes, "testapp", 0, 0, name="AuthorProxy", options={"proxy": True})
-
-
- before = self.make_project_state([self.author_empty, self.author_proxy])
- after = self.make_project_state([self.author_empty, self.author_proxy_notproxy])
- autodetector = MigrationAutodetector(before, after)
- changes = autodetector._detect_changes()
-
- self.assertNumberMigrations(changes, "testapp", 1)
- self.assertOperationTypes(changes, "testapp", 0, ["DeleteModel", "CreateModel"])
- self.assertOperationAttributes(changes, "testapp", 0, 0, name="AuthorProxy")
- self.assertOperationAttributes(changes, "testapp", 0, 1, name="AuthorProxy", options={})
- def test_proxy_custom_pk(self):
- "#23415 - The autodetector must correctly deal with custom FK on proxy models."
-
- before = self.make_project_state([])
- after = self.make_project_state([self.author_empty, self.author_proxy_third, self.book_proxy_fk])
- autodetector = MigrationAutodetector(before, after)
- changes = autodetector._detect_changes()
-
- self.assertEqual(changes['otherapp'][0].operations[0].fields[2][1].rel.field_name, 'id')
-
- before = self.make_project_state([])
- after = self.make_project_state([self.author_custom_pk, self.author_proxy_third, self.book_proxy_fk])
- autodetector = MigrationAutodetector(before, after)
- changes = autodetector._detect_changes()
-
- self.assertEqual(changes['otherapp'][0].operations[0].fields[2][1].rel.field_name, 'pk_field')
- def test_unmanaged(self):
- "Tests that the autodetector correctly deals with managed models"
-
- before = self.make_project_state([self.author_empty])
- after = self.make_project_state([self.author_empty, self.author_unmanaged])
- autodetector = MigrationAutodetector(before, after)
- changes = autodetector._detect_changes()
-
- self.assertNumberMigrations(changes, 'testapp', 1)
- self.assertOperationTypes(changes, 'testapp', 0, ["CreateModel"])
- self.assertOperationAttributes(changes, 'testapp', 0, 0, name="AuthorUnmanaged")
- self.assertEqual(changes['testapp'][0].operations[0].options['managed'], False)
-
- before = self.make_project_state([self.author_empty, self.author_unmanaged])
- after = self.make_project_state([self.author_empty, self.author_unmanaged_managed])
- autodetector = MigrationAutodetector(before, after)
- changes = autodetector._detect_changes()
-
- self.assertNumberMigrations(changes, 'testapp', 1)
- self.assertOperationTypes(changes, 'testapp', 0, ["DeleteModel", "CreateModel"])
- self.assertOperationAttributes(changes, 'testapp', 0, 0, name="AuthorUnmanaged")
- self.assertOperationAttributes(changes, 'testapp', 0, 1, name="AuthorUnmanaged")
- def test_unmanaged_custom_pk(self):
- "#23415 - The autodetector must correctly deal with custom FK on unmanaged models."
-
- before = self.make_project_state([])
- after = self.make_project_state([self.author_unmanaged_default_pk, self.book])
- autodetector = MigrationAutodetector(before, after)
- changes = autodetector._detect_changes()
-
- self.assertEqual(changes['otherapp'][0].operations[0].fields[2][1].rel.field_name, 'id')
-
- before = self.make_project_state([])
- after = self.make_project_state([self.author_unmanaged_custom_pk, self.book])
- autodetector = MigrationAutodetector(before, after)
- changes = autodetector._detect_changes()
-
- self.assertEqual(changes['otherapp'][0].operations[0].fields[2][1].rel.field_name, 'pk_field')
- @override_settings(AUTH_USER_MODEL="thirdapp.CustomUser")
- def test_swappable(self):
- before = self.make_project_state([self.custom_user])
- after = self.make_project_state([self.custom_user, self.author_with_custom_user])
- autodetector = MigrationAutodetector(before, after)
- changes = autodetector._detect_changes()
-
- self.assertEqual(len(changes), 1)
-
- migration = changes['testapp'][0]
- self.assertEqual(migration.dependencies, [("__setting__", "AUTH_USER_MODEL")])
- def test_add_field_with_default(self):
- """
- Adding a field with a default should work (#22030).
- """
-
- before = self.make_project_state([self.author_empty])
- after = self.make_project_state([self.author_name_default])
- autodetector = MigrationAutodetector(before, after)
- changes = autodetector._detect_changes()
-
- self.assertEqual(len(changes['testapp']), 1)
-
- migration = changes['testapp'][0]
- self.assertEqual(len(migration.operations), 1)
-
- action = migration.operations[0]
- self.assertEqual(action.__class__.__name__, "AddField")
- self.assertEqual(action.name, "name")
- def test_custom_deconstructable(self):
- """
- Two instances which deconstruct to the same value aren't considered a
- change.
- """
- before = self.make_project_state([self.author_name_deconstructable_1])
- after = self.make_project_state([self.author_name_deconstructable_2])
- autodetector = MigrationAutodetector(before, after)
- changes = autodetector._detect_changes()
- self.assertEqual(changes, {})
- def test_deconstruct_field_kwarg(self):
- """
- Field instances are handled correctly by nested deconstruction.
- """
- before = self.make_project_state([self.author_name_deconstructable_3])
- after = self.make_project_state([self.author_name_deconstructable_4])
- autodetector = MigrationAutodetector(before, after)
- changes = autodetector._detect_changes()
- self.assertEqual(changes, {})
- def test_deconstruct_type(self):
- """
- #22951 -- Uninstanted classes with deconstruct are correctly returned
- by deep_deconstruct during serialization.
- """
- author = ModelState(
- "testapp",
- "Author",
- [
- ("id", models.AutoField(primary_key=True)),
- ("name", models.CharField(
- max_length=200,
-
- default=models.IntegerField,
- ))
- ],
- )
-
- before = self.make_project_state([])
- after = self.make_project_state([author])
- autodetector = MigrationAutodetector(before, after)
- changes = autodetector._detect_changes()
- self.assertNumberMigrations(changes, 'testapp', 1)
- self.assertOperationTypes(changes, 'testapp', 0, ["CreateModel"])
- def test_replace_string_with_foreignkey(self):
- """
- Adding an FK in the same "spot" as a deleted CharField should work. (#22300).
- """
-
- before = self.make_project_state([self.author_with_publisher_string])
- after = self.make_project_state([self.author_with_publisher, self.publisher])
- autodetector = MigrationAutodetector(before, after)
- changes = autodetector._detect_changes()
-
- self.assertNumberMigrations(changes, 'testapp', 1)
- self.assertOperationTypes(changes, 'testapp', 0, ["CreateModel", "RemoveField", "AddField"])
- self.assertOperationAttributes(changes, 'testapp', 0, 0, name="Publisher")
- self.assertOperationAttributes(changes, 'testapp', 0, 1, name="publisher_name")
- self.assertOperationAttributes(changes, 'testapp', 0, 2, name="publisher")
- def test_foreign_key_removed_before_target_model(self):
- """
- Removing an FK and the model it targets in the same change must remove
- the FK field before the model to maintain consistency.
- """
- before = self.make_project_state([self.author_with_publisher, self.publisher])
- after = self.make_project_state([self.author_name])
- autodetector = MigrationAutodetector(before, after)
- changes = autodetector._detect_changes()
-
- self.assertEqual(len(changes['testapp']), 1)
-
- migration = changes['testapp'][0]
- self.assertEqual(len(migration.operations), 2)
-
- action = migration.operations[0]
- self.assertEqual(action.__class__.__name__, "RemoveField")
- self.assertEqual(action.name, "publisher")
- action = migration.operations[1]
- self.assertEqual(action.__class__.__name__, "DeleteModel")
- self.assertEqual(action.name, "Publisher")
- def test_add_many_to_many(self):
- """
- Adding a ManyToManyField should not prompt for a default (#22435).
- """
- class CustomQuestioner(MigrationQuestioner):
- def ask_not_null_addition(self, field_name, model_name):
- raise Exception("Should not have prompted for not null addition")
- before = self.make_project_state([self.author_empty, self.publisher])
-
- after = self.make_project_state([self.author_with_m2m, self.publisher])
- autodetector = MigrationAutodetector(before, after, CustomQuestioner())
- changes = autodetector._detect_changes()
-
- self.assertEqual(len(changes['testapp']), 1)
- migration = changes['testapp'][0]
-
- self.assertEqual(len(migration.operations), 1)
- action = migration.operations[0]
- self.assertEqual(action.__class__.__name__, "AddField")
- self.assertEqual(action.name, "publishers")
- def test_create_with_through_model(self):
- """
- Adding a m2m with a through model and the models that use it should
- be ordered correctly.
- """
- before = self.make_project_state([])
- after = self.make_project_state([self.author_with_m2m_through, self.publisher, self.contract])
- autodetector = MigrationAutodetector(before, after)
- changes = autodetector._detect_changes()
-
- self.assertNumberMigrations(changes, "testapp", 1)
-
- self.assertOperationTypes(changes, "testapp", 0, ["CreateModel", "CreateModel", "CreateModel", "AddField", "AddField"])
- def test_many_to_many_removed_before_through_model(self):
- """
- Removing a ManyToManyField and the "through" model in the same change must remove
- the field before the model to maintain consistency.
- """
- before = self.make_project_state([self.book_with_multiple_authors_through_attribution, self.author_name, self.attribution])
- after = self.make_project_state([self.book_with_no_author, self.author_name])
- autodetector = MigrationAutodetector(before, after)
- changes = autodetector._detect_changes()
-
- self.assertEqual(len(changes['otherapp']), 1)
-
- migration = changes['otherapp'][0]
- self.assertEqual(len(migration.operations), 4)
-
-
-
- action = migration.operations[0]
- self.assertEqual(action.__class__.__name__, "RemoveField")
- self.assertEqual(action.name, "author")
- action = migration.operations[1]
- self.assertEqual(action.__class__.__name__, "RemoveField")
- self.assertEqual(action.name, "book")
- action = migration.operations[2]
- self.assertEqual(action.__class__.__name__, "RemoveField")
- self.assertEqual(action.name, "authors")
- action = migration.operations[3]
- self.assertEqual(action.__class__.__name__, "DeleteModel")
- self.assertEqual(action.name, "Attribution")
- def test_many_to_many_removed_before_through_model_2(self):
- """
- Removing a model that contains a ManyToManyField and the
- "through" model in the same change must remove
- the field before the model to maintain consistency.
- """
- before = self.make_project_state([self.book_with_multiple_authors_through_attribution, self.author_name, self.attribution])
- after = self.make_project_state([self.author_name])
- autodetector = MigrationAutodetector(before, after)
- changes = autodetector._detect_changes()
-
- self.assertNumberMigrations(changes, 'otherapp', 1)
-
- self.assertOperationTypes(changes, 'otherapp', 0, ["RemoveField", "RemoveField", "RemoveField", "DeleteModel", "DeleteModel"])
- def test_m2m_w_through_multistep_remove(self):
- """
- A model with a m2m field that specifies a "through" model cannot be removed in the same
- migration as that through model as the schema will pass through an inconsistent state.
- The autodetector should produce two migrations to avoid this issue.
- """
- before = self.make_project_state([self.author_with_m2m_through, self.publisher, self.contract])
- after = self.make_project_state([self.publisher])
- autodetector = MigrationAutodetector(before, after)
- changes = autodetector._detect_changes()
-
- self.assertNumberMigrations(changes, "testapp", 1)
-
- self.assertOperationTypes(changes, "testapp", 0, ["RemoveField", "RemoveField", "DeleteModel", "RemoveField", "DeleteModel"])
-
- self.assertOperationAttributes(changes, "testapp", 0, 0, name="publishers")
- self.assertOperationAttributes(changes, "testapp", 0, 1, name="author")
- self.assertOperationAttributes(changes, "testapp", 0, 2, name="Author")
- self.assertOperationAttributes(changes, "testapp", 0, 3, name="publisher")
- self.assertOperationAttributes(changes, "testapp", 0, 4, name="Contract")
- def test_non_circular_foreignkey_dependency_removal(self):
- """
- If two models with a ForeignKey from one to the other are removed at the same time,
- the autodetector should remove them in the correct order.
- """
- before = self.make_project_state([self.author_with_publisher, self.publisher_with_author])
- after = self.make_project_state([])
- autodetector = MigrationAutodetector(before, after)
- changes = autodetector._detect_changes()
-
- self.assertNumberMigrations(changes, "testapp", 1)
-
- self.assertOperationTypes(changes, "testapp", 0, ["RemoveField", "RemoveField", "DeleteModel", "DeleteModel"])
- def test_alter_model_options(self):
- """
- Changing a model's options should make a change
- """
- before = self.make_project_state([self.author_empty])
- after = self.make_project_state([self.author_with_options])
- autodetector = MigrationAutodetector(before, after)
- changes = autodetector._detect_changes()
-
- self.assertNumberMigrations(changes, "testapp", 1)
-
- self.assertOperationTypes(changes, "testapp", 0, ["AlterModelOptions"])
-
- before = self.make_project_state([self.author_with_options])
- after = self.make_project_state([self.author_empty])
- autodetector = MigrationAutodetector(before, after)
- changes = autodetector._detect_changes()
- self.assertNumberMigrations(changes, "testapp", 1)
- self.assertOperationTypes(changes, "testapp", 0, ["AlterModelOptions"])
- def test_alter_model_options_proxy(self):
- """
- Changing a proxy model's options should also make a change
- """
- before = self.make_project_state([self.author_proxy, self.author_empty])
- after = self.make_project_state([self.author_proxy_options, self.author_empty])
- autodetector = MigrationAutodetector(before, after)
- changes = autodetector._detect_changes()
-
- self.assertNumberMigrations(changes, "testapp", 1)
-
- self.assertOperationTypes(changes, "testapp", 0, ["AlterModelOptions"])
- def test_set_alter_order_with_respect_to(self):
- "Tests that setting order_with_respect_to adds a field"
-
- before = self.make_project_state([self.book, self.author_with_book])
- after = self.make_project_state([self.book, self.author_with_book_order_wrt])
- autodetector = MigrationAutodetector(before, after)
- changes = autodetector._detect_changes()
-
- self.assertNumberMigrations(changes, 'testapp', 1)
- self.assertOperationTypes(changes, 'testapp', 0, ["AlterOrderWithRespectTo"])
- self.assertOperationAttributes(changes, 'testapp', 0, 0, name="author", order_with_respect_to="book")
- def test_add_alter_order_with_respect_to(self):
- """
- Tests that setting order_with_respect_to when adding the FK too
- does things in the right order.
- """
-
- before = self.make_project_state([self.author_name])
- after = self.make_project_state([self.book, self.author_with_book_order_wrt])
- autodetector = MigrationAutodetector(before, after)
- changes = autodetector._detect_changes()
-
- self.assertNumberMigrations(changes, 'testapp', 1)
- self.assertOperationTypes(changes, 'testapp', 0, ["AddField", "AlterOrderWithRespectTo"])
- self.assertOperationAttributes(changes, 'testapp', 0, 0, model_name="author", name="book")
- self.assertOperationAttributes(changes, 'testapp', 0, 1, name="author", order_with_respect_to="book")
- def test_remove_alter_order_with_respect_to(self):
- """
- Tests that removing order_with_respect_to when removing the FK too
- does things in the right order.
- """
-
- before = self.make_project_state([self.book, self.author_with_book_order_wrt])
- after = self.make_project_state([self.author_name])
- autodetector = MigrationAutodetector(before, after)
- changes = autodetector._detect_changes()
-
- self.assertNumberMigrations(changes, 'testapp', 1)
- self.assertOperationTypes(changes, 'testapp', 0, ["AlterOrderWithRespectTo", "RemoveField"])
- self.assertOperationAttributes(changes, 'testapp', 0, 0, name="author", order_with_respect_to=None)
- self.assertOperationAttributes(changes, 'testapp', 0, 1, model_name="author", name="book")
- def test_add_model_order_with_respect_to(self):
- """
- Tests that setting order_with_respect_to when adding the whole model
- does things in the right order.
- """
-
- before = self.make_project_state([])
- after = self.make_project_state([self.book, self.author_with_book_order_wrt])
- autodetector = MigrationAutodetector(before, after)
- changes = autodetector._detect_changes()
-
- self.assertNumberMigrations(changes, 'testapp', 1)
- self.assertOperationTypes(changes, 'testapp', 0, ["CreateModel", "AlterOrderWithRespectTo"])
- self.assertOperationAttributes(changes, 'testapp', 0, 1, name="author", order_with_respect_to="book")
-
- self.assertNotIn("_order", [name for name, field in changes['testapp'][0].operations[0].fields])
- def test_swappable_first_inheritance(self):
- """
- Tests that swappable models get their CreateModel first.
- """
-
- before = self.make_project_state([])
- after = self.make_project_state([self.custom_user, self.aardvark])
- autodetector = MigrationAutodetector(before, after)
- changes = autodetector._detect_changes()
-
- self.assertNumberMigrations(changes, 'thirdapp', 1)
- self.assertOperationTypes(changes, 'thirdapp', 0, ["CreateModel", "CreateModel"])
- self.assertOperationAttributes(changes, 'thirdapp', 0, 0, name="CustomUser")
- self.assertOperationAttributes(changes, 'thirdapp', 0, 1, name="Aardvark")
- @override_settings(AUTH_USER_MODEL="thirdapp.CustomUser")
- def test_swappable_first_setting(self):
- """
- Tests that swappable models get their CreateModel first.
- """
-
- before = self.make_project_state([])
- after = self.make_project_state([self.custom_user_no_inherit, self.aardvark])
- autodetector = MigrationAutodetector(before, after)
- changes = autodetector._detect_changes()
-
- self.assertNumberMigrations(changes, 'thirdapp', 1)
- self.assertOperationTypes(changes, 'thirdapp', 0, ["CreateModel", "CreateModel"])
- self.assertOperationAttributes(changes, 'thirdapp', 0, 0, name="CustomUser")
- self.assertOperationAttributes(changes, 'thirdapp', 0, 1, name="Aardvark")
- def test_bases_first(self):
- """
- Tests that bases of other models come first.
- """
-
- before = self.make_project_state([])
- after = self.make_project_state([self.aardvark_based_on_author, self.author_name])
- autodetector = MigrationAutodetector(before, after)
- changes = autodetector._detect_changes()
-
- self.assertNumberMigrations(changes, 'testapp', 1)
- self.assertOperationTypes(changes, 'testapp', 0, ["CreateModel", "CreateModel"])
- self.assertOperationAttributes(changes, 'testapp', 0, 0, name="Author")
- self.assertOperationAttributes(changes, 'testapp', 0, 1, name="Aardvark")
- def test_proxy_bases_first(self):
- """
- Tests that bases of proxies come first.
- """
-
- before = self.make_project_state([])
- after = self.make_project_state([self.author_empty, self.author_proxy, self.author_proxy_proxy])
- autodetector = MigrationAutodetector(before, after)
- changes = autodetector._detect_changes()
-
- self.assertNumberMigrations(changes, 'testapp', 1)
- self.assertOperationTypes(changes, 'testapp', 0, ["CreateModel", "CreateModel", "CreateModel"])
- self.assertOperationAttributes(changes, 'testapp', 0, 0, name="Author")
- self.assertOperationAttributes(changes, 'testapp', 0, 1, name="AuthorProxy")
- self.assertOperationAttributes(changes, 'testapp', 0, 2, name="AAuthorProxyProxy")
- def test_pk_fk_included(self):
- """
- Tests that a relation used as the primary key is kept as part of CreateModel.
- """
-
- before = self.make_project_state([])
- after = self.make_project_state([self.aardvark_pk_fk_author, self.author_name])
- autodetector = MigrationAutodetector(before, after)
- changes = autodetector._detect_changes()
-
- self.assertNumberMigrations(changes, 'testapp', 1)
- self.assertOperationTypes(changes, 'testapp', 0, ["CreateModel", "CreateModel"])
- self.assertOperationAttributes(changes, 'testapp', 0, 0, name="Author")
- self.assertOperationAttributes(changes, 'testapp', 0, 1, name="Aardvark")
- def test_first_dependency(self):
- """
- Tests that a dependency to an app with no migrations uses __first__.
- """
-
- loader = MigrationLoader(connection)
-
- before = self.make_project_state([])
- after = self.make_project_state([self.book_migrations_fk])
- after.real_apps = ["migrations"]
- autodetector = MigrationAutodetector(before, after)
- changes = autodetector._detect_changes(graph=loader.graph)
-
- self.assertNumberMigrations(changes, 'otherapp', 1)
- self.assertOperationTypes(changes, 'otherapp', 0, ["CreateModel"])
- self.assertOperationAttributes(changes, 'otherapp', 0, 0, name="Book")
-
- self.assertEqual(changes['otherapp'][0].dependencies, [("migrations", "__first__")])
- @override_settings(MIGRATION_MODULES={"migrations": "migrations.test_migrations"})
- def test_last_dependency(self):
- """
- Tests that a dependency to an app with existing migrations uses the
- last migration of that app.
- """
-
- loader = MigrationLoader(connection)
-
- before = self.make_project_state([])
- after = self.make_project_state([self.book_migrations_fk])
- after.real_apps = ["migrations"]
- autodetector = MigrationAutodetector(before, after)
- changes = autodetector._detect_changes(graph=loader.graph)
-
- self.assertNumberMigrations(changes, 'otherapp', 1)
- self.assertOperationTypes(changes, 'otherapp', 0, ["CreateModel"])
- self.assertOperationAttributes(changes, 'otherapp', 0, 0, name="Book")
-
- self.assertEqual(changes['otherapp'][0].dependencies, [("migrations", "0002_second")])
- def test_alter_fk_before_model_deletion(self):
- """
- Tests that ForeignKeys are altered _before_ the model they used to
- refer to are deleted.
- """
-
- before = self.make_project_state([self.author_name, self.publisher_with_author])
- after = self.make_project_state([self.aardvark_testapp, self.publisher_with_aardvark_author])
- autodetector = MigrationAutodetector(before, after)
- changes = autodetector._detect_changes()
-
- self.assertNumberMigrations(changes, 'testapp', 1)
- self.assertOperationTypes(changes, 'testapp', 0, ["CreateModel", "AlterField", "DeleteModel"])
- self.assertOperationAttributes(changes, 'testapp', 0, 0, name="Aardvark")
- self.assertOperationAttributes(changes, 'testapp', 0, 1, name="author")
- self.assertOperationAttributes(changes, 'testapp', 0, 2, name="Author")
- def test_fk_dependency_other_app(self):
- """
- Tests that ForeignKeys correctly depend on other apps' models (#23100)
- """
-
- before = self.make_project_state([self.author_name, self.book])
- after = self.make_project_state([self.author_with_book, self.book])
- autodetector = MigrationAutodetector(before, after)
- changes = autodetector._detect_changes()
-
- self.assertNumberMigrations(changes, 'testapp', 1)
- self.assertOperationTypes(changes, 'testapp', 0, ["AddField"])
- self.assertOperationAttributes(changes, 'testapp', 0, 0, name="book")
- self.assertEqual(changes['testapp'][0].dependencies, [("otherapp", "__first__")])
- def test_circular_dependency_mixed_addcreate(self):
- """
- Tests that the dependency resolver knows to put all CreateModel
- before AddField and not become unsolvable (#23315)
- """
- address = ModelState("a", "Address", [
- ("id", models.AutoField(primary_key=True)),
- ("country", models.ForeignKey("b.DeliveryCountry")),
- ])
- person = ModelState("a", "Person", [
- ("id", models.AutoField(primary_key=True)),
- ])
- apackage = ModelState("b", "APackage", [
- ("id", models.AutoField(primary_key=True)),
- ("person", models.ForeignKey("a.Person")),
- ])
- country = ModelState("b", "DeliveryCountry", [
- ("id", models.AutoField(primary_key=True)),
- ])
-
- before = self.make_project_state([])
- after = self.make_project_state([address, person, apackage, country])
- autodetector = MigrationAutodetector(before, after)
- changes = autodetector._detect_changes()
-
- self.assertNumberMigrations(changes, 'a', 2)
- self.assertNumberMigrations(changes, 'b', 1)
- self.assertOperationTypes(changes, 'a', 0, ["CreateModel", "CreateModel"])
- self.assertOperationTypes(changes, 'a', 1, ["AddField"])
- self.assertOperationTypes(changes, 'b', 0, ["CreateModel", "CreateModel"])
- @override_settings(AUTH_USER_MODEL="a.Tenant")
- def test_circular_dependency_swappable(self):
- """
- Tests that the dependency resolver knows to explicitly resolve
- swappable models (#23322)
- """
- tenant = ModelState("a", "Tenant", [
- ("id", models.AutoField(primary_key=True)),
- ("primary_address", models.ForeignKey("b.Address"))],
- bases=(AbstractBaseUser, )
- )
- address = ModelState("b", "Address", [
- ("id", models.AutoField(primary_key=True)),
- ("tenant", models.ForeignKey(settings.AUTH_USER_MODEL)),
- ])
-
- before = self.make_project_state([])
- after = self.make_project_state([address, tenant])
- autodetector = MigrationAutodetector(before, after)
- changes = autodetector._detect_changes()
-
- self.assertNumberMigrations(changes, 'a', 2)
- self.assertNumberMigrations(changes, 'b', 1)
- self.assertOperationTypes(changes, 'a', 0, ["CreateModel"])
- self.assertOperationTypes(changes, 'a', 1, ["AddField"])
- self.assertOperationTypes(changes, 'b', 0, ["CreateModel"])
- self.assertEqual(changes['a'][0].dependencies, [])
- self.assertEqual(set(changes['a'][1].dependencies), {('a', 'auto_1'), ('b', 'auto_1')})
- self.assertEqual(changes['b'][0].dependencies, [('__setting__', 'AUTH_USER_MODEL')])
- @override_settings(AUTH_USER_MODEL="b.Tenant")
- def test_circular_dependency_swappable2(self):
- """
- Tests that the dependency resolver knows to explicitly resolve
- swappable models but with the swappable not being the first migrated
- model (#23322)
- """
- address = ModelState("a", "Address", [
- ("id", models.AutoField(primary_key=True)),
- ("tenant", models.ForeignKey(settings.AUTH_USER_MODEL)),
- ])
- tenant = ModelState("b", "Tenant", [
- ("id", models.AutoField(primary_key=True)),
- ("primary_address", models.ForeignKey("a.Address"))],
- bases=(AbstractBaseUser, )
- )
-
- before = self.make_project_state([])
- after = self.make_project_state([address, tenant])
- autodetector = MigrationAutodetector(before, after)
- changes = autodetector._detect_changes()
-
- self.assertNumberMigrations(changes, 'a', 2)
- self.assertNumberMigrations(changes, 'b', 1)
- self.assertOperationTypes(changes, 'a', 0, ["CreateModel"])
- self.assertOperationTypes(changes, 'a', 1, ["AddField"])
- self.assertOperationTypes(changes, 'b', 0, ["CreateModel"])
- self.assertEqual(changes['a'][0].dependencies, [])
- self.assertEqual(set(changes['a'][1].dependencies), {('__setting__', 'AUTH_USER_MODEL'), ('a', 'auto_1')})
- self.assertEqual(changes['b'][0].dependencies, [('a', 'auto_1')])
- @override_settings(AUTH_USER_MODEL="a.Person")
- def test_circular_dependency_swappable_self(self):
- """
- Tests that the dependency resolver knows to explicitly resolve
- swappable models (#23322)
- """
- person = ModelState("a", "Person", [
- ("id", models.AutoField(primary_key=True)),
- ("parent1", models.ForeignKey(settings.AUTH_USER_MODEL, related_name='children'))
- ])
-
- before = self.make_project_state([])
- after = self.make_project_state([person])
- autodetector = MigrationAutodetector(before, after)
- changes = autodetector._detect_changes()
-
- self.assertNumberMigrations(changes, 'a', 1)
- self.assertOperationTypes(changes, 'a', 0, ["CreateModel"])
- self.assertEqual(changes['a'][0].dependencies, [])
|