Add first control code

Signed-off-by: Roberto Rosario <roberto.rosario@mayan-edms.com>
This commit is contained in:
Roberto Rosario
2019-09-05 01:40:08 -04:00
parent 968abe2cdb
commit a13f033104
3 changed files with 47 additions and 19 deletions

View File

@@ -3,11 +3,14 @@ from __future__ import unicode_literals
from mayan.apps.common.tests.base import BaseTestCase
from mayan.apps.documents.tests.base import GenericDocumentTestCase
from ..control_codes import ControlCodeAttributeEdit
from .mixins import ControlSheetCodeTestMixin
#TODO: use mktmp
TEST_CONTROL_CODE_DOCUMENT_PATH = '/tmp/test_control_code.png'
"""
class ControlCodeTestCase(ControlSheetCodeTestMixin, GenericDocumentTestCase):
auto_upload_document = False
test_document_path = TEST_CONTROL_CODE_DOCUMENT_PATH
@@ -17,8 +20,26 @@ class ControlCodeTestCase(ControlSheetCodeTestMixin, GenericDocumentTestCase):
control_code = self._test_control_code_class(
argument_1='test argument value'
)
control_code.image.save(file_object)
control_code.get_image().save(file_object)
self.upload_document()
print self.test_document.pages.count()
"""
class ControlCodeAttributeEditTestCase(GenericDocumentTestCase):
auto_upload_document = False
test_document_path = TEST_CONTROL_CODE_DOCUMENT_PATH
def test_control_code(self):
TEST_ATTRIBUTE_VALUE = 'test value'
with open(TEST_CONTROL_CODE_DOCUMENT_PATH, mode='wb') as file_object:
control_code = ControlCodeAttributeEdit(
attribute='label', value=TEST_ATTRIBUTE_VALUE
)
control_code.get_image().save(file_object)
self.upload_document()
self.test_document.refresh_from_db()
self.assertEqual(self.test_document.label, TEST_ATTRIBUTE_VALUE)