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
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
|
#!/usr/bin/env python
# This Source Code Form is subject to the terms of the Mozilla Public
# License, v. 2.0. If a copy of the MPL was not distributed with this file,
# You can obtain one at http://mozilla.org/MPL/2.0/.
import os
import shutil
import tempfile
import unittest
import mozunit
from manifestparser import ManifestParser, convert
here = os.path.dirname(os.path.abspath(__file__))
# In some cases tempfile.mkdtemp() may returns a path which contains
# symlinks. Some tests here will then break, as the manifestparser.convert
# function returns paths that does not contains symlinks.
#
# Workaround is to use the following function, if absolute path of temp dir
# must be compared.
def create_realpath_tempdir():
"""
Create a tempdir without symlinks.
"""
return os.path.realpath(tempfile.mkdtemp())
class TestDirectoryConversion(unittest.TestCase):
"""test conversion of a directory tree to a manifest structure"""
def create_stub(self, directory=None):
"""stub out a directory with files in it"""
files = ("foo", "bar", "fleem")
if directory is None:
directory = create_realpath_tempdir()
for i in files:
open(os.path.join(directory, i), "w").write(i)
subdir = os.path.join(directory, "subdir")
os.mkdir(subdir)
open(os.path.join(subdir, "subfile"), "w").write("baz")
return directory
def test_directory_to_manifest(self):
"""
Test our ability to convert a static directory structure to a
manifest.
"""
# create a stub directory
stub = self.create_stub()
try:
stub = stub.replace(os.path.sep, "/")
self.assertTrue(os.path.exists(stub) and os.path.isdir(stub))
# Make a manifest for it
manifest = convert([stub])
out_tmpl = """[%(stub)s/bar]
[%(stub)s/fleem]
[%(stub)s/foo]
[%(stub)s/subdir/subfile]
""" # noqa
self.assertEqual(str(manifest), out_tmpl % dict(stub=stub))
except BaseException:
raise
finally:
shutil.rmtree(stub) # cleanup
def test_convert_directory_manifests_in_place(self):
"""
keep the manifests in place
"""
stub = self.create_stub()
try:
ManifestParser.populate_directory_manifests([stub], filename="manifest.ini")
self.assertEqual(
sorted(os.listdir(stub)),
["bar", "fleem", "foo", "manifest.ini", "subdir"],
)
parser = ManifestParser()
parser.read(os.path.join(stub, "manifest.ini"))
self.assertEqual(
[i["name"] for i in parser.tests], ["subfile", "bar", "fleem", "foo"]
)
parser = ManifestParser()
parser.read(os.path.join(stub, "subdir", "manifest.ini"))
self.assertEqual(len(parser.tests), 1)
self.assertEqual(parser.tests[0]["name"], "subfile")
except BaseException:
raise
finally:
shutil.rmtree(stub)
def test_manifest_ignore(self):
"""test manifest `ignore` parameter for ignoring directories"""
stub = self.create_stub()
try:
ManifestParser.populate_directory_manifests(
[stub], filename="manifest.ini", ignore=("subdir",)
)
parser = ManifestParser()
parser.read(os.path.join(stub, "manifest.ini"))
self.assertEqual([i["name"] for i in parser.tests], ["bar", "fleem", "foo"])
self.assertFalse(
os.path.exists(os.path.join(stub, "subdir", "manifest.ini"))
)
except BaseException:
raise
finally:
shutil.rmtree(stub)
def test_pattern(self):
"""test directory -> manifest with a file pattern"""
stub = self.create_stub()
try:
parser = convert([stub], pattern="f*", relative_to=stub)
self.assertEqual([i["name"] for i in parser.tests], ["fleem", "foo"])
# test multiple patterns
parser = convert([stub], pattern=("f*", "s*"), relative_to=stub)
self.assertEqual(
[i["name"] for i in parser.tests], ["fleem", "foo", "subdir/subfile"]
)
except BaseException:
raise
finally:
shutil.rmtree(stub)
def test_update(self):
"""
Test our ability to update tests from a manifest and a directory of
files
"""
# boilerplate
tempdir = create_realpath_tempdir()
for i in range(10):
open(os.path.join(tempdir, str(i)), "w").write(str(i))
# otherwise empty directory with a manifest file
newtempdir = create_realpath_tempdir()
manifest_file = os.path.join(newtempdir, "manifest.ini")
manifest_contents = str(convert([tempdir], relative_to=tempdir))
with open(manifest_file, "w") as f:
f.write(manifest_contents)
# get the manifest
manifest = ManifestParser(manifests=(manifest_file,))
# All of the tests are initially missing:
paths = [str(i) for i in range(10)]
self.assertEqual([i["name"] for i in manifest.missing()], paths)
# But then we copy one over:
self.assertEqual(manifest.get("name", name="1"), ["1"])
manifest.update(tempdir, name="1")
self.assertEqual(sorted(os.listdir(newtempdir)), ["1", "manifest.ini"])
# Update that one file and copy all the "tests":
open(os.path.join(tempdir, "1"), "w").write("secret door")
manifest.update(tempdir)
self.assertEqual(
sorted(os.listdir(newtempdir)),
["0", "1", "2", "3", "4", "5", "6", "7", "8", "9", "manifest.ini"],
)
self.assertEqual(
open(os.path.join(newtempdir, "1")).read().strip(), "secret door"
)
# clean up:
shutil.rmtree(tempdir)
shutil.rmtree(newtempdir)
if __name__ == "__main__":
mozunit.main()
|
