Normalize handling of temporary file and directory creation.

This commit is contained in:
Roberto Rosario
2016-06-23 20:53:50 -04:00
parent ab831aa493
commit 945158bd60
13 changed files with 65 additions and 36 deletions

View File

@@ -1,8 +1,11 @@
from __future__ import unicode_literals
import os
from django.core.files.base import File
from django.test import TestCase, override_settings
from common.settings import setting_temporary_directory
from documents.models import DocumentType
from documents.tests import (
TEST_DOCUMENT_PATH, TEST_DOCUMENT_TYPE, TEST_HYBRID_DOCUMENT_PATH
@@ -15,7 +18,6 @@ from ..parsers import PDFMinerParser, PopplerParser
@override_settings(OCR_AUTO_OCR=False)
class ParserTestCase(TestCase):
def setUp(self):
self.document_type = DocumentType.objects.create(
label=TEST_DOCUMENT_TYPE
)
@@ -46,6 +48,17 @@ class ParserTestCase(TestCase):
'Mayan EDMS Documentation' in self.document.pages.first().ocr_content.content
)
def test_poppler_parser_cleanup(self):
temp_items = len(os.listdir(setting_temporary_directory.value))
parser = PopplerParser()
parser.process_document_version(self.document.latest_version)
self.assertEqual(
temp_items, len(os.listdir(setting_temporary_directory.value))
)
@override_settings(OCR_AUTO_OCR=False)
class TextExtractorTestCase(TestCase):