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
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
|
import os
import socket
import unittest
try:
import pathlib
except ImportError:
# Python 2 backport
import pathlib2 as pathlib
from testpath.asserts import *
from testpath.tempdir import TemporaryDirectory
class TestAssertFunctions(unittest.TestCase):
def setUp(self):
self.td = TemporaryDirectory()
self.addCleanup(self.td.cleanup)
self.file_path = os.path.join(self.td.name, 'afile')
with open(self.file_path, 'w') as f:
f.write('Blah')
self.dir_path = os.path.join(self.td.name, 'adir')
os.mkdir(self.dir_path)
self.link_path = os.path.join(self.td.name, 'alink')
self.pipe_path = os.path.join(self.td.name, 'apipe')
self.socket_path = os.path.join(self.td.name, 'asocket')
if os.name == 'posix':
# Symlinks are rarely usable on Windows, because a special
# permission is needed to create them.
os.symlink(self.file_path, self.link_path)
os.mkfifo(self.pipe_path)
self.sock = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
self.sock.bind(self.socket_path)
self.nonexistant_path = os.path.join(self.td.name, 'doesntexist')
def tearDown(self):
if hasattr(self, 'sock'):
self.sock.close()
def test_exists(self):
assert_path_exists(self.file_path)
assert_path_exists(pathlib.Path(self.file_path))
assert_not_path_exists(self.nonexistant_path)
with self.assertRaises(AssertionError):
assert_path_exists(self.nonexistant_path)
with self.assertRaises(AssertionError):
assert_not_path_exists(self.file_path)
def test_isfile(self):
assert_isfile(self.file_path)
assert_not_isfile(self.dir_path)
with self.assertRaises(AssertionError):
assert_isfile(self.dir_path)
with self.assertRaises(AssertionError):
assert_not_isfile(self.file_path)
def test_isfile_symlink(self):
if os.name == 'nt':
raise unittest.SkipTest('symlink')
assert_isfile(self.link_path) # Follows the link by default
assert_not_isfile(self.link_path, follow_symlinks=False)
with self.assertRaises(AssertionError):
assert_isfile(self.link_path, follow_symlinks=False)
with self.assertRaises(AssertionError):
assert_not_isfile(self.link_path)
def test_isdir(self):
assert_isdir(self.dir_path)
assert_isdir(pathlib.Path(self.dir_path))
assert_not_isdir(self.file_path)
with self.assertRaises(AssertionError):
assert_isdir(self.file_path)
with self.assertRaises(AssertionError):
assert_not_isdir(self.dir_path)
def test_islink(self):
if os.name == 'nt':
raise unittest.SkipTest('symlink')
assert_islink(self.link_path, to=self.file_path)
assert_islink(pathlib.Path(self.link_path),
to=pathlib.Path(self.file_path))
assert_not_islink(self.file_path)
with self.assertRaises(AssertionError) as c:
assert_islink(self.file_path)
self.assertIn('not a symlink', str(c.exception))
with self.assertRaises(AssertionError) as c:
assert_islink(self.link_path, to=self.dir_path)
self.assertIn('target of', str(c.exception))
with self.assertRaises(AssertionError):
assert_not_islink(self.link_path)
def test_ispipe(self):
if os.name == 'nt':
raise unittest.SkipTest('pipe')
assert_ispipe(self.pipe_path)
assert_not_ispipe(self.dir_path)
with self.assertRaises(AssertionError):
assert_ispipe(self.dir_path)
with self.assertRaises(AssertionError):
assert_not_ispipe(self.pipe_path)
def test_issocket(self):
if os.name == 'nt':
raise unittest.SkipTest('socket')
assert_issocket(self.socket_path)
assert_not_issocket(self.file_path)
with self.assertRaises(AssertionError):
assert_issocket(self.file_path)
with self.assertRaises(AssertionError):
assert_not_issocket(self.socket_path)
|
