41 lines
1.3 KiB
Python
41 lines
1.3 KiB
Python
from __future__ import unicode_literals
|
|
|
|
from mayan.apps.documents.tests.base import GenericDocumentTestCase
|
|
from mayan.apps.storage.utils import fs_cleanup, mkstemp
|
|
|
|
from ..control_codes import ControlCodeCabinetDocumentAdd
|
|
|
|
from .mixins import CabinetTestMixin, CabinetViewTestMixin
|
|
|
|
|
|
class ControlCodeCabinetDocumentAddTestCase(
|
|
CabinetTestMixin, CabinetViewTestMixin,
|
|
GenericDocumentTestCase
|
|
):
|
|
auto_upload_document = False
|
|
|
|
def setUp(self):
|
|
super(ControlCodeCabinetDocumentAddTestCase, self).setUp()
|
|
self.test_document_path = mkstemp()[1]
|
|
|
|
self._create_test_cabinet()
|
|
self._create_test_cabinet_child()
|
|
|
|
def tearDown(self):
|
|
fs_cleanup(filename=self.test_document_path)
|
|
super(ControlCodeCabinetDocumentAddTestCase, self).tearDown()
|
|
|
|
def test_control_code(self):
|
|
with open(self.test_document_path, mode='wb') as file_object:
|
|
control_code = ControlCodeCabinetDocumentAdd(
|
|
label_path=(
|
|
self.test_cabinet.label, self.test_cabinet_child.label
|
|
),
|
|
)
|
|
control_code.get_image().save(file_object)
|
|
|
|
self.upload_document()
|
|
self.assertEqual(
|
|
self.test_document.cabinets.first(), self.test_cabinet_child
|
|
)
|