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 ..widgets import DocumentThumbnailWidget
from ..widgets import DocumentPageThumbnailWidget
from .test_models import GenericDocumentTestCase
class DocumentWidgetTestCase(GenericDocumentTestCase):
class DocumentPageWidgetTestCase(GenericDocumentTestCase):
def test_document_list_view_document_with_no_pages(self):
document_thumbnail_widget = DocumentThumbnailWidget()
document_thumbnail_widget = DocumentPageThumbnailWidget()
self.document.pages.all().delete()
result = document_thumbnail_widget.render(instance=self.document)