--- /dev/null Thu Jan 01 00:00:00 1970 +0000
+++ b/parts/django/tests/regressiontests/admin_util/tests.py Sat Jan 08 11:20:57 2011 +0530
@@ -0,0 +1,239 @@
+from datetime import datetime
+import unittest
+
+from django.conf import settings
+from django.db import models
+from django.utils.formats import localize
+from django.test import TestCase
+
+from django.contrib import admin
+from django.contrib.admin.util import display_for_field, label_for_field, lookup_field
+from django.contrib.admin.views.main import EMPTY_CHANGELIST_VALUE
+from django.contrib.sites.models import Site
+from django.contrib.admin.util import NestedObjects
+
+from models import Article, Count, Event, Location
+
+
+class NestedObjectsTests(TestCase):
+ """
+ Tests for ``NestedObject`` utility collection.
+
+ """
+ def setUp(self):
+ self.n = NestedObjects()
+ self.objs = [Count.objects.create(num=i) for i in range(5)]
+
+ def _check(self, target):
+ self.assertEquals(self.n.nested(lambda obj: obj.num), target)
+
+ def _add(self, obj, parent=None):
+ # don't bother providing the extra args that NestedObjects ignores
+ self.n.add(None, None, obj, None, parent)
+
+ def test_unrelated_roots(self):
+ self._add(self.objs[0])
+ self._add(self.objs[1])
+ self._add(self.objs[2], self.objs[1])
+
+ self._check([0, 1, [2]])
+
+ def test_siblings(self):
+ self._add(self.objs[0])
+ self._add(self.objs[1], self.objs[0])
+ self._add(self.objs[2], self.objs[0])
+
+ self._check([0, [1, 2]])
+
+ def test_duplicate_instances(self):
+ self._add(self.objs[0])
+ self._add(self.objs[1])
+ dupe = Count.objects.get(num=1)
+ self._add(dupe, self.objs[0])
+
+ self._check([0, 1])
+
+ def test_non_added_parent(self):
+ self._add(self.objs[0], self.objs[1])
+
+ self._check([0])
+
+ def test_cyclic(self):
+ self._add(self.objs[0], self.objs[2])
+ self._add(self.objs[1], self.objs[0])
+ self._add(self.objs[2], self.objs[1])
+ self._add(self.objs[0], self.objs[2])
+
+ self._check([0, [1, [2]]])
+
+
+class UtilTests(unittest.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
+
+ article = Article(
+ site=Site(domain=SITE_NAME),
+ 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 = u'<img src="%simg/admin/icon-unknown.gif" alt="None" />' % settings.ADMIN_MEDIA_PREFIX
+ self.assertEqual(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.assertEquals(
+ label_for_field("title", Article),
+ "title"
+ )
+ self.assertEquals(
+ label_for_field("title2", Article),
+ "another name"
+ )
+ self.assertEquals(
+ label_for_field("title2", Article, return_attr=True),
+ ("another name", None)
+ )
+
+ self.assertEquals(
+ label_for_field("__unicode__", Article),
+ "article"
+ )
+ self.assertEquals(
+ label_for_field("__str__", Article),
+ "article"
+ )
+
+ self.assertRaises(
+ AttributeError,
+ lambda: label_for_field("unknown", Article)
+ )
+
+ def test_callable(obj):
+ return "nothing"
+ self.assertEquals(
+ label_for_field(test_callable, Article),
+ "Test callable"
+ )
+ self.assertEquals(
+ label_for_field(test_callable, Article, return_attr=True),
+ ("Test callable", test_callable)
+ )
+
+ self.assertEquals(
+ label_for_field("test_from_model", Article),
+ "Test from model"
+ )
+ self.assertEquals(
+ label_for_field("test_from_model", Article, return_attr=True),
+ ("Test from model", Article.test_from_model)
+ )
+ self.assertEquals(
+ label_for_field("test_from_model_with_override", Article),
+ "not What you Expect"
+ )
+
+ self.assertEquals(
+ 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.assertEquals(
+ label_for_field("test_from_model", Article, model_admin=MockModelAdmin),
+ "not Really the Model"
+ )
+ self.assertEquals(
+ label_for_field("test_from_model", Article,
+ model_admin = MockModelAdmin,
+ return_attr = True
+ ),
+ ("not Really the Model", MockModelAdmin.test_from_model)
+ )
+
+ def test_related_name(self):
+ """
+ Regression test for #13963
+ """
+ self.assertEquals(
+ label_for_field('location', Event, return_attr=True),
+ ('location', None),
+ )
+ self.assertEquals(
+ label_for_field('event', Location, return_attr=True),
+ ('awesome event', None),
+ )
+ self.assertEquals(
+ label_for_field('guest', Event, return_attr=True),
+ ('awesome guest', None),
+ )