2
0

test_relative_fields.py 52 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386
  1. # -*- encoding: utf-8 -*-
  2. from __future__ import unicode_literals
  3. import warnings
  4. from django.core.checks import Error, Warning as DjangoWarning
  5. from django.db import models
  6. from django.test import ignore_warnings
  7. from django.test.testcases import skipIfDBFeature
  8. from django.test.utils import override_settings
  9. from django.utils import six
  10. from django.utils.deprecation import RemovedInDjango20Warning
  11. from django.utils.version import get_docs_version
  12. from .base import IsolatedModelsTestCase
  13. class RelativeFieldTests(IsolatedModelsTestCase):
  14. def test_valid_foreign_key_without_accessor(self):
  15. class Target(models.Model):
  16. # There would be a clash if Model.field installed an accessor.
  17. model = models.IntegerField()
  18. class Model(models.Model):
  19. field = models.ForeignKey(Target, models.CASCADE, related_name='+')
  20. field = Model._meta.get_field('field')
  21. errors = field.check()
  22. self.assertEqual(errors, [])
  23. @ignore_warnings(category=RemovedInDjango20Warning)
  24. def test_valid_foreign_key_without_on_delete(self):
  25. class Target(models.Model):
  26. model = models.IntegerField()
  27. class Model(models.Model):
  28. field = models.ForeignKey(Target, related_name='+')
  29. def test_foreign_key_without_on_delete_warning(self):
  30. with warnings.catch_warnings(record=True) as warns:
  31. warnings.simplefilter('always') # prevent warnings from appearing as errors
  32. class Target(models.Model):
  33. model = models.IntegerField()
  34. class Model(models.Model):
  35. field = models.ForeignKey(Target, related_name='+')
  36. self.assertEqual(len(warns), 1)
  37. self.assertEqual(
  38. str(warns[0].message),
  39. 'on_delete will be a required arg for ForeignKey in Django '
  40. '2.0. Set it to models.CASCADE if you want to maintain the '
  41. 'current default behavior. See '
  42. 'https://docs.djangoproject.com/en/%s/ref/models/fields/'
  43. '#django.db.models.ForeignKey.on_delete' % get_docs_version(),
  44. )
  45. def test_foreign_key_to_field_as_arg(self):
  46. with warnings.catch_warnings(record=True) as warns:
  47. warnings.simplefilter('always') # prevent warnings from appearing as errors
  48. class Target(models.Model):
  49. model = models.IntegerField()
  50. class Model(models.Model):
  51. field = models.ForeignKey(Target, 'id')
  52. self.assertEqual(len(warns), 1)
  53. self.assertEqual(
  54. str(warns[0].message),
  55. "The signature for ForeignKey will change in Django 2.0. "
  56. "Pass to_field='id' as a kwarg instead of as an arg."
  57. )
  58. def test_one_to_one_field_without_on_delete_warning(self):
  59. with warnings.catch_warnings(record=True) as warns:
  60. warnings.simplefilter('always') # prevent warnings from appearing as errors
  61. class Target(models.Model):
  62. model = models.IntegerField()
  63. class Model(models.Model):
  64. field = models.OneToOneField(Target, related_name='+')
  65. self.assertEqual(len(warns), 1)
  66. self.assertEqual(
  67. str(warns[0].message),
  68. 'on_delete will be a required arg for OneToOneField in Django '
  69. '2.0. Set it to models.CASCADE if you want to maintain the '
  70. 'current default behavior. See '
  71. 'https://docs.djangoproject.com/en/%s/ref/models/fields/'
  72. '#django.db.models.ForeignKey.on_delete' % get_docs_version(),
  73. )
  74. def test_one_to_one_field_to_field_as_arg(self):
  75. with warnings.catch_warnings(record=True) as warns:
  76. warnings.simplefilter('always') # prevent warnings from appearing as errors
  77. class Target(models.Model):
  78. model = models.IntegerField()
  79. class Model(models.Model):
  80. field = models.OneToOneField(Target, 'id')
  81. self.assertEqual(len(warns), 1)
  82. self.assertEqual(
  83. str(warns[0].message),
  84. "The signature for OneToOneField will change in Django 2.0. "
  85. "Pass to_field='id' as a kwarg instead of as an arg."
  86. )
  87. def test_foreign_key_to_missing_model(self):
  88. # Model names are resolved when a model is being created, so we cannot
  89. # test relative fields in isolation and we need to attach them to a
  90. # model.
  91. class Model(models.Model):
  92. foreign_key = models.ForeignKey('Rel1', models.CASCADE)
  93. field = Model._meta.get_field('foreign_key')
  94. errors = field.check()
  95. expected = [
  96. Error(
  97. ("Field defines a relation with model 'Rel1', "
  98. "which is either not installed, or is abstract."),
  99. hint=None,
  100. obj=field,
  101. id='fields.E300',
  102. ),
  103. ]
  104. self.assertEqual(errors, expected)
  105. def test_many_to_many_to_missing_model(self):
  106. class Model(models.Model):
  107. m2m = models.ManyToManyField("Rel2")
  108. field = Model._meta.get_field('m2m')
  109. errors = field.check(from_model=Model)
  110. expected = [
  111. Error(
  112. ("Field defines a relation with model 'Rel2', "
  113. "which is either not installed, or is abstract."),
  114. hint=None,
  115. obj=field,
  116. id='fields.E300',
  117. ),
  118. ]
  119. self.assertEqual(errors, expected)
  120. def test_many_to_many_with_useless_options(self):
  121. class Model(models.Model):
  122. name = models.CharField(max_length=20)
  123. class ModelM2M(models.Model):
  124. m2m = models.ManyToManyField(Model, null=True, validators=[''])
  125. errors = ModelM2M.check()
  126. field = ModelM2M._meta.get_field('m2m')
  127. expected = [
  128. DjangoWarning(
  129. 'null has no effect on ManyToManyField.',
  130. hint=None,
  131. obj=field,
  132. id='fields.W340',
  133. )
  134. ]
  135. expected.append(
  136. DjangoWarning(
  137. 'ManyToManyField does not support validators.',
  138. hint=None,
  139. obj=field,
  140. id='fields.W341',
  141. )
  142. )
  143. self.assertEqual(errors, expected)
  144. def test_ambiguous_relationship_model(self):
  145. class Person(models.Model):
  146. pass
  147. class Group(models.Model):
  148. field = models.ManyToManyField('Person',
  149. through="AmbiguousRelationship", related_name='tertiary')
  150. class AmbiguousRelationship(models.Model):
  151. # Too much foreign keys to Person.
  152. first_person = models.ForeignKey(Person, models.CASCADE, related_name="first")
  153. second_person = models.ForeignKey(Person, models.CASCADE, related_name="second")
  154. second_model = models.ForeignKey(Group, models.CASCADE)
  155. field = Group._meta.get_field('field')
  156. errors = field.check(from_model=Group)
  157. expected = [
  158. Error(
  159. ("The model is used as an intermediate model by "
  160. "'invalid_models_tests.Group.field', but it has more than one "
  161. "foreign key to 'Person', which is ambiguous. You must specify "
  162. "which foreign key Django should use via the through_fields "
  163. "keyword argument."),
  164. hint=('If you want to create a recursive relationship, use '
  165. 'ForeignKey("self", symmetrical=False, '
  166. 'through="AmbiguousRelationship").'),
  167. obj=field,
  168. id='fields.E335',
  169. ),
  170. ]
  171. self.assertEqual(errors, expected)
  172. def test_relationship_model_with_foreign_key_to_wrong_model(self):
  173. class WrongModel(models.Model):
  174. pass
  175. class Person(models.Model):
  176. pass
  177. class Group(models.Model):
  178. members = models.ManyToManyField('Person',
  179. through="InvalidRelationship")
  180. class InvalidRelationship(models.Model):
  181. person = models.ForeignKey(Person, models.CASCADE)
  182. wrong_foreign_key = models.ForeignKey(WrongModel, models.CASCADE)
  183. # The last foreign key should point to Group model.
  184. field = Group._meta.get_field('members')
  185. errors = field.check(from_model=Group)
  186. expected = [
  187. Error(
  188. ("The model is used as an intermediate model by "
  189. "'invalid_models_tests.Group.members', but it does not "
  190. "have a foreign key to 'Group' or 'Person'."),
  191. hint=None,
  192. obj=InvalidRelationship,
  193. id='fields.E336',
  194. ),
  195. ]
  196. self.assertEqual(errors, expected)
  197. def test_relationship_model_missing_foreign_key(self):
  198. class Person(models.Model):
  199. pass
  200. class Group(models.Model):
  201. members = models.ManyToManyField('Person',
  202. through="InvalidRelationship")
  203. class InvalidRelationship(models.Model):
  204. group = models.ForeignKey(Group, models.CASCADE)
  205. # No foreign key to Person
  206. field = Group._meta.get_field('members')
  207. errors = field.check(from_model=Group)
  208. expected = [
  209. Error(
  210. ("The model is used as an intermediate model by "
  211. "'invalid_models_tests.Group.members', but it does not have "
  212. "a foreign key to 'Group' or 'Person'."),
  213. hint=None,
  214. obj=InvalidRelationship,
  215. id='fields.E336',
  216. ),
  217. ]
  218. self.assertEqual(errors, expected)
  219. def test_missing_relationship_model(self):
  220. class Person(models.Model):
  221. pass
  222. class Group(models.Model):
  223. members = models.ManyToManyField('Person',
  224. through="MissingM2MModel")
  225. field = Group._meta.get_field('members')
  226. errors = field.check(from_model=Group)
  227. expected = [
  228. Error(
  229. ("Field specifies a many-to-many relation through model "
  230. "'MissingM2MModel', which has not been installed."),
  231. hint=None,
  232. obj=field,
  233. id='fields.E331',
  234. ),
  235. ]
  236. self.assertEqual(errors, expected)
  237. def test_symmetrical_self_referential_field(self):
  238. class Person(models.Model):
  239. # Implicit symmetrical=False.
  240. friends = models.ManyToManyField('self', through="Relationship")
  241. class Relationship(models.Model):
  242. first = models.ForeignKey(Person, models.CASCADE, related_name="rel_from_set")
  243. second = models.ForeignKey(Person, models.CASCADE, related_name="rel_to_set")
  244. field = Person._meta.get_field('friends')
  245. errors = field.check(from_model=Person)
  246. expected = [
  247. Error(
  248. 'Many-to-many fields with intermediate tables must not be symmetrical.',
  249. hint=None,
  250. obj=field,
  251. id='fields.E332',
  252. ),
  253. ]
  254. self.assertEqual(errors, expected)
  255. def test_too_many_foreign_keys_in_self_referential_model(self):
  256. class Person(models.Model):
  257. friends = models.ManyToManyField('self',
  258. through="InvalidRelationship", symmetrical=False)
  259. class InvalidRelationship(models.Model):
  260. first = models.ForeignKey(Person, models.CASCADE, related_name="rel_from_set_2")
  261. second = models.ForeignKey(Person, models.CASCADE, related_name="rel_to_set_2")
  262. third = models.ForeignKey(Person, models.CASCADE, related_name="too_many_by_far")
  263. field = Person._meta.get_field('friends')
  264. errors = field.check(from_model=Person)
  265. expected = [
  266. Error(
  267. ("The model is used as an intermediate model by "
  268. "'invalid_models_tests.Person.friends', but it has more than two "
  269. "foreign keys to 'Person', which is ambiguous. You must specify "
  270. "which two foreign keys Django should use via the through_fields "
  271. "keyword argument."),
  272. hint='Use through_fields to specify which two foreign keys Django should use.',
  273. obj=InvalidRelationship,
  274. id='fields.E333',
  275. ),
  276. ]
  277. self.assertEqual(errors, expected)
  278. def test_symmetric_self_reference_with_intermediate_table(self):
  279. class Person(models.Model):
  280. # Explicit symmetrical=True.
  281. friends = models.ManyToManyField('self',
  282. through="Relationship", symmetrical=True)
  283. class Relationship(models.Model):
  284. first = models.ForeignKey(Person, models.CASCADE, related_name="rel_from_set")
  285. second = models.ForeignKey(Person, models.CASCADE, related_name="rel_to_set")
  286. field = Person._meta.get_field('friends')
  287. errors = field.check(from_model=Person)
  288. expected = [
  289. Error(
  290. 'Many-to-many fields with intermediate tables must not be symmetrical.',
  291. hint=None,
  292. obj=field,
  293. id='fields.E332',
  294. ),
  295. ]
  296. self.assertEqual(errors, expected)
  297. def test_symmetric_self_reference_with_intermediate_table_and_through_fields(self):
  298. """Using through_fields in a m2m with an intermediate model shouldn't mask its incompatibility with symmetry."""
  299. class Person(models.Model):
  300. # Explicit symmetrical=True.
  301. friends = models.ManyToManyField('self',
  302. symmetrical=True,
  303. through="Relationship",
  304. through_fields=('first', 'second'))
  305. class Relationship(models.Model):
  306. first = models.ForeignKey(Person, models.CASCADE, related_name="rel_from_set")
  307. second = models.ForeignKey(Person, models.CASCADE, related_name="rel_to_set")
  308. referee = models.ForeignKey(Person, models.CASCADE, related_name="referred")
  309. field = Person._meta.get_field('friends')
  310. errors = field.check(from_model=Person)
  311. expected = [
  312. Error(
  313. 'Many-to-many fields with intermediate tables must not be symmetrical.',
  314. hint=None,
  315. obj=field,
  316. id='fields.E332',
  317. ),
  318. ]
  319. self.assertEqual(errors, expected)
  320. def test_foreign_key_to_abstract_model(self):
  321. class Model(models.Model):
  322. foreign_key = models.ForeignKey('AbstractModel', models.CASCADE)
  323. class AbstractModel(models.Model):
  324. class Meta:
  325. abstract = True
  326. field = Model._meta.get_field('foreign_key')
  327. errors = field.check()
  328. expected = [
  329. Error(
  330. ("Field defines a relation with model 'AbstractModel', "
  331. "which is either not installed, or is abstract."),
  332. hint=None,
  333. obj=field,
  334. id='fields.E300',
  335. ),
  336. ]
  337. self.assertEqual(errors, expected)
  338. def test_m2m_to_abstract_model(self):
  339. class AbstractModel(models.Model):
  340. class Meta:
  341. abstract = True
  342. class Model(models.Model):
  343. m2m = models.ManyToManyField('AbstractModel')
  344. field = Model._meta.get_field('m2m')
  345. errors = field.check(from_model=Model)
  346. expected = [
  347. Error(
  348. ("Field defines a relation with model 'AbstractModel', "
  349. "which is either not installed, or is abstract."),
  350. hint=None,
  351. obj=field,
  352. id='fields.E300',
  353. ),
  354. ]
  355. self.assertEqual(errors, expected)
  356. def test_unique_m2m(self):
  357. class Person(models.Model):
  358. name = models.CharField(max_length=5)
  359. class Group(models.Model):
  360. members = models.ManyToManyField('Person', unique=True)
  361. field = Group._meta.get_field('members')
  362. errors = field.check(from_model=Group)
  363. expected = [
  364. Error(
  365. 'ManyToManyFields cannot be unique.',
  366. hint=None,
  367. obj=field,
  368. id='fields.E330',
  369. ),
  370. ]
  371. self.assertEqual(errors, expected)
  372. def test_foreign_key_to_non_unique_field(self):
  373. class Target(models.Model):
  374. bad = models.IntegerField() # No unique=True
  375. class Model(models.Model):
  376. foreign_key = models.ForeignKey('Target', models.CASCADE, to_field='bad')
  377. field = Model._meta.get_field('foreign_key')
  378. errors = field.check()
  379. expected = [
  380. Error(
  381. "'Target.bad' must set unique=True because it is referenced by a foreign key.",
  382. hint=None,
  383. obj=field,
  384. id='fields.E311',
  385. ),
  386. ]
  387. self.assertEqual(errors, expected)
  388. def test_foreign_key_to_non_unique_field_under_explicit_model(self):
  389. class Target(models.Model):
  390. bad = models.IntegerField()
  391. class Model(models.Model):
  392. field = models.ForeignKey(Target, models.CASCADE, to_field='bad')
  393. field = Model._meta.get_field('field')
  394. errors = field.check()
  395. expected = [
  396. Error(
  397. "'Target.bad' must set unique=True because it is referenced by a foreign key.",
  398. hint=None,
  399. obj=field,
  400. id='fields.E311',
  401. ),
  402. ]
  403. self.assertEqual(errors, expected)
  404. def test_foreign_object_to_non_unique_fields(self):
  405. class Person(models.Model):
  406. # Note that both fields are not unique.
  407. country_id = models.IntegerField()
  408. city_id = models.IntegerField()
  409. class MMembership(models.Model):
  410. person_country_id = models.IntegerField()
  411. person_city_id = models.IntegerField()
  412. person = models.ForeignObject(Person,
  413. on_delete=models.CASCADE,
  414. from_fields=['person_country_id', 'person_city_id'],
  415. to_fields=['country_id', 'city_id'])
  416. field = MMembership._meta.get_field('person')
  417. errors = field.check()
  418. expected = [
  419. Error(
  420. ("None of the fields 'country_id', 'city_id' on model 'Person' "
  421. "have a unique=True constraint."),
  422. hint=None,
  423. obj=field,
  424. id='fields.E310',
  425. )
  426. ]
  427. self.assertEqual(errors, expected)
  428. def test_on_delete_set_null_on_non_nullable_field(self):
  429. class Person(models.Model):
  430. pass
  431. class Model(models.Model):
  432. foreign_key = models.ForeignKey('Person', models.SET_NULL)
  433. field = Model._meta.get_field('foreign_key')
  434. errors = field.check()
  435. expected = [
  436. Error(
  437. 'Field specifies on_delete=SET_NULL, but cannot be null.',
  438. hint='Set null=True argument on the field, or change the on_delete rule.',
  439. obj=field,
  440. id='fields.E320',
  441. ),
  442. ]
  443. self.assertEqual(errors, expected)
  444. def test_on_delete_set_default_without_default_value(self):
  445. class Person(models.Model):
  446. pass
  447. class Model(models.Model):
  448. foreign_key = models.ForeignKey('Person', models.SET_DEFAULT)
  449. field = Model._meta.get_field('foreign_key')
  450. errors = field.check()
  451. expected = [
  452. Error(
  453. 'Field specifies on_delete=SET_DEFAULT, but has no default value.',
  454. hint='Set a default value, or change the on_delete rule.',
  455. obj=field,
  456. id='fields.E321',
  457. ),
  458. ]
  459. self.assertEqual(errors, expected)
  460. @skipIfDBFeature('interprets_empty_strings_as_nulls')
  461. def test_nullable_primary_key(self):
  462. class Model(models.Model):
  463. field = models.IntegerField(primary_key=True, null=True)
  464. field = Model._meta.get_field('field')
  465. errors = field.check()
  466. expected = [
  467. Error(
  468. 'Primary keys must not have null=True.',
  469. hint='Set null=False on the field, or remove primary_key=True argument.',
  470. obj=field,
  471. id='fields.E007',
  472. ),
  473. ]
  474. self.assertEqual(errors, expected)
  475. def test_not_swapped_model(self):
  476. class SwappableModel(models.Model):
  477. # A model that can be, but isn't swapped out. References to this
  478. # model should *not* raise any validation error.
  479. class Meta:
  480. swappable = 'TEST_SWAPPABLE_MODEL'
  481. class Model(models.Model):
  482. explicit_fk = models.ForeignKey(SwappableModel,
  483. models.CASCADE,
  484. related_name='explicit_fk')
  485. implicit_fk = models.ForeignKey('invalid_models_tests.SwappableModel',
  486. models.CASCADE,
  487. related_name='implicit_fk')
  488. explicit_m2m = models.ManyToManyField(SwappableModel,
  489. related_name='explicit_m2m')
  490. implicit_m2m = models.ManyToManyField(
  491. 'invalid_models_tests.SwappableModel',
  492. related_name='implicit_m2m')
  493. explicit_fk = Model._meta.get_field('explicit_fk')
  494. self.assertEqual(explicit_fk.check(), [])
  495. implicit_fk = Model._meta.get_field('implicit_fk')
  496. self.assertEqual(implicit_fk.check(), [])
  497. explicit_m2m = Model._meta.get_field('explicit_m2m')
  498. self.assertEqual(explicit_m2m.check(from_model=Model), [])
  499. implicit_m2m = Model._meta.get_field('implicit_m2m')
  500. self.assertEqual(implicit_m2m.check(from_model=Model), [])
  501. @override_settings(TEST_SWAPPED_MODEL='invalid_models_tests.Replacement')
  502. def test_referencing_to_swapped_model(self):
  503. class Replacement(models.Model):
  504. pass
  505. class SwappedModel(models.Model):
  506. class Meta:
  507. swappable = 'TEST_SWAPPED_MODEL'
  508. class Model(models.Model):
  509. explicit_fk = models.ForeignKey(SwappedModel,
  510. models.CASCADE,
  511. related_name='explicit_fk')
  512. implicit_fk = models.ForeignKey('invalid_models_tests.SwappedModel',
  513. models.CASCADE,
  514. related_name='implicit_fk')
  515. explicit_m2m = models.ManyToManyField(SwappedModel,
  516. related_name='explicit_m2m')
  517. implicit_m2m = models.ManyToManyField(
  518. 'invalid_models_tests.SwappedModel',
  519. related_name='implicit_m2m')
  520. fields = [
  521. Model._meta.get_field('explicit_fk'),
  522. Model._meta.get_field('implicit_fk'),
  523. Model._meta.get_field('explicit_m2m'),
  524. Model._meta.get_field('implicit_m2m'),
  525. ]
  526. expected_error = Error(
  527. ("Field defines a relation with the model "
  528. "'invalid_models_tests.SwappedModel', which has been swapped out."),
  529. hint="Update the relation to point at 'settings.TEST_SWAPPED_MODEL'.",
  530. id='fields.E301',
  531. )
  532. for field in fields:
  533. expected_error.obj = field
  534. errors = field.check(from_model=Model)
  535. self.assertEqual(errors, [expected_error])
  536. def test_related_field_has_invalid_related_name(self):
  537. digit = 0
  538. illegal_non_alphanumeric = '!'
  539. whitespace = '\t'
  540. invalid_related_names = [
  541. '%s_begins_with_digit' % digit,
  542. '%s_begins_with_illegal_non_alphanumeric' % illegal_non_alphanumeric,
  543. '%s_begins_with_whitespace' % whitespace,
  544. 'contains_%s_illegal_non_alphanumeric' % illegal_non_alphanumeric,
  545. 'contains_%s_whitespace' % whitespace,
  546. 'ends_with_with_illegal_non_alphanumeric_%s' % illegal_non_alphanumeric,
  547. 'ends_with_whitespace_%s' % whitespace,
  548. 'with', # a Python keyword
  549. 'related_name\n',
  550. ]
  551. # Python 2 crashes on non-ASCII strings.
  552. if six.PY3:
  553. invalid_related_names.append(',')
  554. class Parent(models.Model):
  555. pass
  556. for invalid_related_name in invalid_related_names:
  557. Child = type(str('Child_%s') % str(invalid_related_name), (models.Model,), {
  558. 'parent': models.ForeignKey('Parent', models.CASCADE, related_name=invalid_related_name),
  559. '__module__': Parent.__module__,
  560. })
  561. field = Child._meta.get_field('parent')
  562. errors = Child.check()
  563. expected = [
  564. Error(
  565. "The name '%s' is invalid related_name for field Child_%s.parent"
  566. % (invalid_related_name, invalid_related_name),
  567. hint="Related name must be a valid Python identifier or end with a '+'",
  568. obj=field,
  569. id='fields.E306',
  570. ),
  571. ]
  572. self.assertEqual(errors, expected)
  573. def test_related_field_has_valid_related_name(self):
  574. lowercase = 'a'
  575. uppercase = 'A'
  576. digit = 0
  577. related_names = [
  578. '%s_starts_with_lowercase' % lowercase,
  579. '%s_tarts_with_uppercase' % uppercase,
  580. '_starts_with_underscore',
  581. 'contains_%s_digit' % digit,
  582. 'ends_with_plus+',
  583. '_',
  584. '_+',
  585. '+',
  586. ]
  587. # Python 2 crashes on non-ASCII strings.
  588. if six.PY3:
  589. related_names.extend(['試', '試驗+'])
  590. class Parent(models.Model):
  591. pass
  592. for related_name in related_names:
  593. Child = type(str('Child_%s') % str(related_name), (models.Model,), {
  594. 'parent': models.ForeignKey('Parent', models.CASCADE, related_name=related_name),
  595. '__module__': Parent.__module__,
  596. })
  597. errors = Child.check()
  598. self.assertFalse(errors)
  599. class AccessorClashTests(IsolatedModelsTestCase):
  600. def test_fk_to_integer(self):
  601. self._test_accessor_clash(
  602. target=models.IntegerField(),
  603. relative=models.ForeignKey('Target', models.CASCADE))
  604. def test_fk_to_fk(self):
  605. self._test_accessor_clash(
  606. target=models.ForeignKey('Another', models.CASCADE),
  607. relative=models.ForeignKey('Target', models.CASCADE))
  608. def test_fk_to_m2m(self):
  609. self._test_accessor_clash(
  610. target=models.ManyToManyField('Another'),
  611. relative=models.ForeignKey('Target', models.CASCADE))
  612. def test_m2m_to_integer(self):
  613. self._test_accessor_clash(
  614. target=models.IntegerField(),
  615. relative=models.ManyToManyField('Target'))
  616. def test_m2m_to_fk(self):
  617. self._test_accessor_clash(
  618. target=models.ForeignKey('Another', models.CASCADE),
  619. relative=models.ManyToManyField('Target'))
  620. def test_m2m_to_m2m(self):
  621. self._test_accessor_clash(
  622. target=models.ManyToManyField('Another'),
  623. relative=models.ManyToManyField('Target'))
  624. def _test_accessor_clash(self, target, relative):
  625. class Another(models.Model):
  626. pass
  627. class Target(models.Model):
  628. model_set = target
  629. class Model(models.Model):
  630. rel = relative
  631. errors = Model.check()
  632. expected = [
  633. Error(
  634. "Reverse accessor for 'Model.rel' clashes with field name 'Target.model_set'.",
  635. hint=("Rename field 'Target.model_set', or add/change "
  636. "a related_name argument to the definition "
  637. "for field 'Model.rel'."),
  638. obj=Model._meta.get_field('rel'),
  639. id='fields.E302',
  640. ),
  641. ]
  642. self.assertEqual(errors, expected)
  643. def test_clash_between_accessors(self):
  644. class Target(models.Model):
  645. pass
  646. class Model(models.Model):
  647. foreign = models.ForeignKey(Target, models.CASCADE)
  648. m2m = models.ManyToManyField(Target)
  649. errors = Model.check()
  650. expected = [
  651. Error(
  652. "Reverse accessor for 'Model.foreign' clashes with reverse accessor for 'Model.m2m'.",
  653. hint=("Add or change a related_name argument to the definition "
  654. "for 'Model.foreign' or 'Model.m2m'."),
  655. obj=Model._meta.get_field('foreign'),
  656. id='fields.E304',
  657. ),
  658. Error(
  659. "Reverse accessor for 'Model.m2m' clashes with reverse accessor for 'Model.foreign'.",
  660. hint=("Add or change a related_name argument to the definition "
  661. "for 'Model.m2m' or 'Model.foreign'."),
  662. obj=Model._meta.get_field('m2m'),
  663. id='fields.E304',
  664. ),
  665. ]
  666. self.assertEqual(errors, expected)
  667. def test_m2m_to_m2m_with_inheritance(self):
  668. """ Ref #22047. """
  669. class Target(models.Model):
  670. pass
  671. class Model(models.Model):
  672. children = models.ManyToManyField('Child',
  673. related_name="m2m_clash", related_query_name="no_clash")
  674. class Parent(models.Model):
  675. m2m_clash = models.ManyToManyField('Target')
  676. class Child(Parent):
  677. pass
  678. errors = Model.check()
  679. expected = [
  680. Error(
  681. "Reverse accessor for 'Model.children' clashes with field name 'Child.m2m_clash'.",
  682. hint=("Rename field 'Child.m2m_clash', or add/change "
  683. "a related_name argument to the definition "
  684. "for field 'Model.children'."),
  685. obj=Model._meta.get_field('children'),
  686. id='fields.E302',
  687. )
  688. ]
  689. self.assertEqual(errors, expected)
  690. class ReverseQueryNameClashTests(IsolatedModelsTestCase):
  691. def test_fk_to_integer(self):
  692. self._test_reverse_query_name_clash(
  693. target=models.IntegerField(),
  694. relative=models.ForeignKey('Target', models.CASCADE))
  695. def test_fk_to_fk(self):
  696. self._test_reverse_query_name_clash(
  697. target=models.ForeignKey('Another', models.CASCADE),
  698. relative=models.ForeignKey('Target', models.CASCADE))
  699. def test_fk_to_m2m(self):
  700. self._test_reverse_query_name_clash(
  701. target=models.ManyToManyField('Another'),
  702. relative=models.ForeignKey('Target', models.CASCADE))
  703. def test_m2m_to_integer(self):
  704. self._test_reverse_query_name_clash(
  705. target=models.IntegerField(),
  706. relative=models.ManyToManyField('Target'))
  707. def test_m2m_to_fk(self):
  708. self._test_reverse_query_name_clash(
  709. target=models.ForeignKey('Another', models.CASCADE),
  710. relative=models.ManyToManyField('Target'))
  711. def test_m2m_to_m2m(self):
  712. self._test_reverse_query_name_clash(
  713. target=models.ManyToManyField('Another'),
  714. relative=models.ManyToManyField('Target'))
  715. def _test_reverse_query_name_clash(self, target, relative):
  716. class Another(models.Model):
  717. pass
  718. class Target(models.Model):
  719. model = target
  720. class Model(models.Model):
  721. rel = relative
  722. errors = Model.check()
  723. expected = [
  724. Error(
  725. "Reverse query name for 'Model.rel' clashes with field name 'Target.model'.",
  726. hint=("Rename field 'Target.model', or add/change "
  727. "a related_name argument to the definition "
  728. "for field 'Model.rel'."),
  729. obj=Model._meta.get_field('rel'),
  730. id='fields.E303',
  731. ),
  732. ]
  733. self.assertEqual(errors, expected)
  734. class ExplicitRelatedNameClashTests(IsolatedModelsTestCase):
  735. def test_fk_to_integer(self):
  736. self._test_explicit_related_name_clash(
  737. target=models.IntegerField(),
  738. relative=models.ForeignKey('Target', models.CASCADE, related_name='clash'))
  739. def test_fk_to_fk(self):
  740. self._test_explicit_related_name_clash(
  741. target=models.ForeignKey('Another', models.CASCADE),
  742. relative=models.ForeignKey('Target', models.CASCADE, related_name='clash'))
  743. def test_fk_to_m2m(self):
  744. self._test_explicit_related_name_clash(
  745. target=models.ManyToManyField('Another'),
  746. relative=models.ForeignKey('Target', models.CASCADE, related_name='clash'))
  747. def test_m2m_to_integer(self):
  748. self._test_explicit_related_name_clash(
  749. target=models.IntegerField(),
  750. relative=models.ManyToManyField('Target', related_name='clash'))
  751. def test_m2m_to_fk(self):
  752. self._test_explicit_related_name_clash(
  753. target=models.ForeignKey('Another', models.CASCADE),
  754. relative=models.ManyToManyField('Target', related_name='clash'))
  755. def test_m2m_to_m2m(self):
  756. self._test_explicit_related_name_clash(
  757. target=models.ManyToManyField('Another'),
  758. relative=models.ManyToManyField('Target', related_name='clash'))
  759. def _test_explicit_related_name_clash(self, target, relative):
  760. class Another(models.Model):
  761. pass
  762. class Target(models.Model):
  763. clash = target
  764. class Model(models.Model):
  765. rel = relative
  766. errors = Model.check()
  767. expected = [
  768. Error(
  769. "Reverse accessor for 'Model.rel' clashes with field name 'Target.clash'.",
  770. hint=("Rename field 'Target.clash', or add/change "
  771. "a related_name argument to the definition "
  772. "for field 'Model.rel'."),
  773. obj=Model._meta.get_field('rel'),
  774. id='fields.E302',
  775. ),
  776. Error(
  777. "Reverse query name for 'Model.rel' clashes with field name 'Target.clash'.",
  778. hint=("Rename field 'Target.clash', or add/change "
  779. "a related_name argument to the definition "
  780. "for field 'Model.rel'."),
  781. obj=Model._meta.get_field('rel'),
  782. id='fields.E303',
  783. ),
  784. ]
  785. self.assertEqual(errors, expected)
  786. class ExplicitRelatedQueryNameClashTests(IsolatedModelsTestCase):
  787. def test_fk_to_integer(self):
  788. self._test_explicit_related_query_name_clash(
  789. target=models.IntegerField(),
  790. relative=models.ForeignKey('Target',
  791. models.CASCADE,
  792. related_query_name='clash'))
  793. def test_fk_to_fk(self):
  794. self._test_explicit_related_query_name_clash(
  795. target=models.ForeignKey('Another', models.CASCADE),
  796. relative=models.ForeignKey('Target',
  797. models.CASCADE,
  798. related_query_name='clash'))
  799. def test_fk_to_m2m(self):
  800. self._test_explicit_related_query_name_clash(
  801. target=models.ManyToManyField('Another'),
  802. relative=models.ForeignKey('Target',
  803. models.CASCADE,
  804. related_query_name='clash'))
  805. def test_m2m_to_integer(self):
  806. self._test_explicit_related_query_name_clash(
  807. target=models.IntegerField(),
  808. relative=models.ManyToManyField('Target',
  809. related_query_name='clash'))
  810. def test_m2m_to_fk(self):
  811. self._test_explicit_related_query_name_clash(
  812. target=models.ForeignKey('Another', models.CASCADE),
  813. relative=models.ManyToManyField('Target',
  814. related_query_name='clash'))
  815. def test_m2m_to_m2m(self):
  816. self._test_explicit_related_query_name_clash(
  817. target=models.ManyToManyField('Another'),
  818. relative=models.ManyToManyField('Target',
  819. related_query_name='clash'))
  820. def _test_explicit_related_query_name_clash(self, target, relative):
  821. class Another(models.Model):
  822. pass
  823. class Target(models.Model):
  824. clash = target
  825. class Model(models.Model):
  826. rel = relative
  827. errors = Model.check()
  828. expected = [
  829. Error(
  830. "Reverse query name for 'Model.rel' clashes with field name 'Target.clash'.",
  831. hint=("Rename field 'Target.clash', or add/change a related_name "
  832. "argument to the definition for field 'Model.rel'."),
  833. obj=Model._meta.get_field('rel'),
  834. id='fields.E303',
  835. ),
  836. ]
  837. self.assertEqual(errors, expected)
  838. class SelfReferentialM2MClashTests(IsolatedModelsTestCase):
  839. def test_clash_between_accessors(self):
  840. class Model(models.Model):
  841. first_m2m = models.ManyToManyField('self', symmetrical=False)
  842. second_m2m = models.ManyToManyField('self', symmetrical=False)
  843. errors = Model.check()
  844. expected = [
  845. Error(
  846. "Reverse accessor for 'Model.first_m2m' clashes with reverse accessor for 'Model.second_m2m'.",
  847. hint=("Add or change a related_name argument to the definition "
  848. "for 'Model.first_m2m' or 'Model.second_m2m'."),
  849. obj=Model._meta.get_field('first_m2m'),
  850. id='fields.E304',
  851. ),
  852. Error(
  853. "Reverse accessor for 'Model.second_m2m' clashes with reverse accessor for 'Model.first_m2m'.",
  854. hint=("Add or change a related_name argument to the definition "
  855. "for 'Model.second_m2m' or 'Model.first_m2m'."),
  856. obj=Model._meta.get_field('second_m2m'),
  857. id='fields.E304',
  858. ),
  859. ]
  860. self.assertEqual(errors, expected)
  861. def test_accessor_clash(self):
  862. class Model(models.Model):
  863. model_set = models.ManyToManyField("self", symmetrical=False)
  864. errors = Model.check()
  865. expected = [
  866. Error(
  867. "Reverse accessor for 'Model.model_set' clashes with field name 'Model.model_set'.",
  868. hint=("Rename field 'Model.model_set', or add/change "
  869. "a related_name argument to the definition "
  870. "for field 'Model.model_set'."),
  871. obj=Model._meta.get_field('model_set'),
  872. id='fields.E302',
  873. ),
  874. ]
  875. self.assertEqual(errors, expected)
  876. def test_reverse_query_name_clash(self):
  877. class Model(models.Model):
  878. model = models.ManyToManyField("self", symmetrical=False)
  879. errors = Model.check()
  880. expected = [
  881. Error(
  882. "Reverse query name for 'Model.model' clashes with field name 'Model.model'.",
  883. hint=("Rename field 'Model.model', or add/change a related_name "
  884. "argument to the definition for field 'Model.model'."),
  885. obj=Model._meta.get_field('model'),
  886. id='fields.E303',
  887. ),
  888. ]
  889. self.assertEqual(errors, expected)
  890. def test_clash_under_explicit_related_name(self):
  891. class Model(models.Model):
  892. clash = models.IntegerField()
  893. m2m = models.ManyToManyField("self",
  894. symmetrical=False, related_name='clash')
  895. errors = Model.check()
  896. expected = [
  897. Error(
  898. "Reverse accessor for 'Model.m2m' clashes with field name 'Model.clash'.",
  899. hint=("Rename field 'Model.clash', or add/change a related_name "
  900. "argument to the definition for field 'Model.m2m'."),
  901. obj=Model._meta.get_field('m2m'),
  902. id='fields.E302',
  903. ),
  904. Error(
  905. "Reverse query name for 'Model.m2m' clashes with field name 'Model.clash'.",
  906. hint=("Rename field 'Model.clash', or add/change a related_name "
  907. "argument to the definition for field 'Model.m2m'."),
  908. obj=Model._meta.get_field('m2m'),
  909. id='fields.E303',
  910. ),
  911. ]
  912. self.assertEqual(errors, expected)
  913. def test_valid_model(self):
  914. class Model(models.Model):
  915. first = models.ManyToManyField("self",
  916. symmetrical=False, related_name='first_accessor')
  917. second = models.ManyToManyField("self",
  918. symmetrical=False, related_name='second_accessor')
  919. errors = Model.check()
  920. self.assertEqual(errors, [])
  921. class SelfReferentialFKClashTests(IsolatedModelsTestCase):
  922. def test_accessor_clash(self):
  923. class Model(models.Model):
  924. model_set = models.ForeignKey("Model", models.CASCADE)
  925. errors = Model.check()
  926. expected = [
  927. Error(
  928. "Reverse accessor for 'Model.model_set' clashes with field name 'Model.model_set'.",
  929. hint=("Rename field 'Model.model_set', or add/change "
  930. "a related_name argument to the definition "
  931. "for field 'Model.model_set'."),
  932. obj=Model._meta.get_field('model_set'),
  933. id='fields.E302',
  934. ),
  935. ]
  936. self.assertEqual(errors, expected)
  937. def test_reverse_query_name_clash(self):
  938. class Model(models.Model):
  939. model = models.ForeignKey("Model", models.CASCADE)
  940. errors = Model.check()
  941. expected = [
  942. Error(
  943. "Reverse query name for 'Model.model' clashes with field name 'Model.model'.",
  944. hint=("Rename field 'Model.model', or add/change "
  945. "a related_name argument to the definition "
  946. "for field 'Model.model'."),
  947. obj=Model._meta.get_field('model'),
  948. id='fields.E303',
  949. ),
  950. ]
  951. self.assertEqual(errors, expected)
  952. def test_clash_under_explicit_related_name(self):
  953. class Model(models.Model):
  954. clash = models.CharField(max_length=10)
  955. foreign = models.ForeignKey("Model", models.CASCADE, related_name='clash')
  956. errors = Model.check()
  957. expected = [
  958. Error(
  959. "Reverse accessor for 'Model.foreign' clashes with field name 'Model.clash'.",
  960. hint=("Rename field 'Model.clash', or add/change "
  961. "a related_name argument to the definition "
  962. "for field 'Model.foreign'."),
  963. obj=Model._meta.get_field('foreign'),
  964. id='fields.E302',
  965. ),
  966. Error(
  967. "Reverse query name for 'Model.foreign' clashes with field name 'Model.clash'.",
  968. hint=("Rename field 'Model.clash', or add/change "
  969. "a related_name argument to the definition "
  970. "for field 'Model.foreign'."),
  971. obj=Model._meta.get_field('foreign'),
  972. id='fields.E303',
  973. ),
  974. ]
  975. self.assertEqual(errors, expected)
  976. class ComplexClashTests(IsolatedModelsTestCase):
  977. # New tests should not be included here, because this is a single,
  978. # self-contained sanity check, not a test of everything.
  979. def test_complex_clash(self):
  980. class Target(models.Model):
  981. tgt_safe = models.CharField(max_length=10)
  982. clash = models.CharField(max_length=10)
  983. model = models.CharField(max_length=10)
  984. clash1_set = models.CharField(max_length=10)
  985. class Model(models.Model):
  986. src_safe = models.CharField(max_length=10)
  987. foreign_1 = models.ForeignKey(Target, models.CASCADE, related_name='id')
  988. foreign_2 = models.ForeignKey(Target, models.CASCADE, related_name='src_safe')
  989. m2m_1 = models.ManyToManyField(Target, related_name='id')
  990. m2m_2 = models.ManyToManyField(Target, related_name='src_safe')
  991. errors = Model.check()
  992. expected = [
  993. Error(
  994. "Reverse accessor for 'Model.foreign_1' clashes with field name 'Target.id'.",
  995. hint=("Rename field 'Target.id', or add/change a related_name "
  996. "argument to the definition for field 'Model.foreign_1'."),
  997. obj=Model._meta.get_field('foreign_1'),
  998. id='fields.E302',
  999. ),
  1000. Error(
  1001. "Reverse query name for 'Model.foreign_1' clashes with field name 'Target.id'.",
  1002. hint=("Rename field 'Target.id', or add/change a related_name "
  1003. "argument to the definition for field 'Model.foreign_1'."),
  1004. obj=Model._meta.get_field('foreign_1'),
  1005. id='fields.E303',
  1006. ),
  1007. Error(
  1008. "Reverse accessor for 'Model.foreign_1' clashes with reverse accessor for 'Model.m2m_1'.",
  1009. hint=("Add or change a related_name argument to "
  1010. "the definition for 'Model.foreign_1' or 'Model.m2m_1'."),
  1011. obj=Model._meta.get_field('foreign_1'),
  1012. id='fields.E304',
  1013. ),
  1014. Error(
  1015. "Reverse query name for 'Model.foreign_1' clashes with reverse query name for 'Model.m2m_1'.",
  1016. hint=("Add or change a related_name argument to "
  1017. "the definition for 'Model.foreign_1' or 'Model.m2m_1'."),
  1018. obj=Model._meta.get_field('foreign_1'),
  1019. id='fields.E305',
  1020. ),
  1021. Error(
  1022. "Reverse accessor for 'Model.foreign_2' clashes with reverse accessor for 'Model.m2m_2'.",
  1023. hint=("Add or change a related_name argument "
  1024. "to the definition for 'Model.foreign_2' or 'Model.m2m_2'."),
  1025. obj=Model._meta.get_field('foreign_2'),
  1026. id='fields.E304',
  1027. ),
  1028. Error(
  1029. "Reverse query name for 'Model.foreign_2' clashes with reverse query name for 'Model.m2m_2'.",
  1030. hint=("Add or change a related_name argument to "
  1031. "the definition for 'Model.foreign_2' or 'Model.m2m_2'."),
  1032. obj=Model._meta.get_field('foreign_2'),
  1033. id='fields.E305',
  1034. ),
  1035. Error(
  1036. "Reverse accessor for 'Model.m2m_1' clashes with field name 'Target.id'.",
  1037. hint=("Rename field 'Target.id', or add/change a related_name "
  1038. "argument to the definition for field 'Model.m2m_1'."),
  1039. obj=Model._meta.get_field('m2m_1'),
  1040. id='fields.E302',
  1041. ),
  1042. Error(
  1043. "Reverse query name for 'Model.m2m_1' clashes with field name 'Target.id'.",
  1044. hint=("Rename field 'Target.id', or add/change a related_name "
  1045. "argument to the definition for field 'Model.m2m_1'."),
  1046. obj=Model._meta.get_field('m2m_1'),
  1047. id='fields.E303',
  1048. ),
  1049. Error(
  1050. "Reverse accessor for 'Model.m2m_1' clashes with reverse accessor for 'Model.foreign_1'.",
  1051. hint=("Add or change a related_name argument to the definition "
  1052. "for 'Model.m2m_1' or 'Model.foreign_1'."),
  1053. obj=Model._meta.get_field('m2m_1'),
  1054. id='fields.E304',
  1055. ),
  1056. Error(
  1057. "Reverse query name for 'Model.m2m_1' clashes with reverse query name for 'Model.foreign_1'.",
  1058. hint=("Add or change a related_name argument to "
  1059. "the definition for 'Model.m2m_1' or 'Model.foreign_1'."),
  1060. obj=Model._meta.get_field('m2m_1'),
  1061. id='fields.E305',
  1062. ),
  1063. Error(
  1064. "Reverse accessor for 'Model.m2m_2' clashes with reverse accessor for 'Model.foreign_2'.",
  1065. hint=("Add or change a related_name argument to the definition "
  1066. "for 'Model.m2m_2' or 'Model.foreign_2'."),
  1067. obj=Model._meta.get_field('m2m_2'),
  1068. id='fields.E304',
  1069. ),
  1070. Error(
  1071. "Reverse query name for 'Model.m2m_2' clashes with reverse query name for 'Model.foreign_2'.",
  1072. hint=("Add or change a related_name argument to the definition "
  1073. "for 'Model.m2m_2' or 'Model.foreign_2'."),
  1074. obj=Model._meta.get_field('m2m_2'),
  1075. id='fields.E305',
  1076. ),
  1077. ]
  1078. self.assertEqual(errors, expected)
  1079. class M2mThroughFieldsTests(IsolatedModelsTestCase):
  1080. def test_m2m_field_argument_validation(self):
  1081. """
  1082. Tests that ManyToManyField accepts the ``through_fields`` kwarg
  1083. only if an intermediary table is specified.
  1084. """
  1085. class Fan(models.Model):
  1086. pass
  1087. self.assertRaisesMessage(
  1088. ValueError, 'Cannot specify through_fields without a through model',
  1089. models.ManyToManyField, Fan, through_fields=('f1', 'f2'))
  1090. def test_invalid_order(self):
  1091. """
  1092. Tests that mixing up the order of link fields to ManyToManyField.through_fields
  1093. triggers validation errors.
  1094. """
  1095. class Fan(models.Model):
  1096. pass
  1097. class Event(models.Model):
  1098. invitees = models.ManyToManyField(Fan, through='Invitation', through_fields=('invitee', 'event'))
  1099. class Invitation(models.Model):
  1100. event = models.ForeignKey(Event, models.CASCADE)
  1101. invitee = models.ForeignKey(Fan, models.CASCADE)
  1102. inviter = models.ForeignKey(Fan, models.CASCADE, related_name='+')
  1103. field = Event._meta.get_field('invitees')
  1104. errors = field.check(from_model=Event)
  1105. expected = [
  1106. Error(
  1107. ("'Invitation.invitee' is not a foreign key to 'Event'."),
  1108. hint="Did you mean one of the following foreign keys to 'Event': event?",
  1109. obj=field,
  1110. id='fields.E339'),
  1111. Error(
  1112. ("'Invitation.event' is not a foreign key to 'Fan'."),
  1113. hint="Did you mean one of the following foreign keys to 'Fan': invitee, inviter?",
  1114. obj=field,
  1115. id='fields.E339'),
  1116. ]
  1117. self.assertEqual(expected, errors)
  1118. def test_invalid_field(self):
  1119. """
  1120. Tests that providing invalid field names to ManyToManyField.through_fields
  1121. triggers validation errors.
  1122. """
  1123. class Fan(models.Model):
  1124. pass
  1125. class Event(models.Model):
  1126. invitees = models.ManyToManyField(Fan, through='Invitation', through_fields=('invalid_field_1', 'invalid_field_2'))
  1127. class Invitation(models.Model):
  1128. event = models.ForeignKey(Event, models.CASCADE)
  1129. invitee = models.ForeignKey(Fan, models.CASCADE)
  1130. inviter = models.ForeignKey(Fan, models.CASCADE, related_name='+')
  1131. field = Event._meta.get_field('invitees')
  1132. errors = field.check(from_model=Event)
  1133. expected = [
  1134. Error(
  1135. ("The intermediary model 'invalid_models_tests.Invitation' has no field 'invalid_field_1'."),
  1136. hint="Did you mean one of the following foreign keys to 'Event': event?",
  1137. obj=field,
  1138. id='fields.E338'),
  1139. Error(
  1140. ("The intermediary model 'invalid_models_tests.Invitation' has no field 'invalid_field_2'."),
  1141. hint="Did you mean one of the following foreign keys to 'Fan': invitee, inviter?",
  1142. obj=field,
  1143. id='fields.E338'),
  1144. ]
  1145. self.assertEqual(expected, errors)
  1146. def test_explicit_field_names(self):
  1147. """
  1148. Tests that if ``through_fields`` kwarg is given, it must specify both
  1149. link fields of the intermediary table.
  1150. """
  1151. class Fan(models.Model):
  1152. pass
  1153. class Event(models.Model):
  1154. invitees = models.ManyToManyField(Fan, through='Invitation', through_fields=(None, 'invitee'))
  1155. class Invitation(models.Model):
  1156. event = models.ForeignKey(Event, models.CASCADE)
  1157. invitee = models.ForeignKey(Fan, models.CASCADE)
  1158. inviter = models.ForeignKey(Fan, models.CASCADE, related_name='+')
  1159. field = Event._meta.get_field('invitees')
  1160. errors = field.check(from_model=Event)
  1161. expected = [
  1162. Error(
  1163. ("Field specifies 'through_fields' but does not provide the names "
  1164. "of the two link fields that should be used for the relation "
  1165. "through model 'invalid_models_tests.Invitation'."),
  1166. hint=("Make sure you specify 'through_fields' as "
  1167. "through_fields=('field1', 'field2')"),
  1168. obj=field,
  1169. id='fields.E337')]
  1170. self.assertEqual(expected, errors)