121 lines
4.2 KiB
Python
121 lines
4.2 KiB
Python
from __future__ import unicode_literals
|
|
|
|
import csv
|
|
|
|
from django.core import management
|
|
from django.utils.encoding import force_bytes
|
|
|
|
from mayan.apps.documents.models import DocumentType, Document
|
|
from mayan.apps.documents.tests import GenericDocumentTestCase
|
|
from mayan.apps.documents.tests.literals import TEST_SMALL_DOCUMENT_PATH
|
|
from mayan.apps.storage.utils import fs_cleanup, mkstemp
|
|
|
|
|
|
class ImportManagementCommandTestCase(GenericDocumentTestCase):
|
|
auto_generate_test_csv_file = True
|
|
auto_upload_document = False
|
|
random_primary_key_enable = False
|
|
test_import_count = 1
|
|
|
|
def setUp(self):
|
|
super(ImportManagementCommandTestCase, self).setUp()
|
|
if self.auto_generate_test_csv_file:
|
|
self._create_test_csv_file()
|
|
|
|
def tearDown(self):
|
|
self._destroy_test_csv_file()
|
|
super(ImportManagementCommandTestCase, self).tearDown()
|
|
|
|
def _create_test_csv_file(self):
|
|
self.test_csv_file_descriptor, self.test_csv_path = mkstemp()
|
|
|
|
print('Test CSV file: {}'.format(self.test_csv_path))
|
|
|
|
with open(self.test_csv_path, mode='wb') as csvfile:
|
|
filewriter = csv.writer(
|
|
csvfile, delimiter=force_bytes(','), quotechar=force_bytes('"'),
|
|
quoting=csv.QUOTE_MINIMAL
|
|
)
|
|
print(
|
|
'Generating test CSV for {} documents'.format(
|
|
self.test_import_count
|
|
)
|
|
)
|
|
for times in range(self.test_import_count):
|
|
filewriter.writerow(
|
|
[
|
|
self.test_document_type.label, TEST_SMALL_DOCUMENT_PATH,
|
|
'column 2', 'column 3', 'column 4', 'column 5',
|
|
'part #', 'value',
|
|
]
|
|
)
|
|
filewriter.writerow(
|
|
[
|
|
self.test_document_type.label, TEST_SMALL_DOCUMENT_PATH,
|
|
'column 2', 'column 3', 'column 4', 'column 5',
|
|
'part#', 'value',
|
|
]
|
|
)
|
|
|
|
def _destroy_test_csv_file(self):
|
|
fs_cleanup(
|
|
filename=self.test_csv_path,
|
|
file_descriptor=self.test_csv_file_descriptor
|
|
)
|
|
|
|
def test_import_csv_read(self):
|
|
self.test_document_type.delete()
|
|
management.call_command('import', self.test_csv_path)
|
|
|
|
self.assertTrue(DocumentType.objects.count() > 0)
|
|
self.assertTrue(Document.objects.count() > 0)
|
|
|
|
def test_import_document_type_column_mapping(self):
|
|
self.test_document_type.delete()
|
|
management.call_command(
|
|
'import', self.test_csv_path, '--document_type_column', '2'
|
|
)
|
|
|
|
self.assertTrue(DocumentType.objects.first().label == 'column 2')
|
|
self.assertTrue(Document.objects.count() > 0)
|
|
|
|
def test_import_document_path_column_mapping(self):
|
|
self.test_document_type.delete()
|
|
with self.assertRaises(IOError):
|
|
management.call_command(
|
|
'import', self.test_csv_path, '--document_path_column', '2'
|
|
)
|
|
|
|
def test_import_metadata_column_mapping(self):
|
|
self.test_document_type.delete()
|
|
management.call_command(
|
|
'import', self.test_csv_path, '--metadata_pairs_column', '2:3,4:5',
|
|
)
|
|
|
|
self.assertTrue(DocumentType.objects.count() > 0)
|
|
self.assertTrue(Document.objects.count() > 0)
|
|
self.assertTrue(Document.objects.first().metadata.count() > 0)
|
|
self.assertEqual(
|
|
Document.objects.first().metadata.get(
|
|
metadata_type__name='column_2'
|
|
).value, 'column 3'
|
|
)
|
|
|
|
def test_import_ambiguous_metadata(self):
|
|
self.auto_generate_test_csv_file = False
|
|
self.test_import_count = 2
|
|
|
|
self.test_document_type.delete()
|
|
management.call_command(
|
|
'import', self.test_csv_path, '--metadata_pairs_column', '6:7',
|
|
)
|
|
|
|
self.assertTrue(DocumentType.objects.count() > 0)
|
|
self.assertTrue(Document.objects.count() > 0)
|
|
self.assertTrue(Document.objects.first().metadata.count() > 0)
|
|
self.assertEqual(
|
|
Document.objects.first().metadata.get(
|
|
metadata_type__name='part'
|
|
).value, 'value'
|
|
)
|