def test_actions_column(self):
"""
The action column should have an "add" button inserted.
"""
from mptt.admin import FeinCMSModelAdmin
model_admin = FeinCMSModelAdmin(Category, admin.site)
category = Category.objects.get(id=1)
self.assertTrue('<a href="add/?parent=1" title="Add child">' in
model_admin._actions_column(category)[0])
def test_actions_column(self):
"""
The action column should have an "add" button inserted.
"""
from mptt.admin import FeinCMSModelAdmin
model_admin = FeinCMSModelAdmin(Category, admin.site)
category = Category.objects.get(id=1)
self.assertTrue(
'<a href="add/?parent=1" title="Add child">' in
model_admin._actions_column(category)[0]
)
class FeinCMSModelAdminTestCase(TestCase):
"""
Tests for FeinCMSModelAdmin.
"""
fixtures = ['categories.json']
def setUp(self):
from mptt.admin import FeinCMSModelAdmin
self.model_admin = FeinCMSModelAdmin(Category, admin.site)
def test_actions_column(self):
"""
The action column should have an "add" button inserted.
"""
# See implementation notes.
if django.VERSION < (1, 4):
prefix = '/static/admin/img/admin/'
else:
prefix = '/static/admin/img/'
category = Category.objects.get(id=1)
self.assertEqual(self.model_admin._actions_column(category), [
u'<a href="add/?parent=1" title="Add child">'
u'<img src="%sicon_addlink.gif" alt="Add child" /></a>' % prefix,
'<div class="drag_handle"></div>'])
def setUp(self):
from mptt.admin import FeinCMSModelAdmin
self.model_admin = FeinCMSModelAdmin(Category, admin.site)
