Coverage for tests / test_python.py: 100.000%

1# Licensed under the Apache License: http://www.apache.org/licenses/LICENSE-2.0 

2# For details: https://github.com/coveragepy/coveragepy/blob/main/NOTICE.txt 

3 

4"""Tests of coverage/python.py""" 

5 

6from __future__ import annotations 69 ctx1xyzAabBCDEcdFGHIefJKLMghNOPQijRSTUVWklmXYZ012nop345678qrs9!#$%'tuv()w

7 

8import pathlib 69 ctx1xyzAabBCDEcdFGHIefJKLMghNOPQijRSTUVWklmXYZ012nop345678qrs9!#$%'tuv()w

9import sys 69 ctx1xyzAabBCDEcdFGHIefJKLMghNOPQijRSTUVWklmXYZ012nop345678qrs9!#$%'tuv()w

10 

11import pytest 69 ctx1xyzAabBCDEcdFGHIefJKLMghNOPQijRSTUVWklmXYZ012nop345678qrs9!#$%'tuv()w

12 

13from coverage import env 69 ctx1xyzAabBCDEcdFGHIefJKLMghNOPQijRSTUVWklmXYZ012nop345678qrs9!#$%'tuv()w

14from coverage.python import get_zip_bytes, source_for_file 69 ctx1xyzAabBCDEcdFGHIefJKLMghNOPQijRSTUVWklmXYZ012nop345678qrs9!#$%'tuv()w

15 

16from tests.coveragetest import CoverageTest 69 ctx1xyzAabBCDEcdFGHIefJKLMghNOPQijRSTUVWklmXYZ012nop345678qrs9!#$%'tuv()w

17from tests.helpers import os_sep 69 ctx1xyzAabBCDEcdFGHIefJKLMghNOPQijRSTUVWklmXYZ012nop345678qrs9!#$%'tuv()w

18 

19 

20@pytest.mark.xdist_group(name="get_zip_bytes_test") 69 ctx1xyzAabBCDEcdFGHIefJKLMghNOPQijRSTUVWklmXYZ012nop345678qrs9!#$%'tuv()w

21class GetZipBytesTest(CoverageTest): 69 ctx1xyzAabBCDEcdFGHIefJKLMghNOPQijRSTUVWklmXYZ012nop345678qrs9!#$%'tuv()w

22 """Tests of `get_zip_bytes`.""" 

23 

24 run_in_temp_dir = False 69 ctx1xyzAabBCDEcdFGHIefJKLMghNOPQijRSTUVWklmXYZ012nop345678qrs9!#$%'tuv()w

25 

26 @pytest.mark.parametrize( 69 ctx1xyzAabBCDEcdFGHIefJKLMghNOPQijRSTUVWklmXYZ012nop345678qrs9!#$%'tuv()w

27 "encoding", 

28 ["utf-8", "gb2312", "hebrew", "shift_jis", "cp1252"], 

29 ) 

30 def test_get_encoded_zip_files(self, encoding: str) -> None: 69 ctx1xyzAabBCDEcdFGHIefJKLMghNOPQijRSTUVWklmXYZ012nop345678qrs9!#$%'tuv()w

31 # See igor.py, do_zipmods, for the text of these files. 

32 zip_file = "tests/zipmods.zip" 69 ctx1xyzAabBCDEcdFGHIefJKLMghNOPQijRSTUVWklmXYZ012nop345678qrs9!#$%'tuv()w

33 sys.path.append(zip_file) # So we can import the files. 69 ctx1xyzAabBCDEcdFGHIefJKLMghNOPQijRSTUVWklmXYZ012nop345678qrs9!#$%'tuv()w

34 filename = zip_file + "/encoded_" + encoding + ".py" 69 ctx1xyzAabBCDEcdFGHIefJKLMghNOPQijRSTUVWklmXYZ012nop345678qrs9!#$%'tuv()w

35 filename = os_sep(filename) 69 ctx1xyzAabBCDEcdFGHIefJKLMghNOPQijRSTUVWklmXYZ012nop345678qrs9!#$%'tuv()w

36 zip_data = get_zip_bytes(filename) 69 ctx1xyzAabBCDEcdFGHIefJKLMghNOPQijRSTUVWklmXYZ012nop345678qrs9!#$%'tuv()w

37 assert zip_data is not None 69 ctx1xyzAabBCDEcdFGHIefJKLMghNOPQijRSTUVWklmXYZ012nop345678qrs9!#$%'tuv()w

38 zip_text = zip_data.decode(encoding) 69 ctx1xyzAabBCDEcdFGHIefJKLMghNOPQijRSTUVWklmXYZ012nop345678qrs9!#$%'tuv()w

39 assert "All OK" in zip_text 69 ctx1xyzAabBCDEcdFGHIefJKLMghNOPQijRSTUVWklmXYZ012nop345678qrs9!#$%'tuv()w

40 # Run the code to see that we really got it encoded properly. 

41 mod = __import__("encoded_" + encoding) 69 ctx1xyzAabBCDEcdFGHIefJKLMghNOPQijRSTUVWklmXYZ012nop345678qrs9!#$%'tuv()w

42 assert mod.encoding == encoding 69 ctx1xyzAabBCDEcdFGHIefJKLMghNOPQijRSTUVWklmXYZ012nop345678qrs9!#$%'tuv()w

43 

44 

45def test_source_for_file(tmp_path: pathlib.Path) -> None: 69 ctx1xyzAabBCDEcdFGHIefJKLMghNOPQijRSTUVWklmXYZ012nop345678qrs9!#$%'tuv()w

46 src = str(tmp_path / "a.py") 69 ctx1xyzAabBCDEcdFGHIefJKLMghNOPQijRSTUVWklmXYZ012nop345678qrs9!#$%'tuv()w

47 assert source_for_file(src) == src 69 ctx1xyzAabBCDEcdFGHIefJKLMghNOPQijRSTUVWklmXYZ012nop345678qrs9!#$%'tuv()w

48 assert source_for_file(src + "c") == src 69 ctx1xyzAabBCDEcdFGHIefJKLMghNOPQijRSTUVWklmXYZ012nop345678qrs9!#$%'tuv()w

49 assert source_for_file(src + "o") == src 69 ctx1xyzAabBCDEcdFGHIefJKLMghNOPQijRSTUVWklmXYZ012nop345678qrs9!#$%'tuv()w

50 unknown = src + "FOO" 69 ctx1xyzAabBCDEcdFGHIefJKLMghNOPQijRSTUVWklmXYZ012nop345678qrs9!#$%'tuv()w

51 assert source_for_file(unknown) == unknown 69 ctx1xyzAabBCDEcdFGHIefJKLMghNOPQijRSTUVWklmXYZ012nop345678qrs9!#$%'tuv()w

52 

53 

54@pytest.mark.skipif(not env.WINDOWS, reason="not windows") 69 ctx1xyzAabBCDEcdFGHIefJKLMghNOPQijRSTUVWklmXYZ012nop345678qrs9!#$%'tuv()w

55def test_source_for_file_windows(tmp_path: pathlib.Path) -> None: 69 ctx1xyzAabBCDEcdFGHIefJKLMghNOPQijRSTUVWklmXYZ012nop345678qrs9!#$%'tuv()w

56 a_py = tmp_path / "a.py" 23 ctx1abcdefghijklmnopqrstuvw

57 src = str(a_py) 23 ctx1abcdefghijklmnopqrstuvw

58 

59 # On windows if a pyw exists, it is an acceptable source 

60 path_windows = tmp_path / "a.pyw" 23 ctx1abcdefghijklmnopqrstuvw

61 path_windows.write_text("", encoding="utf-8") 23 ctx1abcdefghijklmnopqrstuvw

62 assert str(path_windows) == source_for_file(src + "c") 23 ctx1abcdefghijklmnopqrstuvw

63 

64 # If both pyw and py exist, py is preferred 

65 a_py.write_text("", encoding="utf-8") 23 ctx1abcdefghijklmnopqrstuvw

66 assert source_for_file(src + "c") == src 23 ctx1abcdefghijklmnopqrstuvw

67 

68 

69class RunpyTest(CoverageTest): 69 ctx1xyzAabBCDEcdFGHIefJKLMghNOPQijRSTUVWklmXYZ012nop345678qrs9!#$%'tuv()w

70 """Tests using runpy.""" 

71 

72 @pytest.mark.parametrize("convert_to", ["str", "Path"]) 69 ctx1xyzAabBCDEcdFGHIefJKLMghNOPQijRSTUVWklmXYZ012nop345678qrs9!#$%'tuv()w

73 def test_runpy_path(self, convert_to: str) -> None: 69 ctx1xyzAabBCDEcdFGHIefJKLMghNOPQijRSTUVWklmXYZ012nop345678qrs9!#$%'tuv()w

74 # Ensure runpy.run_path(path) works when path is pathlib.Path or str. 

75 # 

76 # runpy.run_path(pathlib.Path(...)) causes __file__ to be a Path, 

77 # which may make source_for_file() stumble (#1819) with: 

78 # 

79 # AttributeError: 'PosixPath' object has no attribute 'endswith' 

80 

81 self.check_coverage(f"""\ 69 ctx1xyzAabBCDEcdFGHIefJKLMghNOPQijRSTUVWklmXYZ012nop345678qrs9!#$%'tuv()w

82 import runpy 

83 from pathlib import Path 

84 pyfile = Path('script.py') 

85 pyfile.write_text('', encoding='utf-8') 

86 runpy.run_path({convert_to}(pyfile)) 

87 """)