99 lines
3.5 KiB
Python
99 lines
3.5 KiB
Python
from __future__ import unicode_literals
|
|
|
|
from furl import furl
|
|
|
|
from django.urls import reverse
|
|
|
|
from mayan.apps.documents.models import Document
|
|
from mayan.apps.documents.permissions import permission_document_create
|
|
from mayan.apps.documents.tests import (
|
|
GenericDocumentViewTestCase, TEST_SMALL_DOCUMENT_PATH,
|
|
)
|
|
from mayan.apps.sources.models import WebFormSource
|
|
from mayan.apps.sources.tests.literals import (
|
|
TEST_SOURCE_LABEL, TEST_SOURCE_UNCOMPRESS_N,
|
|
)
|
|
|
|
from .literals import (
|
|
TEST_METADATA_VALUE_UNICODE, TEST_METADATA_VALUE_WITH_AMPERSAND
|
|
)
|
|
from .mixins import MetadataTypeTestMixin
|
|
|
|
|
|
class DocumentUploadMetadataTestCase(MetadataTypeTestMixin, GenericDocumentViewTestCase):
|
|
auto_upload_document = False
|
|
|
|
def setUp(self):
|
|
super(DocumentUploadMetadataTestCase, self).setUp()
|
|
self.source = WebFormSource.objects.create(
|
|
enabled=True, label=TEST_SOURCE_LABEL,
|
|
uncompress=TEST_SOURCE_UNCOMPRESS_N
|
|
)
|
|
|
|
#self.document.delete()
|
|
|
|
self.document_type.metadata_type_relations.create(
|
|
metadata_type=self.metadata_type, required=True
|
|
)
|
|
|
|
self.upload_url = furl(reverse(viewname='sources:upload_interactive'))
|
|
|
|
def test_upload_interactive_with_unicode_metadata(self):
|
|
self.upload_url.args['metadata0_id'] = self.metadata_type.pk
|
|
self.upload_url.args['metadata0_value'] = TEST_METADATA_VALUE_UNICODE
|
|
|
|
self.grant_access(
|
|
permission=permission_document_create, obj=self.document_type
|
|
)
|
|
|
|
# Upload the test document
|
|
with open(TEST_SMALL_DOCUMENT_PATH, mode='rb') as file_descriptor:
|
|
response = self.post(
|
|
path=self.upload_url, data={
|
|
'document-language': 'eng', 'source-file': file_descriptor,
|
|
'document_type_id': self.document_type.pk,
|
|
}
|
|
)
|
|
self.assertEqual(response.status_code, 302)
|
|
self.assertEqual(Document.objects.count(), 1)
|
|
self.assertEqual(
|
|
Document.objects.first().metadata.first().value,
|
|
TEST_METADATA_VALUE_UNICODE
|
|
)
|
|
|
|
def test_upload_interactive_with_ampersand_metadata(self):
|
|
self.upload_url.args['metadata0_id'] = self.metadata_type.pk
|
|
self.upload_url.args['metadata0_value'] = TEST_METADATA_VALUE_WITH_AMPERSAND
|
|
|
|
self.grant_access(
|
|
permission=permission_document_create, obj=self.document_type
|
|
)
|
|
# Upload the test document
|
|
with open(TEST_SMALL_DOCUMENT_PATH, mode='rb') as file_descriptor:
|
|
response = self.post(
|
|
path=self.upload_url, data={
|
|
'document-language': 'eng', 'source-file': file_descriptor,
|
|
'document_type_id': self.document_type.pk,
|
|
}
|
|
)
|
|
self.assertEqual(response.status_code, 302)
|
|
self.assertEqual(Document.objects.count(), 1)
|
|
self.assertEqual(
|
|
Document.objects.first().metadata.first().value,
|
|
TEST_METADATA_VALUE_WITH_AMPERSAND
|
|
)
|
|
|
|
def test_initial_step_conditions(self):
|
|
self.grant_access(
|
|
permission=permission_document_create, obj=self.document_type
|
|
)
|
|
|
|
response = self.post(
|
|
viewname='sources:document_create_multiple', data={
|
|
'document_type_selection-document_type': self.document_type.pk,
|
|
'document_create_wizard-current_step': 0
|
|
}
|
|
)
|
|
|
|
self.assertContains(response=response, text='Step 2', status_code=200)
|