blob: 26360091f7ecbb846f90556343dbb59d9037ef22 (
plain)
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
|
"""Tests of coverage/collector.py and other collectors."""
import os.path
import coverage
from tests.coveragetest import CoverageTest
from tests.helpers import CheckUniqueFilenames
class CollectorTest(CoverageTest):
"""Test specific aspects of the collection process."""
def test_should_trace_cache(self):
# The tracers should only invoke should_trace once for each file name.
# Make some files that invoke each other.
self.make_file("f1.py", """\
def f1(x, f):
return f(x)
""")
self.make_file("f2.py", """\
import f1
def func(x):
return f1.f1(x, otherfunc)
def otherfunc(x):
return x*x
for i in range(10):
func(i)
""")
# Trace one file, but not the other. CheckUniqueFilenames will assert
# that _should_trace hasn't been called twice for the same file.
cov = coverage.coverage(include=["f1.py"])
should_trace_hook = CheckUniqueFilenames.hook(cov, '_should_trace')
# Import the Python file, executing it.
self.start_import_stop(cov, "f2")
# Double-check that our files were checked.
abs_files = set(os.path.abspath(f) for f in should_trace_hook.filenames)
self.assertIn(os.path.abspath("f1.py"), abs_files)
self.assertIn(os.path.abspath("f2.py"), abs_files)
|
