Improve wizard upload view test.

Signed-off-by: Roberto Rosario <roberto.rosario.gonzalez@gmail.com>
This commit is contained in:
Roberto Rosario
2017-07-13 18:21:02 -04:00
parent ba6387021e
commit 5f9a052236
2 changed files with 17 additions and 17 deletions

View File

@@ -15,8 +15,8 @@ __all__ = (
'TEST_NON_ASCII_COMPRESSED_DOCUMENT_FILENAME',
'TEST_NON_ASCII_COMPRESSED_DOCUMENT_PATH',
'TEST_NON_ASCII_DOCUMENT_FILENAME', 'TEST_NON_ASCII_DOCUMENT_PATH',
'TEST_SMALL_DOCUMENT_FILENAME', 'TEST_SMALL_DOCUMENT_PATH',
'TEST_DOCUMENT_VERSION_COMMENT_EDITED',
'TEST_SMALL_DOCUMENT_CHECKSUM', 'TEST_SMALL_DOCUMENT_FILENAME',
'TEST_SMALL_DOCUMENT_PATH', 'TEST_DOCUMENT_VERSION_COMMENT_EDITED',
)
# Filenames

View File

@@ -12,8 +12,8 @@ from common.utils import fs_cleanup, mkdtemp
from documents.models import Document, DocumentType
from documents.permissions import permission_document_create
from documents.tests import (
TEST_DOCUMENT_PATH, TEST_SMALL_DOCUMENT_PATH, TEST_DOCUMENT_DESCRIPTION,
TEST_DOCUMENT_TYPE
TEST_DOCUMENT_DESCRIPTION, TEST_DOCUMENT_PATH, TEST_DOCUMENT_TYPE,
TEST_SMALL_DOCUMENT_CHECKSUM, TEST_SMALL_DOCUMENT_PATH
)
from documents.tests.test_views import GenericDocumentViewTestCase
@@ -40,17 +40,20 @@ class DocumentUploadTestCase(GenericDocumentViewTestCase):
self.document.delete()
def test_upload_wizard_without_permission(self):
self.login_user()
with open(TEST_DOCUMENT_PATH) as file_object:
response = self.post(
def _request_upload_wizard(self):
with open(TEST_SMALL_DOCUMENT_PATH) as file_object:
return self.post(
'sources:upload_interactive', args=(self.source.pk,), data={
'source-file': file_object,
'document_type_id': self.document_type.pk,
}
}, follow=True
)
def test_upload_wizard_without_permission(self):
self.login_user()
response = self._request_upload_wizard()
self.assertEqual(response.status_code, 403)
self.assertEqual(Document.objects.count(), 0)
@@ -59,16 +62,13 @@ class DocumentUploadTestCase(GenericDocumentViewTestCase):
self.grant(permission_document_create)
with open(TEST_DOCUMENT_PATH) as file_object:
response = self.post(
'sources:upload_interactive', args=(self.source.pk,), data={
'source-file': file_object,
'document_type_id': self.document_type.pk,
}, follow=True
)
response = self._request_upload_wizard()
self.assertTrue(b'queued' in response.content)
self.assertEqual(Document.objects.count(), 1)
self.assertEqual(
Document.objects.first().checksum, TEST_SMALL_DOCUMENT_CHECKSUM
)
def test_upload_wizard_with_document_type_access(self):
"""