Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(416)

Unified Diff: tools/telemetry/third_party/coverage/tests/test_process.py

Issue 1366913004: Add coverage Base URL: https://chromium.googlesource.com/chromium/src.git@master
Patch Set: Created 5 years, 3 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
Index: tools/telemetry/third_party/coverage/tests/test_process.py
diff --git a/tools/telemetry/third_party/coverage/tests/test_process.py b/tools/telemetry/third_party/coverage/tests/test_process.py
new file mode 100644
index 0000000000000000000000000000000000000000..3023a18c465d8f8b252cabec2c4f69196023e098
--- /dev/null
+++ b/tools/telemetry/third_party/coverage/tests/test_process.py
@@ -0,0 +1,1034 @@
+# coding: utf8
+# Licensed under the Apache License: http://www.apache.org/licenses/LICENSE-2.0
+# For details: https://bitbucket.org/ned/coveragepy/src/default/NOTICE.txt
+
+"""Tests for process behavior of coverage.py."""
+
+import glob
+import os
+import os.path
+import re
+import sys
+import textwrap
+
+import coverage
+from coverage import env, CoverageData
+
+from tests.coveragetest import CoverageTest
+
+HERE = os.path.dirname(__file__)
+
+
+class ProcessTest(CoverageTest):
+ """Tests of the per-process behavior of coverage.py."""
+
+ def number_of_data_files(self):
+ """Return the number of coverage data files in this directory."""
+ num = 0
+ for f in os.listdir('.'):
+ if f.startswith('.coverage.') or f == '.coverage':
+ num += 1
+ return num
+
+ def test_save_on_exit(self):
+ self.make_file("mycode.py", """\
+ h = "Hello"
+ w = "world"
+ """)
+
+ self.assert_doesnt_exist(".coverage")
+ self.run_command("coverage run mycode.py")
+ self.assert_exists(".coverage")
+
+ def test_environment(self):
+ # Checks that we can import modules from the test directory at all!
+ self.make_file("mycode.py", """\
+ import covmod1
+ import covmodzip1
+ a = 1
+ print('done')
+ """)
+
+ self.assert_doesnt_exist(".coverage")
+ out = self.run_command("coverage run mycode.py")
+ self.assert_exists(".coverage")
+ self.assertEqual(out, 'done\n')
+
+ def make_b_or_c_py(self):
+ """Create b_or_c.py, used in a few of these tests."""
+ self.make_file("b_or_c.py", """\
+ import sys
+ a = 1
+ if sys.argv[1] == 'b':
+ b = 1
+ else:
+ c = 1
+ d = 1
+ print('done')
+ """)
+
+ def test_combine_parallel_data(self):
+ self.make_b_or_c_py()
+ out = self.run_command("coverage run -p b_or_c.py b")
+ self.assertEqual(out, 'done\n')
+ self.assert_doesnt_exist(".coverage")
+ self.assertEqual(self.number_of_data_files(), 1)
+
+ out = self.run_command("coverage run -p b_or_c.py c")
+ self.assertEqual(out, 'done\n')
+ self.assert_doesnt_exist(".coverage")
+
+ # After two -p runs, there should be two .coverage.machine.123 files.
+ self.assertEqual(self.number_of_data_files(), 2)
+
+ # Combine the parallel coverage data files into .coverage .
+ self.run_command("coverage combine")
+ self.assert_exists(".coverage")
+
+ # After combining, there should be only the .coverage file.
+ self.assertEqual(self.number_of_data_files(), 1)
+
+ # Read the coverage file and see that b_or_c.py has all 7 lines
+ # executed.
+ data = coverage.CoverageData()
+ data.read_file(".coverage")
+ self.assertEqual(data.line_counts()['b_or_c.py'], 7)
+
+ def test_combine_parallel_data_in_two_steps(self):
+ self.make_b_or_c_py()
+
+ out = self.run_command("coverage run -p b_or_c.py b")
+ self.assertEqual(out, 'done\n')
+ self.assert_doesnt_exist(".coverage")
+ self.assertEqual(self.number_of_data_files(), 1)
+
+ # Combine the (one) parallel coverage data file into .coverage .
+ self.run_command("coverage combine")
+ self.assert_exists(".coverage")
+ self.assertEqual(self.number_of_data_files(), 1)
+
+ out = self.run_command("coverage run -p b_or_c.py c")
+ self.assertEqual(out, 'done\n')
+ self.assert_exists(".coverage")
+ self.assertEqual(self.number_of_data_files(), 2)
+
+ # Combine the parallel coverage data files into .coverage .
+ self.run_command("coverage combine")
+ self.assert_exists(".coverage")
+
+ # After combining, there should be only the .coverage file.
+ self.assertEqual(self.number_of_data_files(), 1)
+
+ # Read the coverage file and see that b_or_c.py has all 7 lines
+ # executed.
+ data = coverage.CoverageData()
+ data.read_file(".coverage")
+ self.assertEqual(data.line_counts()['b_or_c.py'], 7)
+
+ def test_append_data(self):
+ self.make_b_or_c_py()
+
+ out = self.run_command("coverage run b_or_c.py b")
+ self.assertEqual(out, 'done\n')
+ self.assert_exists(".coverage")
+ self.assertEqual(self.number_of_data_files(), 1)
+
+ out = self.run_command("coverage run --append b_or_c.py c")
+ self.assertEqual(out, 'done\n')
+ self.assert_exists(".coverage")
+ self.assertEqual(self.number_of_data_files(), 1)
+
+ # Read the coverage file and see that b_or_c.py has all 7 lines
+ # executed.
+ data = coverage.CoverageData()
+ data.read_file(".coverage")
+ self.assertEqual(data.line_counts()['b_or_c.py'], 7)
+
+ def test_append_data_with_different_file(self):
+ self.make_b_or_c_py()
+
+ self.make_file(".coveragerc", """\
+ [run]
+ data_file = .mycovdata
+ """)
+
+ out = self.run_command("coverage run b_or_c.py b")
+ self.assertEqual(out, 'done\n')
+ self.assert_doesnt_exist(".coverage")
+ self.assert_exists(".mycovdata")
+
+ out = self.run_command("coverage run --append b_or_c.py c")
+ self.assertEqual(out, 'done\n')
+ self.assert_doesnt_exist(".coverage")
+ self.assert_exists(".mycovdata")
+
+ # Read the coverage file and see that b_or_c.py has all 7 lines
+ # executed.
+ data = coverage.CoverageData()
+ data.read_file(".mycovdata")
+ self.assertEqual(data.line_counts()['b_or_c.py'], 7)
+
+ def test_append_can_create_a_data_file(self):
+ self.make_b_or_c_py()
+
+ out = self.run_command("coverage run --append b_or_c.py b")
+ self.assertEqual(out, 'done\n')
+ self.assert_exists(".coverage")
+ self.assertEqual(self.number_of_data_files(), 1)
+
+ # Read the coverage file and see that b_or_c.py has only 6 lines
+ # executed.
+ data = coverage.CoverageData()
+ data.read_file(".coverage")
+ self.assertEqual(data.line_counts()['b_or_c.py'], 6)
+
+ def test_combine_with_rc(self):
+ self.make_b_or_c_py()
+
+ self.make_file(".coveragerc", """\
+ [run]
+ parallel = true
+ """)
+
+ out = self.run_command("coverage run b_or_c.py b")
+ self.assertEqual(out, 'done\n')
+ self.assert_doesnt_exist(".coverage")
+
+ out = self.run_command("coverage run b_or_c.py c")
+ self.assertEqual(out, 'done\n')
+ self.assert_doesnt_exist(".coverage")
+
+ # After two runs, there should be two .coverage.machine.123 files.
+ self.assertEqual(self.number_of_data_files(), 2)
+
+ # Combine the parallel coverage data files into .coverage .
+ self.run_command("coverage combine")
+ self.assert_exists(".coverage")
+ self.assert_exists(".coveragerc")
+
+ # After combining, there should be only the .coverage file.
+ self.assertEqual(self.number_of_data_files(), 1)
+
+ # Read the coverage file and see that b_or_c.py has all 7 lines
+ # executed.
+ data = coverage.CoverageData()
+ data.read_file(".coverage")
+ self.assertEqual(data.line_counts()['b_or_c.py'], 7)
+
+ # Reporting should still work even with the .rc file
+ out = self.run_command("coverage report")
+ self.assertMultiLineEqual(out, textwrap.dedent("""\
+ Name Stmts Miss Cover
+ -------------------------------
+ b_or_c.py 7 0 100%
+ """))
+
+ def test_combine_with_aliases(self):
+ self.make_file("d1/x.py", """\
+ a = 1
+ b = 2
+ print("%s %s" % (a, b))
+ """)
+
+ self.make_file("d2/x.py", """\
+ # 1
+ # 2
+ # 3
+ c = 4
+ d = 5
+ print("%s %s" % (c, d))
+ """)
+
+ self.make_file(".coveragerc", """\
+ [run]
+ parallel = True
+
+ [paths]
+ source =
+ src
+ */d1
+ */d2
+ """)
+
+ out = self.run_command("coverage run " + os.path.normpath("d1/x.py"))
+ self.assertEqual(out, '1 2\n')
+ out = self.run_command("coverage run " + os.path.normpath("d2/x.py"))
+ self.assertEqual(out, '4 5\n')
+
+ self.assertEqual(self.number_of_data_files(), 2)
+
+ self.run_command("coverage combine")
+ self.assert_exists(".coverage")
+
+ # After combining, there should be only the .coverage file.
+ self.assertEqual(self.number_of_data_files(), 1)
+
+ # Read the coverage data file and see that the two different x.py
+ # files have been combined together.
+ data = coverage.CoverageData()
+ data.read_file(".coverage")
+ summary = data.line_counts(fullpath=True)
+ self.assertEqual(len(summary), 1)
+ actual = os.path.normcase(os.path.abspath(list(summary.keys())[0]))
+ expected = os.path.normcase(os.path.abspath('src/x.py'))
+ self.assertEqual(actual, expected)
+ self.assertEqual(list(summary.values())[0], 6)
+
+ def test_erase_parallel(self):
+ self.make_file(".coveragerc", """\
+ [run]
+ data_file = data.dat
+ parallel = True
+ """)
+ self.make_file("data.dat")
+ self.make_file("data.dat.fooey")
+ self.make_file("data.dat.gooey")
+ self.make_file(".coverage")
+
+ self.run_command("coverage erase")
+ self.assert_doesnt_exist("data.dat")
+ self.assert_doesnt_exist("data.dat.fooey")
+ self.assert_doesnt_exist("data.dat.gooey")
+ self.assert_exists(".coverage")
+
+ def test_missing_source_file(self):
+ # Check what happens if the source is missing when reporting happens.
+ self.make_file("fleeting.py", """\
+ s = 'goodbye, cruel world!'
+ """)
+
+ self.run_command("coverage run fleeting.py")
+ os.remove("fleeting.py")
+ out = self.run_command("coverage html -d htmlcov")
+ self.assertRegex(out, "No source for code: '.*fleeting.py'")
+ self.assertNotIn("Traceback", out)
+
+ # It happens that the code paths are different for *.py and other
+ # files, so try again with no extension.
+ self.make_file("fleeting", """\
+ s = 'goodbye, cruel world!'
+ """)
+
+ self.run_command("coverage run fleeting")
+ os.remove("fleeting")
+ status, out = self.run_command_status("coverage html -d htmlcov")
+ self.assertRegex(out, "No source for code: '.*fleeting'")
+ self.assertNotIn("Traceback", out)
+ self.assertEqual(status, 1)
+
+ def test_running_missing_file(self):
+ status, out = self.run_command_status("coverage run xyzzy.py")
+ self.assertRegex(out, "No file to run: .*xyzzy.py")
+ self.assertNotIn("raceback", out)
+ self.assertNotIn("rror", out)
+ self.assertEqual(status, 1)
+
+ def test_code_throws(self):
+ self.make_file("throw.py", """\
+ def f1():
+ raise Exception("hey!")
+
+ def f2():
+ f1()
+
+ f2()
+ """)
+
+ # The important thing is for "coverage run" and "python" to report the
+ # same traceback.
+ status, out = self.run_command_status("coverage run throw.py")
+ out2 = self.run_command("python throw.py")
+ if env.PYPY:
+ # Pypy has an extra frame in the traceback for some reason
+ lines2 = out2.splitlines()
+ out2 = "".join(l+"\n" for l in lines2 if "toplevel" not in l)
+ self.assertMultiLineEqual(out, out2)
+
+ # But also make sure that the output is what we expect.
+ self.assertIn('File "throw.py", line 5, in f2', out)
+ self.assertIn('raise Exception("hey!")', out)
+ self.assertNotIn('coverage', out)
+ self.assertEqual(status, 1)
+
+ def test_code_exits(self):
+ self.make_file("exit.py", """\
+ import sys
+ def f1():
+ print("about to exit..")
+ sys.exit(17)
+
+ def f2():
+ f1()
+
+ f2()
+ """)
+
+ # The important thing is for "coverage run" and "python" to have the
+ # same output. No traceback.
+ status, out = self.run_command_status("coverage run exit.py")
+ status2, out2 = self.run_command_status("python exit.py")
+ self.assertMultiLineEqual(out, out2)
+ self.assertMultiLineEqual(out, "about to exit..\n")
+ self.assertEqual(status, status2)
+ self.assertEqual(status, 17)
+
+ def test_code_exits_no_arg(self):
+ self.make_file("exit_none.py", """\
+ import sys
+ def f1():
+ print("about to exit quietly..")
+ sys.exit()
+
+ f1()
+ """)
+ status, out = self.run_command_status("coverage run exit_none.py")
+ status2, out2 = self.run_command_status("python exit_none.py")
+ self.assertMultiLineEqual(out, out2)
+ self.assertMultiLineEqual(out, "about to exit quietly..\n")
+ self.assertEqual(status, status2)
+ self.assertEqual(status, 0)
+
+ def test_coverage_run_is_like_python(self):
+ tryfile = os.path.join(HERE, "try_execfile.py")
+ with open(tryfile) as f:
+ self.make_file("run_me.py", f.read())
+ out_cov = self.run_command("coverage run run_me.py")
+ out_py = self.run_command("python run_me.py")
+ self.assertMultiLineEqual(out_cov, out_py)
+
+ def test_coverage_run_dashm_is_like_python_dashm(self):
+ # These -m commands assume the coverage tree is on the path.
+ out_cov = self.run_command("coverage run -m tests.try_execfile")
+ out_py = self.run_command("python -m tests.try_execfile")
+ self.assertMultiLineEqual(out_cov, out_py)
+
+ def test_coverage_run_dir_is_like_python_dir(self):
+ tryfile = os.path.join(HERE, "try_execfile.py")
+ with open(tryfile) as f:
+ self.make_file("with_main/__main__.py", f.read())
+ out_cov = self.run_command("coverage run with_main")
+ out_py = self.run_command("python with_main")
+
+ # The coverage.py results are not identical to the Python results, and
+ # I don't know why. For now, ignore those failures. If someone finds
+ # a real problem with the discrepancies, we can work on it some more.
+ ignored = r"__file__|__loader__|__package__"
+ # PyPy includes the current directory in the path when running a
+ # directory, while CPython and coverage.py do not. Exclude that from
+ # the comparison also...
+ if env.PYPY:
+ ignored += "|"+re.escape(os.getcwd())
+ out_cov = remove_matching_lines(out_cov, ignored)
+ out_py = remove_matching_lines(out_py, ignored)
+ self.assertMultiLineEqual(out_cov, out_py)
+
+ def test_coverage_run_dashm_equal_to_doubledashsource(self):
+ """regression test for #328
+
+ When imported by -m, a module's __name__ is __main__, but we need the
+ --source machinery to know and respect the original name.
+ """
+ # These -m commands assume the coverage tree is on the path.
+ out_cov = self.run_command(
+ "coverage run --source tests.try_execfile -m tests.try_execfile"
+ )
+ out_py = self.run_command("python -m tests.try_execfile")
+ self.assertMultiLineEqual(out_cov, out_py)
+
+ def test_coverage_run_dashm_superset_of_doubledashsource(self):
+ """Edge case: --source foo -m foo.bar"""
+ # These -m commands assume the coverage tree is on the path.
+ out_cov = self.run_command(
+ "coverage run --source tests -m tests.try_execfile"
+ )
+ out_py = self.run_command("python -m tests.try_execfile")
+ self.assertMultiLineEqual(out_cov, out_py)
+
+ st, out = self.run_command_status("coverage report")
+ self.assertEqual(st, 0)
+ self.assertEqual(self.line_count(out), 6, out)
+
+ def test_coverage_run_script_imports_doubledashsource(self):
+ # This file imports try_execfile, which compiles it to .pyc, so the
+ # first run will have __file__ == "try_execfile.py" and the second will
+ # have __file__ == "try_execfile.pyc", which throws off the comparison.
+ # Setting dont_write_bytecode True stops the compilation to .pyc and
+ # keeps the test working.
+ self.make_file("myscript", """\
+ import sys; sys.dont_write_bytecode = True
+ import tests.try_execfile
+ """)
+
+ # These -m commands assume the coverage tree is on the path.
+ out_cov = self.run_command(
+ "coverage run --source tests myscript"
+ )
+ out_py = self.run_command("python myscript")
+ self.assertMultiLineEqual(out_cov, out_py)
+
+ st, out = self.run_command_status("coverage report")
+ self.assertEqual(st, 0)
+ self.assertEqual(self.line_count(out), 6, out)
+
+ def test_coverage_run_dashm_is_like_python_dashm_off_path(self):
+ # https://bitbucket.org/ned/coveragepy/issue/242
+ tryfile = os.path.join(HERE, "try_execfile.py")
+ self.make_file("sub/__init__.py", "")
+ with open(tryfile) as f:
+ self.make_file("sub/run_me.py", f.read())
+ out_cov = self.run_command("coverage run -m sub.run_me")
+ out_py = self.run_command("python -m sub.run_me")
+ self.assertMultiLineEqual(out_cov, out_py)
+
+ def test_coverage_run_dashm_is_like_python_dashm_with__main__207(self):
+ if sys.version_info < (2, 7):
+ # Coverage.py isn't bug-for-bug compatible in the behavior of -m for
+ # Pythons < 2.7
+ self.skip("-m doesn't work the same < Python 2.7")
+ # https://bitbucket.org/ned/coveragepy/issue/207
+ self.make_file("package/__init__.py", "print('init')")
+ self.make_file("package/__main__.py", "print('main')")
+ out_cov = self.run_command("coverage run -m package")
+ out_py = self.run_command("python -m package")
+ self.assertMultiLineEqual(out_cov, out_py)
+
+ def test_fork(self):
+ if not hasattr(os, 'fork'):
+ self.skip("Can't test os.fork since it doesn't exist.")
+
+ self.make_file("fork.py", """\
+ import os
+
+ def child():
+ print('Child!')
+
+ def main():
+ ret = os.fork()
+
+ if ret == 0:
+ child()
+ else:
+ os.waitpid(ret, 0)
+
+ main()
+ """)
+
+ out = self.run_command("coverage run -p fork.py")
+ self.assertEqual(out, 'Child!\n')
+ self.assert_doesnt_exist(".coverage")
+
+ # After running the forking program, there should be two
+ # .coverage.machine.123 files.
+ self.assertEqual(self.number_of_data_files(), 2)
+
+ # Combine the parallel coverage data files into .coverage .
+ self.run_command("coverage combine")
+ self.assert_exists(".coverage")
+
+ # After combining, there should be only the .coverage file.
+ self.assertEqual(self.number_of_data_files(), 1)
+
+ # Read the coverage file and see that b_or_c.py has all 7 lines
+ # executed.
+ data = coverage.CoverageData()
+ data.read_file(".coverage")
+ self.assertEqual(data.line_counts()['fork.py'], 9)
+
+ def test_warnings(self):
+ self.make_file("hello.py", """\
+ import sys, os
+ print("Hello")
+ """)
+ out = self.run_command("coverage run --source=sys,xyzzy,quux hello.py")
+
+ self.assertIn("Hello\n", out)
+ self.assertIn(textwrap.dedent("""\
+ Coverage.py warning: Module sys has no Python source.
+ Coverage.py warning: Module xyzzy was never imported.
+ Coverage.py warning: Module quux was never imported.
+ Coverage.py warning: No data was collected.
+ """), out)
+
+ def test_warnings_during_reporting(self):
+ # While fixing issue #224, the warnings were being printed far too
+ # often. Make sure they're not any more.
+ self.make_file("hello.py", """\
+ import sys, os, the_other
+ print("Hello")
+ """)
+ self.make_file("the_other.py", """\
+ print("What?")
+ """)
+ self.make_file(".coveragerc", """\
+ [run]
+ source =
+ .
+ xyzzy
+ """)
+
+ self.run_command("coverage run hello.py")
+ out = self.run_command("coverage html")
+ self.assertEqual(out.count("Module xyzzy was never imported."), 0)
+
+ def test_warnings_if_never_run(self):
+ out = self.run_command("coverage run i_dont_exist.py")
+ self.assertIn("No file to run: 'i_dont_exist.py'", out)
+ self.assertNotIn("warning", out)
+ self.assertNotIn("Exception", out)
+
+ out = self.run_command("coverage run -m no_such_module")
+ self.assertTrue(
+ ("No module named no_such_module" in out) or
+ ("No module named 'no_such_module'" in out)
+ )
+ self.assertNotIn("warning", out)
+ self.assertNotIn("Exception", out)
+
+ def test_warnings_trace_function_changed_with_threads(self):
+ # https://bitbucket.org/ned/coveragepy/issue/164
+ self.make_file("bug164.py", """\
+ import threading
+ import time
+
+ class MyThread (threading.Thread):
+ def run(self):
+ print("Hello")
+
+ thr = MyThread()
+ thr.start()
+ thr.join()
+ """)
+ out = self.run_command("coverage run --timid bug164.py")
+
+ self.assertIn("Hello\n", out)
+ self.assertNotIn("warning", out)
+
+ def test_warning_trace_function_changed(self):
+ self.make_file("settrace.py", """\
+ import sys
+ print("Hello")
+ sys.settrace(None)
+ print("Goodbye")
+ """)
+ out = self.run_command("coverage run --timid settrace.py")
+ self.assertIn("Hello\n", out)
+ self.assertIn("Goodbye\n", out)
+
+ self.assertIn("Trace function changed", out)
+
+ def test_note(self):
+ self.make_file(".coveragerc", """\
+ [run]
+ data_file = mydata.dat
+ note = These are musical notes: ♫𝅗𝅥♩
+ """)
+ self.make_file("simple.py", """print('hello')""")
+ self.run_command("coverage run simple.py")
+
+ data = CoverageData()
+ data.read_file("mydata.dat")
+ infos = data.run_infos()
+ self.assertEqual(len(infos), 1)
+ self.assertEqual(infos[0]['note'], u"These are musical notes: ♫𝅗𝅥♩")
+
+ def test_fullcoverage(self): # pragma: not covered
+ if env.PY2: # This doesn't work on Python 2.
+ self.skip("fullcoverage doesn't work on Python 2.")
+ # It only works with the C tracer, and if we aren't measuring ourselves.
+ if not env.C_TRACER or env.METACOV:
+ self.skip("fullcoverage only works with the C tracer.")
+
+ # fullcoverage is a trick to get stdlib modules measured from
+ # the very beginning of the process. Here we import os and
+ # then check how many lines are measured.
+ self.make_file("getenv.py", """\
+ import os
+ print("FOOEY == %s" % os.getenv("FOOEY"))
+ """)
+
+ fullcov = os.path.join(
+ os.path.dirname(coverage.__file__), "fullcoverage"
+ )
+ self.set_environ("FOOEY", "BOO")
+ self.set_environ("PYTHONPATH", fullcov)
+ out = self.run_command("python -m coverage run -L getenv.py")
+ self.assertEqual(out, "FOOEY == BOO\n")
+ data = coverage.CoverageData()
+ data.read_file(".coverage")
+ # The actual number of executed lines in os.py when it's
+ # imported is 120 or so. Just running os.getenv executes
+ # about 5.
+ self.assertGreater(data.line_counts()['os.py'], 50)
+
+ def test_deprecation_warnings(self):
+ # Test that coverage doesn't trigger deprecation warnings.
+ # https://bitbucket.org/ned/coveragepy/issue/305/pendingdeprecationwarning-the-imp-module
+ self.make_file("allok.py", """\
+ import warnings
+ warnings.simplefilter('default')
+ import coverage
+ print("No warnings!")
+ """)
+ out = self.run_command("python allok.py")
+ self.assertEqual(out, "No warnings!\n")
+
+ def test_run_twice(self):
+ # https://bitbucket.org/ned/coveragepy/issue/353/40a3-introduces-an-unexpected-third-case
+ self.make_file("foo.py", """\
+ def foo():
+ pass
+ """)
+ self.make_file("run_twice.py", """\
+ import coverage
+
+ for _ in [1, 2]:
+ inst = coverage.Coverage(source=['foo'])
+ inst.load()
+ inst.start()
+ import foo
+ inst.stop()
+ inst.combine()
+ inst.save()
+ """)
+ out = self.run_command("python run_twice.py")
+ self.assertEqual(
+ out,
+ "Coverage.py warning: Module foo was previously imported, but not measured.\n"
+ )
+
+
+class AliasedCommandTest(CoverageTest):
+ """Tests of the version-specific command aliases."""
+
+ run_in_temp_dir = False
+
+ def test_major_version_works(self):
+ # "coverage2" works on py2
+ cmd = "coverage%d" % sys.version_info[0]
+ out = self.run_command(cmd)
+ self.assertIn("Code coverage for Python", out)
+
+ def test_wrong_alias_doesnt_work(self):
+ # "coverage3" doesn't work on py2
+ badcmd = "coverage%d" % (5 - sys.version_info[0])
+ out = self.run_command(badcmd)
+ self.assertNotIn("Code coverage for Python", out)
+
+ def test_specific_alias_works(self):
+ # "coverage-2.7" works on py2.7
+ cmd = "coverage-%d.%d" % sys.version_info[:2]
+ out = self.run_command(cmd)
+ self.assertIn("Code coverage for Python", out)
+
+
+class PydocTest(CoverageTest):
+ """Test that pydoc can get our information."""
+
+ run_in_temp_dir = False
+
+ def assert_pydoc_ok(self, name, thing):
+ """Check that pydoc of `name` finds the docstring from `thing`."""
+ # Run pydoc.
+ out = self.run_command("python -m pydoc " + name)
+ # It should say "Help on..", and not have a traceback
+ self.assert_starts_with(out, "Help on ")
+ self.assertNotIn("Traceback", out)
+
+ # All of the lines in the docstring should be there somewhere.
+ for line in thing.__doc__.splitlines():
+ self.assertIn(line.strip(), out)
+
+ def test_pydoc_coverage(self):
+ self.assert_pydoc_ok("coverage", coverage)
+
+ def test_pydoc_coverage_coverage(self):
+ self.assert_pydoc_ok("coverage.Coverage", coverage.Coverage)
+
+
+class FailUnderTest(CoverageTest):
+ """Tests of the --fail-under switch."""
+
+ def setUp(self):
+ super(FailUnderTest, self).setUp()
+ self.make_file("forty_two_plus.py", """\
+ # I have 42.857% (3/7) coverage!
+ a = 1
+ b = 2
+ if a > 3:
+ b = 4
+ c = 5
+ d = 6
+ e = 7
+ """)
+ st, _ = self.run_command_status("coverage run forty_two_plus.py")
+ self.assertEqual(st, 0)
+ st, out = self.run_command_status("coverage report")
+ self.assertEqual(st, 0)
+ self.assertEqual(
+ self.last_line_squeezed(out),
+ "forty_two_plus.py 7 4 43%"
+ )
+
+ def test_report(self):
+ st, _ = self.run_command_status("coverage report --fail-under=42")
+ self.assertEqual(st, 0)
+ st, _ = self.run_command_status("coverage report --fail-under=43")
+ self.assertEqual(st, 0)
+ st, _ = self.run_command_status("coverage report --fail-under=44")
+ self.assertEqual(st, 2)
+
+ def test_html_report(self):
+ st, _ = self.run_command_status("coverage html --fail-under=42")
+ self.assertEqual(st, 0)
+ st, _ = self.run_command_status("coverage html --fail-under=43")
+ self.assertEqual(st, 0)
+ st, _ = self.run_command_status("coverage html --fail-under=44")
+ self.assertEqual(st, 2)
+
+ def test_xml_report(self):
+ st, _ = self.run_command_status("coverage xml --fail-under=42")
+ self.assertEqual(st, 0)
+ st, _ = self.run_command_status("coverage xml --fail-under=43")
+ self.assertEqual(st, 0)
+ st, _ = self.run_command_status("coverage xml --fail-under=44")
+ self.assertEqual(st, 2)
+
+ def test_fail_under_in_config(self):
+ self.make_file(".coveragerc", "[report]\nfail_under = 43\n")
+ st, _ = self.run_command_status("coverage report")
+ self.assertEqual(st, 0)
+
+ self.make_file(".coveragerc", "[report]\nfail_under = 44\n")
+ st, _ = self.run_command_status("coverage report")
+ self.assertEqual(st, 2)
+
+
+class FailUnderNoFilesTest(CoverageTest):
+ """Test that nothing to report results in an error exit status."""
+ def setUp(self):
+ super(FailUnderNoFilesTest, self).setUp()
+ self.make_file(".coveragerc", "[report]\nfail_under = 99\n")
+
+ def test_report(self):
+ st, out = self.run_command_status("coverage report")
+ self.assertIn('No data to report.', out)
+ self.assertEqual(st, 1)
+
+ def test_xml(self):
+ st, out = self.run_command_status("coverage xml")
+ self.assertIn('No data to report.', out)
+ self.assertEqual(st, 1)
+
+ def test_html(self):
+ st, out = self.run_command_status("coverage html")
+ self.assertIn('No data to report.', out)
+ self.assertEqual(st, 1)
+
+
+class FailUnderEmptyFilesTest(CoverageTest):
+ """Test that empty files produce the proper fail_under exit status."""
+ def setUp(self):
+ super(FailUnderEmptyFilesTest, self).setUp()
+
+ self.make_file(".coveragerc", "[report]\nfail_under = 99\n")
+ self.make_file("empty.py", "")
+ st, _ = self.run_command_status("coverage run empty.py")
+ self.assertEqual(st, 0)
+
+ def test_report(self):
+ st, _ = self.run_command_status("coverage report")
+ self.assertEqual(st, 2)
+
+ def test_xml(self):
+ st, _ = self.run_command_status("coverage xml")
+ self.assertEqual(st, 2)
+
+ def test_html(self):
+ st, _ = self.run_command_status("coverage html")
+ self.assertEqual(st, 2)
+
+
+def possible_pth_dirs():
+ """Produce a sequence of directories for trying to write .pth files."""
+ # First look through sys.path, and we find a .pth file, then it's a good
+ # place to put ours.
+ for d in sys.path:
+ g = glob.glob(os.path.join(d, "*.pth"))
+ if g:
+ yield d
+
+ # If we're still looking, then try the Python library directory.
+ # https://bitbucket.org/ned/coveragepy/issue/339/pth-test-malfunctions
+ import distutils.sysconfig # pylint: disable=import-error
+ yield distutils.sysconfig.get_python_lib()
+
+
+class ProcessCoverageMixin(object):
+ """Set up a .pth file to coverage-measure all sub-processes."""
+
+ def setUp(self):
+ super(ProcessCoverageMixin, self).setUp()
+ # Find a place to put a .pth file.
+ pth_contents = "import coverage; coverage.process_startup()\n"
+ for pth_dir in possible_pth_dirs(): # pragma: part covered
+ pth_path = os.path.join(pth_dir, "subcover.pth")
+ with open(pth_path, "w") as pth:
+ try:
+ pth.write(pth_contents)
+ self.pth_path = pth_path
+ break
+ except (IOError, OSError): # pragma: not covered
+ pass
+ else: # pragma: not covered
+ raise Exception("Couldn't find a place for the .pth file")
+
+ self.addCleanup(os.remove, self.pth_path)
+
+
+class ProcessStartupTest(ProcessCoverageMixin, CoverageTest):
+ """Test that we can measure coverage in sub-processes."""
+
+ def test_subprocess_with_pth_files(self): # pragma: not covered
+ if env.METACOV:
+ self.skip("Can't test sub-process pth file suppport during metacoverage")
+
+ # Main will run sub.py
+ self.make_file("main.py", """\
+ import os, os.path, sys
+ ex = os.path.basename(sys.executable)
+ os.system(ex + " sub.py")
+ """)
+ # sub.py will write a few lines.
+ self.make_file("sub.py", """\
+ with open("out.txt", "w") as f:
+ f.write("Hello, world!\\n")
+ """)
+ self.make_file("coverage.ini", """\
+ [run]
+ data_file = .mycovdata
+ """)
+ self.set_environ("COVERAGE_PROCESS_START", "coverage.ini")
+ import main # pylint: disable=import-error,unused-variable
+
+ with open("out.txt") as f:
+ self.assertEqual(f.read(), "Hello, world!\n")
+
+ # Read the data from .coverage
+ self.assert_exists(".mycovdata")
+ data = coverage.CoverageData()
+ data.read_file(".mycovdata")
+ self.assertEqual(data.line_counts()['sub.py'], 2)
+
+
+class ProcessStartupWithSourceTest(ProcessCoverageMixin, CoverageTest):
+ """Show that we can configure {[run]source} during process-level coverage.
+
+ There are three interesting variables, for a total of eight tests:
+
+ 1. -m versus a simple script argument (for example, `python myscript`),
+
+ 2. filtering for the top-level (main.py) or second-level (sub.py)
+ module, and
+
+ 3. whether the files are in a package or not.
+
+ """
+
+ def assert_pth_and_source_work_together(
+ self, dashm, package, source
+ ): # pragma: not covered
+ """Run the test for a particular combination of factors.
+
+ The arguments are all strings:
+
+ * `dashm`: Either "" (run the program as a file) or "-m" (run the
+ program as a module).
+
+ * `package`: Either "" (put the source at the top level) or a
+ package name to use to hold the source.
+
+ * `source`: Either "main" or "sub", which file to use as the
+ ``--source`` argument.
+
+ """
+ if env.METACOV:
+ self.skip("Can't test sub-process pth file suppport during metacoverage")
+
+ def fullname(modname):
+ """What is the full module name for `modname` for this test?"""
+ if package and dashm:
+ return '.'.join((package, modname))
+ else:
+ return modname
+
+ def path(basename):
+ """Where should `basename` be created for this test?"""
+ return os.path.join(package, basename)
+
+ # Main will run sub.py.
+ self.make_file(path("main.py"), """\
+ import %s
+ if True: pass
+ """ % fullname('sub'))
+ if package:
+ self.make_file(path("__init__.py"), "")
+ # sub.py will write a few lines.
+ self.make_file(path("sub.py"), """\
+ with open("out.txt", "w") as f:
+ f.write("Hello, world!")
+ """)
+ self.make_file("coverage.ini", """\
+ [run]
+ source = %s
+ """ % fullname(source))
+
+ self.set_environ("COVERAGE_PROCESS_START", "coverage.ini")
+
+ if dashm:
+ cmd = "python -m %s" % fullname('main')
+ else:
+ cmd = "python %s" % path('main.py')
+
+ self.run_command(cmd)
+
+ with open("out.txt") as f:
+ self.assertEqual(f.read(), "Hello, world!")
+
+ # Read the data from .coverage
+ self.assert_exists(".coverage")
+ data = coverage.CoverageData()
+ data.read_file(".coverage")
+ summary = data.line_counts()
+ print(summary)
+ self.assertEqual(summary[source + '.py'], 2)
+ self.assertEqual(len(summary), 1)
+
+ def test_dashm_main(self):
+ self.assert_pth_and_source_work_together('-m', '', 'main')
+
+ def test_script_main(self):
+ self.assert_pth_and_source_work_together('', '', 'main')
+
+ def test_dashm_sub(self):
+ self.assert_pth_and_source_work_together('-m', '', 'sub')
+
+ def test_script_sub(self):
+ self.assert_pth_and_source_work_together('', '', 'sub')
+
+ def test_dashm_pkg_main(self):
+ self.assert_pth_and_source_work_together('-m', 'pkg', 'main')
+
+ def test_script_pkg_main(self):
+ self.assert_pth_and_source_work_together('', 'pkg', 'main')
+
+ def test_dashm_pkg_sub(self):
+ self.assert_pth_and_source_work_together('-m', 'pkg', 'sub')
+
+ def test_script_pkg_sub(self):
+ self.assert_pth_and_source_work_together('', 'pkg', 'sub')
+
+
+def remove_matching_lines(text, pat):
+ """Return `text` with all lines matching `pat` removed."""
+ lines = [l for l in text.splitlines(True) if not re.search(pat, l)]
+ return "".join(lines)
« no previous file with comments | « tools/telemetry/third_party/coverage/tests/test_plugins.py ('k') | tools/telemetry/third_party/coverage/tests/test_python.py » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698