36 lines
974 B
Python
36 lines
974 B
Python
from __future__ import unicode_literals
|
|
|
|
import os
|
|
import shutil
|
|
|
|
from common.tests import BaseTestCase
|
|
from common.utils import mkdtemp
|
|
from documents.tests import TEST_NON_ASCII_DOCUMENT_PATH
|
|
|
|
from ..classes import StagingFile
|
|
|
|
|
|
class StagingFileTestCase(BaseTestCase):
|
|
def test_unicode_staging_file(self):
|
|
temporary_directory = mkdtemp()
|
|
shutil.copy(TEST_NON_ASCII_DOCUMENT_PATH, temporary_directory)
|
|
|
|
filename = os.path.basename(TEST_NON_ASCII_DOCUMENT_PATH)
|
|
|
|
class MockStagingFolder(object):
|
|
self.folder_path = temporary_directory
|
|
|
|
staging_file_1 = StagingFile(
|
|
staging_folder=MockStagingFolder(),
|
|
filename=filename
|
|
)
|
|
|
|
staging_file_2 = StagingFile(
|
|
staging_folder=MockStagingFolder(),
|
|
encoded_filename=staging_file_1.encoded_filename
|
|
)
|
|
|
|
self.assertEqual(filename, staging_file_2.filename)
|
|
|
|
shutil.rmtree(temporary_directory)
|