bc072f7b7e
Add support for specifying metadata columns. Signed-off-by: Roberto Rosario <roberto.rosario.gonzalez@gmail.com>
95 lines
3.3 KiB
Python
95 lines
3.3 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_upload_document = False
|
|
random_primary_key_enable = False
|
|
test_import_count = 1
|
|
|
|
def setUp(self):
|
|
super(ImportManagementCommandTestCase, self).setUp()
|
|
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',
|
|
'column 6', 'column 7', 'column 8', 'column 9',
|
|
'column 10', 'column 11',
|
|
]
|
|
)
|
|
|
|
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'
|
|
)
|