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
|
"""Tests for coverage.codeunit"""
import os, sys
from coverage.codeunit import code_unit_factory
from coverage.files import FileLocator
sys.path.insert(0, os.path.split(__file__)[0]) # Force relative import for Py3k
from coveragetest import CoverageTest
# pylint: disable-msg=F0401
# Unable to import 'aa' (No module named aa)
class CodeUnitTest(CoverageTest):
"""Tests for coverage.codeunit"""
def setUp(self):
super(CodeUnitTest, self).setUp()
# Parent class saves and restores sys.path, we can just modify it.
testmods = self.nice_file(os.path.dirname(__file__), 'modules')
sys.path.append(testmods)
def test_filenames(self):
acu = code_unit_factory("aa/afile.py", FileLocator())
bcu = code_unit_factory("aa/bb/bfile.py", FileLocator())
ccu = code_unit_factory("aa/bb/cc/cfile.py", FileLocator())
self.assertEqual(acu[0].name, "aa/afile")
self.assertEqual(bcu[0].name, "aa/bb/bfile")
self.assertEqual(ccu[0].name, "aa/bb/cc/cfile")
self.assertEqual(acu[0].flat_rootname(), "aa_afile")
self.assertEqual(bcu[0].flat_rootname(), "aa_bb_bfile")
self.assertEqual(ccu[0].flat_rootname(), "aa_bb_cc_cfile")
self.assertEqual(acu[0].source_file().read(), "# afile.py\n")
self.assertEqual(bcu[0].source_file().read(), "# bfile.py\n")
self.assertEqual(ccu[0].source_file().read(), "# cfile.py\n")
def test_modules(self):
import aa, aa.bb, aa.bb.cc
cu = code_unit_factory([aa, aa.bb, aa.bb.cc], FileLocator())
self.assertEqual(cu[0].name, "aa")
self.assertEqual(cu[1].name, "aa.bb")
self.assertEqual(cu[2].name, "aa.bb.cc")
self.assertEqual(cu[0].flat_rootname(), "aa")
self.assertEqual(cu[1].flat_rootname(), "aa_bb")
self.assertEqual(cu[2].flat_rootname(), "aa_bb_cc")
self.assertEqual(cu[0].source_file().read(), "# aa\n")
self.assertEqual(cu[1].source_file().read(), "# bb\n")
self.assertEqual(cu[2].source_file().read(), "") # yes, empty
def test_module_files(self):
import aa.afile, aa.bb.bfile, aa.bb.cc.cfile
cu = code_unit_factory([aa.afile, aa.bb.bfile, aa.bb.cc.cfile],
FileLocator())
self.assertEqual(cu[0].name, "aa.afile")
self.assertEqual(cu[1].name, "aa.bb.bfile")
self.assertEqual(cu[2].name, "aa.bb.cc.cfile")
self.assertEqual(cu[0].flat_rootname(), "aa_afile")
self.assertEqual(cu[1].flat_rootname(), "aa_bb_bfile")
self.assertEqual(cu[2].flat_rootname(), "aa_bb_cc_cfile")
self.assertEqual(cu[0].source_file().read(), "# afile.py\n")
self.assertEqual(cu[1].source_file().read(), "# bfile.py\n")
self.assertEqual(cu[2].source_file().read(), "# cfile.py\n")
def test_comparison(self):
acu = code_unit_factory("aa/afile.py", FileLocator())
acu2 = code_unit_factory("aa/afile.py", FileLocator())
zcu = code_unit_factory("aa/zfile.py", FileLocator())
bcu = code_unit_factory("aa/bb/bfile.py", FileLocator())
assert acu == acu2 and acu <= acu2 and acu >= acu2
assert acu < zcu and acu <= zcu and acu != zcu
assert zcu > acu and zcu >= acu and zcu != acu
assert acu < bcu and acu <= bcu and acu != bcu
assert bcu > acu and bcu >= acu and bcu != acu
def test_egg(self):
import egg1, egg1.egg1
cu = code_unit_factory([egg1, egg1.egg1], FileLocator())
self.assertEqual(cu[0].source_file().read(), "")
self.assertEqual(cu[1].source_file().read().split("\n")[0],
"# My egg file!"
)
|
