From 27544d5a51082ad121a53512263a8cc64b8c9854 Mon Sep 17 00:00:00 2001 From: Andrews Medina Date: Wed, 12 Jun 2013 17:08:21 -0300 Subject: [PATCH] Added more tests for get_attr templatetag. --- djadmin2/tests/test_admin2tags.py | 22 ++++++++++++++++++++++ 1 file changed, 22 insertions(+) diff --git a/djadmin2/tests/test_admin2tags.py b/djadmin2/tests/test_admin2tags.py index 41dd02c..b51bb87 100644 --- a/djadmin2/tests/test_admin2tags.py +++ b/djadmin2/tests/test_admin2tags.py @@ -74,3 +74,25 @@ class TagsTests(TestCase): admin2_tags.get_attr(Klass(), "hello"), "hello" ) + + def test_get_attr_str(self): + class Klass(object): + def __str__(self): + return "str" + + def __unicode__(self): + return "unicode" + + self.assertEquals( + admin2_tags.get_attr(Klass(), "__str__"), + "unicode" + ) + + def test_get_attr(self): + class Klass(object): + attr = "value" + + self.assertEquals( + admin2_tags.get_attr(Klass(), "attr"), + "value" + )