Add unicode filename test for the StagingFile class.

2016-03-20 15:59:49 -04:00
parent 6ec7a2ca4d
commit 7be6e8c6ed
2 changed files with 37 additions and 1 deletions
@@ -0,0 +1,34 @@
+from __future__ import unicode_literals
+
+import os
+import shutil
+import tempfile
+
+from django.test import TestCase
+
+from documents.tests import TEST_NON_ASCII_DOCUMENT_PATH
+
+from ..classes import StagingFile
+
+
+class StagingFileTestCase(TestCase):
+    def test_unicode_staging_file(self):
+        temporary_directory = tempfile.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)