示例#1
0
from django.contrib.admin import ModelAdmin
from django.contrib.admin.templatetags.admin_list import result_list
from suit.templatetags.suit_list import paginator_number, paginator_info, \
    pagination, suit_list_filter_select, headers_handler, dict_to_attrs, \
    result_row_attrs, cells_handler
from suit.tests.mixins import UserTestCaseMixin, ModelsTestCaseMixin
from suit.tests.models import Album, Book, test_app_label

try:
    from django.core.urlresolvers import reverse
except ImportError:
    # For Django >= 2.0
    from django.urls import reverse

app_label = test_app_label()


class ModelAdminMock(object):
    def suit_row_attributes(self, obj):
        return {'class': obj.name, 'data': obj.pk}

    def suit_cell_attributes(self, obj, column):
        return {'class': 'col-' + column, 'data': obj.pk}


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')]

示例#2
0
from django.contrib.admin.templatetags.admin_list import result_list
from django.core.urlresolvers import reverse
from suit.templatetags.suit_list import (
    paginator_number,
    paginator_info,
    pagination,
    suit_list_filter_select,
    headers_handler,
    dict_to_attrs,
    result_row_attrs,
    cells_handler,
)
from suit.tests.mixins import UserTestCaseMixin, ModelsTestCaseMixin
from suit.tests.models import Album, Book, test_app_label

app_label = test_app_label()


class ModelAdminMock(object):
    def suit_row_attributes(self, obj):
        return {"class": obj.name, "data": obj.pk}

    def suit_cell_attributes(self, obj, column):
        return {"class": "col-" + column, "data": obj.pk}


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")]