class PollAdmin(easy.MixinEasyViews, admin.ModelAdmin):
list_display = ('name', 'count_question')
count_question = easy.LinkChangeListAdminField('test_app', 'question', 'question_set.count', {'poll': 'id'}, 'Count')
def easy_view_test(self, request, *args):
return HttpResponse('test is ok with %s' % (args or 'list'))
def test_link(self):
poll = mommy.make(Poll, )
custom_field = easy.LinkChangeListAdminField('test_app', 'question',
'question_set.count',
{'pool': 'id'})
ret = custom_field(poll)
expected = u'<a href="/admin/test_app/question/?pool=1">0</a>'
self.assertEqual(expected, ret)
self.assertTrue(custom_field.allow_tags)
def test_link(self):
poll = baker.make(Poll, )
custom_field = easy.LinkChangeListAdminField('test_app', 'question',
'question_set.count',
{'pool': 'id'},
{'static': 1})
ret = custom_field(poll)
q = urlencode({'pool': poll.id, 'static': 1})
expected = u'<a href="/admin/test_app/question/?' + q + '">0</a>'
self.assertEqual(expected, ret)
self.assertTrue(custom_field.allow_tags)