python - 自定义 django 管理模板的单元测试

Question

不知道如何处理这种情况。 最近开始使用 django 进行单元测试。 在我的项目中，我有一个自定义的 change_list.html 来在管理页面上添加一个按钮。

我正在寻找可以验证此自定义按钮的单元测试。 这是我在 admin.py 中的代码:

class LocationAdmin(OSMGeoAdmin):

    def get_urls(self):
        urls = patterns('', url(
            r'^import/$',
            self.admin_site.admin_view(self.import_process)
        ))

        super_urls = super(LocationAdmin, self).get_urls()

        return urls + super_urls

    def import_process(self, request):
        pass

admin.site.register(Location, LocationAdmin)

此代码自动加载 app/admin/app/app/change_list.html 中的模板 这是:

{% extends "admin/change_list.html" %}
{% load staticfiles %}
{% block extrahead %}
    <link rel="stylesheet" href="{% static "css/custom_upload.css" %}">
{% endblock %}
{% block object-tools-items %}
    {{ block.super }}
    <li><a role="button" href="import/" class="btn btn-primary btn-margin-left">Import</a></li>

{% endblock %}

但是，如果您想通过单元测试来证明它的有效性呢？ 当我可以使用 View 和函数 render_to_string 时，我可以测试模板 像这样:

response = my_view(HttpRequest())
expected_html = render_to_string(
    'template.html',
    {'arg_key': arg}
)
self.assertEqual(response.content.decode(), expected_html)

但是在这里，我不知道如何使用关联模板部分调用管理 View 。 这是我发现在管理页面上添加单元测试的开始。

class MockRequest(object):
    pass


class MockSuperUser(object):

    def has_perm(self, perm):
        return True

request = MockRequest()
request.user = MockSuperUser()


class ModelAdminTests(TestCase):

    def setUp(self):
        self.site = AdminSite()

    def test_admin_import_button_on_location_admin_page(self):
        ma = ModelAdmin(Location, self.site)
        #Here the correct assert ?

谢谢大家

Answer 1

这可能不是一个好主意(可能是更好的选择:using the test client)。但作为引用，以下是如何直接从管理方法以编程方式呈现更改 ListView :

from django.contrib.admin import AdminSite, ModelAdmin
from django.test.client import RequestFactory
from django.contrib.admin.templatetags.admin_urls import admin_urlname
from django.shortcuts import resolve_url

class MockUser:
    is_active = True
    is_staff = True

    def has_perm(self, *args):
        return True

url = resolve_url(admin_urlname(MyModel._meta, "changelist"))
request = RequestFactory().get(url)
request.user = MockUser()

class MyAdmin(ModelAdmin):
    change_list_template = "path/to/custom/template/if/needed"

admin = MyAdmin(Work, AdminSite())
html = admin.changelist_view(request).rendered_content