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
|
# Licensed under the Apache License: http://www.apache.org/licenses/LICENSE-2.0
# For details: https://github.com/nedbat/coveragepy/blob/master/NOTICE.txt
"""Tests of code in tests/mixins.py"""
import pytest
from coverage.misc import import_local_file
from tests.mixins import TempDirMixin, RestoreModulesMixin
class TempDirMixinTest(TempDirMixin):
"""Test the methods in TempDirMixin."""
def file_text(self, fname):
"""Return the text read from a file."""
with open(fname, "rb") as f:
return f.read().decode('ascii')
def test_make_file(self):
# A simple file.
self.make_file("fooey.boo", "Hello there")
assert self.file_text("fooey.boo") == "Hello there"
# A file in a sub-directory
self.make_file("sub/another.txt", "Another")
assert self.file_text("sub/another.txt") == "Another"
# A second file in that sub-directory
self.make_file("sub/second.txt", "Second")
assert self.file_text("sub/second.txt") == "Second"
# A deeper directory
self.make_file("sub/deeper/evenmore/third.txt")
assert self.file_text("sub/deeper/evenmore/third.txt") == ""
# Dedenting
self.make_file("dedented.txt", """\
Hello
Bye
""")
assert self.file_text("dedented.txt") == "Hello\nBye\n"
def test_make_file_newline(self):
self.make_file("unix.txt", "Hello\n")
assert self.file_text("unix.txt") == "Hello\n"
self.make_file("dos.txt", "Hello\n", newline="\r\n")
assert self.file_text("dos.txt") == "Hello\r\n"
self.make_file("mac.txt", "Hello\n", newline="\r")
assert self.file_text("mac.txt") == "Hello\r"
def test_make_file_non_ascii(self):
self.make_file("unicode.txt", "tablo: «ταБℓσ»")
with open("unicode.txt", "rb") as f:
text = f.read()
assert text == b"tablo: \xc2\xab\xcf\x84\xce\xb1\xd0\x91\xe2\x84\x93\xcf\x83\xc2\xbb"
def test_make_bytes_file(self):
self.make_file("binary.dat", bytes=b"\x99\x33\x66hello\0")
with open("binary.dat", "rb") as f:
data = f.read()
assert data == b"\x99\x33\x66hello\0"
class RestoreModulessMixinTest(TempDirMixin, RestoreModulesMixin):
"""Tests of SysPathModulesMixin."""
@pytest.mark.parametrize("val", [17, 42])
def test_module_independence(self, val):
self.make_file("xyzzy.py", f"A = {val}")
import xyzzy # pylint: disable=import-error
assert xyzzy.A == val
def test_cleanup_and_reimport(self):
self.make_file("xyzzy.py", "A = 17")
xyzzy = import_local_file("xyzzy")
assert xyzzy.A == 17
self.clean_local_file_imports()
self.make_file("xyzzy.py", "A = 42")
xyzzy = import_local_file("xyzzy")
assert xyzzy.A == 42
|