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)