1
0
mirror of https://github.com/django/django.git synced 2024-12-23 01:25:58 +00:00
django/tests/model_package/tests.py
2022-02-07 20:37:05 +01:00

75 lines
2.6 KiB
Python

from django.db import connection, models
from django.db.backends.utils import truncate_name
from django.test import TestCase
from .models.article import Article, Site
from .models.publication import Publication
class Advertisement(models.Model):
customer = models.CharField(max_length=100)
publications = models.ManyToManyField("model_package.Publication", blank=True)
class ModelPackageTests(TestCase):
def test_m2m_tables_in_subpackage_models(self):
"""
Regression for #12168: models split into subpackages still get M2M
tables.
"""
p = Publication.objects.create(title="FooBar")
site = Site.objects.create(name="example.com")
a = Article.objects.create(headline="a foo headline")
a.publications.add(p)
a.sites.add(site)
a = Article.objects.get(id=a.pk)
self.assertEqual(a.id, a.pk)
self.assertEqual(a.sites.count(), 1)
def test_models_in_the_test_package(self):
"""
Regression for #12245 - Models can exist in the test package, too.
"""
p = Publication.objects.create(title="FooBar")
ad = Advertisement.objects.create(customer="Lawrence Journal-World")
ad.publications.add(p)
ad = Advertisement.objects.get(id=ad.pk)
self.assertEqual(ad.publications.count(), 1)
def test_automatic_m2m_column_names(self):
"""
Regression for #12386 - field names on the autogenerated intermediate
class that are specified as dotted strings don't retain any path
component for the field or column name.
"""
self.assertEqual(Article.publications.through._meta.fields[1].name, "article")
self.assertEqual(
Article.publications.through._meta.fields[1].get_attname_column(),
("article_id", "article_id"),
)
self.assertEqual(
Article.publications.through._meta.fields[2].name, "publication"
)
self.assertEqual(
Article.publications.through._meta.fields[2].get_attname_column(),
("publication_id", "publication_id"),
)
self.assertEqual(
Article._meta.get_field("publications").m2m_db_table(),
truncate_name(
"model_package_article_publications", connection.ops.max_name_length()
),
)
self.assertEqual(
Article._meta.get_field("publications").m2m_column_name(), "article_id"
)
self.assertEqual(
Article._meta.get_field("publications").m2m_reverse_name(), "publication_id"
)