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)