Fix widget test.

Signed-off-by: Michael Price <loneviking72@gmail.com>
This commit is contained in:
Michael Price
2018-03-13 01:54:46 -04:00
committed by Roberto Rosario
parent 1fdb452c9d
commit 395be18151

View File

@@ -1,13 +1,13 @@
from __future__ import unicode_literals from __future__ import unicode_literals
from ..widgets import DocumentThumbnailWidget from ..widgets import DocumentPageThumbnailWidget
from .test_models import GenericDocumentTestCase from .test_models import GenericDocumentTestCase
class DocumentWidgetTestCase(GenericDocumentTestCase): class DocumentPageWidgetTestCase(GenericDocumentTestCase):
def test_document_list_view_document_with_no_pages(self): def test_document_list_view_document_with_no_pages(self):
document_thumbnail_widget = DocumentThumbnailWidget() document_thumbnail_widget = DocumentPageThumbnailWidget()
self.document.pages.all().delete() self.document.pages.all().delete()
result = document_thumbnail_widget.render(instance=self.document) result = document_thumbnail_widget.render(instance=self.document)