mirror of
https://github.com/django/django.git
synced 2024-12-25 02:26:12 +00:00
267346678d
git-svn-id: http://code.djangoproject.com/svn/django/trunk@12366 bcc190cf-cafb-0310-a4f2-bffc1f526a37
165 lines
5.1 KiB
Python
165 lines
5.1 KiB
Python
from datetime import datetime
|
|
import unittest
|
|
|
|
from django.db import models
|
|
|
|
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.utils.formats import localize
|
|
|
|
from models import Article
|
|
|
|
|
|
|
|
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)
|
|
|
|
display_value = display_for_field(None, models.NullBooleanField())
|
|
self.assertEqual(display_value, EMPTY_CHANGELIST_VALUE)
|
|
|
|
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)
|
|
)
|