aboutsummaryrefslogtreecommitdiff
path: root/sysutils/py-scandir/files/patch-test_test__scandir.py
blob: c65b270b000d55b836e4eb27d9c145784e9753c7 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
--- test/test_scandir.py.orig	2017-09-29 12:38:51 UTC
+++ test/test_scandir.py
@@ -14,6 +14,9 @@ try:
 except ImportError:
     has_scandir = False
 
+reload(sys)
+sys.setdefaultencoding('utf8')
+
 FILE_ATTRIBUTE_DIRECTORY = 16
 
 TEST_PATH = os.path.abspath(os.path.join(os.path.dirname(__file__), 'testdir'))
@@ -48,24 +51,24 @@ def create_file(path, contents='1234'):
 def setup_main():
     join = os.path.join
 
-    os.mkdir(TEST_PATH)
-    os.mkdir(join(TEST_PATH, 'subdir'))
+    os.makedirs(TEST_PATH)
+    os.makedirs(join(TEST_PATH, 'subdir'))
     create_file(join(TEST_PATH, 'file1.txt'))
     create_file(join(TEST_PATH, 'file2.txt'), contents='12345678')
 
-    os.mkdir(join(TEST_PATH, 'subdir', 'unidir\u018F'))
+    os.makedirs(join(TEST_PATH, 'subdir', u'unidir\u018F'))
     create_file(join(TEST_PATH, 'subdir', 'file1.txt'))
-    create_file(join(TEST_PATH, 'subdir', 'unicod\u018F.txt'))
+    create_file(join(TEST_PATH, 'subdir', u'unicod\u018F.txt'))
 
-    create_file(join(TEST_PATH, 'subdir', 'unidir\u018F', 'file1.txt'))
+    create_file(join(TEST_PATH, 'subdir', u'unidir\u018F', 'file1.txt'))
 
-    os.mkdir(join(TEST_PATH, 'linkdir'))
+    os.makedirs(join(TEST_PATH, 'linkdir'))
 
 
 def setup_symlinks():
     join = os.path.join
 
-    os.mkdir(join(TEST_PATH, 'linkdir', 'linksubdir'))
+    os.makedirs(join(TEST_PATH, 'linkdir', 'linksubdir'))
     create_file(join(TEST_PATH, 'linkdir', 'file1.txt'))
 
     os.symlink(os.path.abspath(join(TEST_PATH, 'linkdir', 'file1.txt')),
@@ -217,7 +220,7 @@ class TestMixin(object):
         self.assertTrue(isinstance(entry.path, bytes))
 
         # b'unicod?.txt' on Windows, b'unicod\xc6\x8f.txt' (UTF-8) or similar on POSIX
-        entry_name = 'unicod\u018f.txt'.encode(sys.getfilesystemencoding(), 'replace')
+        entry_name = u'unicod\u018f.txt'.encode(sys.getfilesystemencoding(), 'replace')
         self.assertEqual(entry.name, entry_name)
         self.assertEqual(entry.path, os.path.join(path, entry_name))
 
@@ -234,12 +237,12 @@ class TestMixin(object):
         self.assertTrue(isinstance(entry.name, str))
         self.assertTrue(isinstance(entry.path, str))
 
-        entry_name = 'unicod\u018f.txt'
+        entry_name = u'unicod\u018f.txt'
         self.assertEqual(entry.name, entry_name)
-        self.assertEqual(entry.path, os.path.join(path, 'unicod\u018f.txt'))
+        self.assertEqual(entry.path, os.path.join(path, u'unicod\u018f.txt'))
 
         # Check that it handles unicode input properly
-        path = os.path.join(TEST_PATH, 'subdir', 'unidir\u018f')
+        path = os.path.join(TEST_PATH, 'subdir', u'unidir\u018f')
         self.assertTrue(isinstance(path, str))
         entries = list(self.scandir_func(path))
         self.assertEqual(len(entries), 1)