39 lines
1.0 KiB
Python
39 lines
1.0 KiB
Python
from __future__ import unicode_literals
|
|
|
|
from django.test import override_settings
|
|
|
|
from mayan.apps.common.tests.base import BaseTestCase
|
|
from mayan.apps.documents.tests.base import GenericDocumentTestCase
|
|
from mayan.apps.documents.tests.mixins import DocumentTestMixin
|
|
|
|
from ..classes import ControlCode
|
|
|
|
TEST_CONTROL_CODE_DOCUMENT_PATH = '/tmp/test_control_code.png'
|
|
|
|
|
|
class ControlCodeTest(ControlCode):
|
|
arguments = ('argument_1',)
|
|
label = 'Test'
|
|
name = 'test'
|
|
|
|
def execute(self):
|
|
pass
|
|
|
|
|
|
ControlCode.register(control_code=ControlCodeTest)
|
|
|
|
|
|
class ControlCodeTestCase(GenericDocumentTestCase):
|
|
auto_upload_document = False
|
|
test_document_path = TEST_CONTROL_CODE_DOCUMENT_PATH
|
|
|
|
def test_control_code_detection(self):
|
|
|
|
with open(TEST_CONTROL_CODE_DOCUMENT_PATH, mode='wb') as file_object:
|
|
control_code = ControlCodeTest(argument_1='test argument value')
|
|
control_code.image.save(file_object)
|
|
|
|
self.upload_document()
|
|
|
|
print self.test_document.pages.count()
|