summaryrefslogtreecommitdiff
path: root/tests/admin_utils
diff options
context:
space:
mode:
authorTim Graham <timograham@gmail.com>2014-12-31 11:25:11 -0500
committerTim Graham <timograham@gmail.com>2014-12-31 11:26:27 -0500
commit8a9b0c15a6c0ef60dea3ba3042317520bc201206 (patch)
treea018a2b6016fe1cae7d8e4facb66b5e7ac5800bc /tests/admin_utils
parent837fc2d8cdfefce375697d95e241836c7be12696 (diff)
Renamed tests for util -> utils moves; refs #17627.
Diffstat (limited to 'tests/admin_utils')
-rw-r--r--tests/admin_utils/__init__.py0
-rw-r--r--tests/admin_utils/models.py72
-rw-r--r--tests/admin_utils/tests.py353
3 files changed, 425 insertions, 0 deletions
diff --git a/tests/admin_utils/__init__.py b/tests/admin_utils/__init__.py
new file mode 100644
index 0000000000..e69de29bb2
--- /dev/null
+++ b/tests/admin_utils/__init__.py
diff --git a/tests/admin_utils/models.py b/tests/admin_utils/models.py
new file mode 100644
index 0000000000..ce845c64eb
--- /dev/null
+++ b/tests/admin_utils/models.py
@@ -0,0 +1,72 @@
+from django.db import models
+from django.utils import six
+from django.utils.encoding import python_2_unicode_compatible
+
+
+@python_2_unicode_compatible
+class Site(models.Model):
+ domain = models.CharField(max_length=100)
+
+ def __str__(self):
+ return self.domain
+
+
+class Article(models.Model):
+ """
+ A simple Article model for testing
+ """
+ site = models.ForeignKey(Site, related_name="admin_articles")
+ title = models.CharField(max_length=100)
+ title2 = models.CharField(max_length=100, verbose_name="another name")
+ created = models.DateTimeField()
+
+ def test_from_model(self):
+ return "nothing"
+
+ def test_from_model_with_override(self):
+ return "nothing"
+ test_from_model_with_override.short_description = "not What you Expect"
+
+
+@python_2_unicode_compatible
+class Count(models.Model):
+ num = models.PositiveSmallIntegerField()
+ parent = models.ForeignKey('self', null=True)
+
+ def __str__(self):
+ return six.text_type(self.num)
+
+
+class Event(models.Model):
+ date = models.DateTimeField(auto_now_add=True)
+
+
+class Location(models.Model):
+ event = models.OneToOneField(Event, verbose_name='awesome event')
+
+
+class Guest(models.Model):
+ event = models.OneToOneField(Event)
+ name = models.CharField(max_length=255)
+
+ class Meta:
+ verbose_name = "awesome guest"
+
+
+class EventGuide(models.Model):
+ event = models.ForeignKey(Event, on_delete=models.DO_NOTHING)
+
+
+class Vehicle(models.Model):
+ pass
+
+
+class VehicleMixin(Vehicle):
+ vehicle = models.OneToOneField(Vehicle, parent_link=True, related_name='vehicle_%(app_label)s_%(class)s')
+
+ class Meta:
+ abstract = True
+
+
+class Car(VehicleMixin):
+ pass
diff --git a/tests/admin_utils/tests.py b/tests/admin_utils/tests.py
new file mode 100644
index 0000000000..01710e51ce
--- /dev/null
+++ b/tests/admin_utils/tests.py
@@ -0,0 +1,353 @@
+from __future__ import unicode_literals
+
+from datetime import datetime
+
+from django.conf import settings
+from django.contrib import admin
+from django.contrib.admin import helpers
+from django.contrib.admin.utils import (display_for_field, flatten,
+ flatten_fieldsets, label_for_field, lookup_field, NestedObjects)
+from django.contrib.admin.views.main import EMPTY_CHANGELIST_VALUE
+from django.db import models, DEFAULT_DB_ALIAS
+from django import forms
+from django.test import TestCase
+from django.utils.formats import localize
+from django.utils.safestring import mark_safe
+from django.utils import six
+
+from .models import Site, Article, Count, Event, Location, EventGuide, Vehicle, Car
+
+
+class NestedObjectsTests(TestCase):
+ """
+ Tests for ``NestedObject`` utility collection.
+ """
+ def setUp(self):
+ self.n = NestedObjects(using=DEFAULT_DB_ALIAS)
+ self.objs = [Count.objects.create(num=i) for i in range(5)]
+
+ def _check(self, target):
+ self.assertEqual(self.n.nested(lambda obj: obj.num), target)
+
+ def _connect(self, i, j):
+ self.objs[i].parent = self.objs[j]
+ self.objs[i].save()
+
+ def _collect(self, *indices):
+ self.n.collect([self.objs[i] for i in indices])
+
+ def test_unrelated_roots(self):
+ self._connect(2, 1)
+ self._collect(0)
+ self._collect(1)
+ self._check([0, 1, [2]])
+
+ def test_siblings(self):
+ self._connect(1, 0)
+ self._connect(2, 0)
+ self._collect(0)
+ self._check([0, [1, 2]])
+
+ def test_non_added_parent(self):
+ self._connect(0, 1)
+ self._collect(0)
+ self._check([0])
+
+ def test_cyclic(self):
+ self._connect(0, 2)
+ self._connect(1, 0)
+ self._connect(2, 1)
+ self._collect(0)
+ self._check([0, [1, [2]]])
+
+ def test_queries(self):
+ self._connect(1, 0)
+ self._connect(2, 0)
+ # 1 query to fetch all children of 0 (1 and 2)
+ # 1 query to fetch all children of 1 and 2 (none)
+ # Should not require additional queries to populate the nested graph.
+ self.assertNumQueries(2, self._collect, 0)
+
+ def test_on_delete_do_nothing(self):
+ """
+ Check that the nested collector doesn't query for DO_NOTHING objects.
+ """
+ n = NestedObjects(using=DEFAULT_DB_ALIAS)
+ objs = [Event.objects.create()]
+ EventGuide.objects.create(event=objs[0])
+ with self.assertNumQueries(2):
+ # One for Location, one for Guest, and no query for EventGuide
+ n.collect(objs)
+
+ def test_relation_on_abstract(self):
+ """
+ #21846 -- Check that `NestedObjects.collect()` doesn't trip
+ (AttributeError) on the special notation for relations on abstract
+ models (related_name that contains %(app_label)s and/or %(class)s).
+ """
+ n = NestedObjects(using=DEFAULT_DB_ALIAS)
+ Car.objects.create()
+ n.collect([Vehicle.objects.first()])
+
+
+class UtilsTests(TestCase):
+ def test_values_from_lookup_field(self):
+ """
+ Regression test for #12654: lookup_field
+ """
+ SITE_NAME = 'example.com'
+ TITLE_TEXT = 'Some title'
+ CREATED_DATE = datetime.min
+ ADMIN_METHOD = 'admin method'
+ SIMPLE_FUNCTION = 'function'
+ INSTANCE_ATTRIBUTE = 'attr'
+
+ class MockModelAdmin(object):
+ def get_admin_value(self, obj):
+ return ADMIN_METHOD
+
+ simple_function = lambda obj: SIMPLE_FUNCTION
+
+ site_obj = Site.objects.create(domain=SITE_NAME)
+ article = Article(
+ site=site_obj,
+ title=TITLE_TEXT,
+ created=CREATED_DATE,
+ )
+ article.non_field = INSTANCE_ATTRIBUTE
+
+ verifications = (
+ ('site', SITE_NAME),
+ ('created', localize(CREATED_DATE)),
+ ('title', TITLE_TEXT),
+ ('get_admin_value', ADMIN_METHOD),
+ (simple_function, SIMPLE_FUNCTION),
+ ('test_from_model', article.test_from_model()),
+ ('non_field', INSTANCE_ATTRIBUTE)
+ )
+
+ mock_admin = MockModelAdmin()
+ for name, value in verifications:
+ field, attr, resolved_value = lookup_field(name, article, mock_admin)
+
+ if field is not None:
+ resolved_value = display_for_field(resolved_value, field)
+
+ self.assertEqual(value, resolved_value)
+
+ def test_null_display_for_field(self):
+ """
+ Regression test for #12550: display_for_field should handle None
+ value.
+ """
+ display_value = display_for_field(None, models.CharField())
+ self.assertEqual(display_value, EMPTY_CHANGELIST_VALUE)
+
+ display_value = display_for_field(None, models.CharField(
+ choices=(
+ (None, "test_none"),
+ )
+ ))
+ self.assertEqual(display_value, "test_none")
+
+ display_value = display_for_field(None, models.DateField())
+ self.assertEqual(display_value, EMPTY_CHANGELIST_VALUE)
+
+ display_value = display_for_field(None, models.TimeField())
+ self.assertEqual(display_value, EMPTY_CHANGELIST_VALUE)
+
+ # Regression test for #13071: NullBooleanField has special
+ # handling.
+ display_value = display_for_field(None, models.NullBooleanField())
+ expected = '<img src="%sadmin/img/icon-unknown.gif" alt="None" />' % settings.STATIC_URL
+ self.assertHTMLEqual(display_value, expected)
+
+ display_value = display_for_field(None, models.DecimalField())
+ self.assertEqual(display_value, EMPTY_CHANGELIST_VALUE)
+
+ display_value = display_for_field(None, models.FloatField())
+ self.assertEqual(display_value, EMPTY_CHANGELIST_VALUE)
+
+ def test_label_for_field(self):
+ """
+ Tests for label_for_field
+ """
+ self.assertEqual(
+ label_for_field("title", Article),
+ "title"
+ )
+ self.assertEqual(
+ label_for_field("title2", Article),
+ "another name"
+ )
+ self.assertEqual(
+ label_for_field("title2", Article, return_attr=True),
+ ("another name", None)
+ )
+
+ self.assertEqual(
+ label_for_field("__unicode__", Article),
+ "article"
+ )
+ self.assertEqual(
+ label_for_field("__str__", Article),
+ str("article")
+ )
+
+ self.assertRaises(
+ AttributeError,
+ lambda: label_for_field("unknown", Article)
+ )
+
+ def test_callable(obj):
+ return "nothing"
+ self.assertEqual(
+ label_for_field(test_callable, Article),
+ "Test callable"
+ )
+ self.assertEqual(
+ label_for_field(test_callable, Article, return_attr=True),
+ ("Test callable", test_callable)
+ )
+
+ self.assertEqual(
+ label_for_field("test_from_model", Article),
+ "Test from model"
+ )
+ self.assertEqual(
+ label_for_field("test_from_model", Article, return_attr=True),
+ ("Test from model", Article.test_from_model)
+ )
+ self.assertEqual(
+ label_for_field("test_from_model_with_override", Article),
+ "not What you Expect"
+ )
+
+ self.assertEqual(
+ label_for_field(lambda x: "nothing", Article),
+ "--"
+ )
+
+ class MockModelAdmin(object):
+ def test_from_model(self, obj):
+ return "nothing"
+ test_from_model.short_description = "not Really the Model"
+
+ self.assertEqual(
+ label_for_field("test_from_model", Article, model_admin=MockModelAdmin),
+ "not Really the Model"
+ )
+ self.assertEqual(
+ label_for_field("test_from_model", Article,
+ model_admin=MockModelAdmin,
+ return_attr=True),
+ ("not Really the Model", MockModelAdmin.test_from_model)
+ )
+
+ def test_label_for_property(self):
+ # NOTE: cannot use @property decorator, because of
+ # AttributeError: 'property' object has no attribute 'short_description'
+ class MockModelAdmin(object):
+ def my_property(self):
+ return "this if from property"
+ my_property.short_description = 'property short description'
+ test_from_property = property(my_property)
+
+ self.assertEqual(
+ label_for_field("test_from_property", Article, model_admin=MockModelAdmin),
+ 'property short description'
+ )
+
+ def test_related_name(self):
+ """
+ Regression test for #13963
+ """
+ self.assertEqual(
+ label_for_field('location', Event, return_attr=True),
+ ('location', None),
+ )
+ self.assertEqual(
+ label_for_field('event', Location, return_attr=True),
+ ('awesome event', None),
+ )
+ self.assertEqual(
+ label_for_field('guest', Event, return_attr=True),
+ ('awesome guest', None),
+ )
+
+ def test_logentry_unicode(self):
+ """
+ Regression test for #15661
+ """
+ log_entry = admin.models.LogEntry()
+
+ log_entry.action_flag = admin.models.ADDITION
+ self.assertTrue(
+ six.text_type(log_entry).startswith('Added ')
+ )
+
+ log_entry.action_flag = admin.models.CHANGE
+ self.assertTrue(
+ six.text_type(log_entry).startswith('Changed ')
+ )
+
+ log_entry.action_flag = admin.models.DELETION
+ self.assertTrue(
+ six.text_type(log_entry).startswith('Deleted ')
+ )
+
+ # Make sure custom action_flags works
+ log_entry.action_flag = 4
+ self.assertEqual(six.text_type(log_entry), 'LogEntry Object')
+
+ def test_safestring_in_field_label(self):
+ # safestring should not be escaped
+ class MyForm(forms.Form):
+ text = forms.CharField(label=mark_safe('<i>text</i>'))
+ cb = forms.BooleanField(label=mark_safe('<i>cb</i>'))
+
+ form = MyForm()
+ self.assertHTMLEqual(helpers.AdminField(form, 'text', is_first=False).label_tag(),
+ '<label for="id_text" class="required inline"><i>text</i>:</label>')
+ self.assertHTMLEqual(helpers.AdminField(form, 'cb', is_first=False).label_tag(),
+ '<label for="id_cb" class="vCheckboxLabel required inline"><i>cb</i></label>')
+
+ # normal strings needs to be escaped
+ class MyForm(forms.Form):
+ text = forms.CharField(label='&text')
+ cb = forms.BooleanField(label='&cb')
+
+ form = MyForm()
+ self.assertHTMLEqual(helpers.AdminField(form, 'text', is_first=False).label_tag(),
+ '<label for="id_text" class="required inline">&amp;text:</label>')
+ self.assertHTMLEqual(helpers.AdminField(form, 'cb', is_first=False).label_tag(),
+ '<label for="id_cb" class="vCheckboxLabel required inline">&amp;cb</label>')
+
+ def test_flatten(self):
+ flat_all = ['url', 'title', 'content', 'sites']
+ inputs = (
+ ((), []),
+ (('url', 'title', ('content', 'sites')), flat_all),
+ (('url', 'title', 'content', 'sites'), flat_all),
+ ((('url', 'title'), ('content', 'sites')), flat_all)
+ )
+ for orig, expected in inputs:
+ self.assertEqual(flatten(orig), expected)
+
+ def test_flatten_fieldsets(self):
+ """
+ Regression test for #18051
+ """
+ fieldsets = (
+ (None, {
+ 'fields': ('url', 'title', ('content', 'sites'))
+ }),
+ )
+ self.assertEqual(flatten_fieldsets(fieldsets), ['url', 'title', 'content', 'sites'])
+
+ fieldsets = (
+ (None, {
+ 'fields': ('url', 'title', ['content', 'sites'])
+ }),
+ )
+ self.assertEqual(flatten_fieldsets(fieldsets), ['url', 'title', 'content', 'sites'])