88 lines
3.0 KiB
Python
88 lines
3.0 KiB
Python
from __future__ import unicode_literals
|
|
|
|
from django.contrib.sites.models import Site
|
|
from django.db import models, connection
|
|
from django.db.backends.utils import truncate_name
|
|
from django.test import TestCase
|
|
|
|
from .models.publication import Publication
|
|
from .models.article import Article
|
|
|
|
|
|
class Advertisement(models.Model):
|
|
customer = models.CharField(max_length=100)
|
|
publications = models.ManyToManyField("model_package.Publication", blank=True)
|
|
|
|
|
|
class ModelPackageTests(TestCase):
|
|
|
|
@classmethod
|
|
def setUpClass(cls):
|
|
super(ModelPackageTests, cls).setUpClass()
|
|
# This cleanup is necessary because contrib.sites cache
|
|
# makes tests interfere with each other, see #11505
|
|
Site.objects.clear_cache()
|
|
|
|
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")
|
|
|
|
current_site = Site.objects.get_current()
|
|
self.assertEqual(current_site.domain, "example.com")
|
|
|
|
a = Article.objects.create(headline="a foo headline")
|
|
a.publications.add(p)
|
|
a.sites.add(current_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'
|
|
)
|