def test_suit_list_result_row_attrs_by_response(self):
Book.objects.all().delete()
for x in range(2):
book = Book(pk=x, name='sky-%s' % x)
book.save()
self.get_changelist()
result = result_row_attrs(self.changelist, 1)
self.assertTrue('data="1"' in result)
self.assertTrue('class="row1 suit_row_attr_class-sky-1"' in result)
def test_pagination_many_pages(self):
per_page_original = ModelAdmin.list_per_page
ModelAdmin.list_per_page = 20
for x in range(25):
book = Book(name='Test %d' % x)
book.save()
self.get_changelist()
pg = pagination(self.changelist)
ModelAdmin.list_per_page = per_page_original
self.assertEqual(pg['cl'], self.changelist)
self.assertEqual(len(pg['page_range']), 2)
self.assertEqual(pg['pagination_required'], True)
def test_suit_list_cells_handler_by_response(self):
Book.objects.all().delete()
for x in range(2):
book = Book(pk=x, name='sky-%s' % x)
book.save()
self.get_changelist()
cl = self.changelist
results = result_list(cl)['results']
result_cells = cells_handler(results, cl)
self.assertTrue(
'class="suit_cell_attr_class-name-sky-1' in result_cells[0][-1])
self.assertTrue(' data="1"' in result_cells[0][-1])
def setUp(self):
self.login_superuser()
self.book = Book(name='Test')
self.book.save()
self.get_changelist()
class ChangeListMock(object):
list_display = ('action_checkbox', 'name', 'order', 'status')
model_admin = ModelAdminMock()
result_list = [Book(pk=1, name='beach'), Book(pk=2, name='sky')]
def create_book(self):
book = Book(pk=2, name='Some book')
book.save()
return book