diff options
Diffstat (limited to 'tests/test_files.py')
-rw-r--r-- | tests/test_files.py | 26 |
1 files changed, 23 insertions, 3 deletions
diff --git a/tests/test_files.py b/tests/test_files.py index 00142ac2..62d3e320 100644 --- a/tests/test_files.py +++ b/tests/test_files.py @@ -1,11 +1,13 @@ """Tests for files.py""" -import os, os.path +import os +import os.path +import sys from coverage.files import ( - FileLocator, TreeMatcher, FnmatchMatcher, ModuleMatcher + FileLocator, TreeMatcher, FnmatchMatcher, ModuleMatcher, PathAliases, + find_python_files, abs_file, get_zip_bytes ) -from coverage.files import PathAliases, find_python_files, abs_file from coverage.misc import CoverageException from tests.coveragetest import CoverageTest @@ -247,3 +249,21 @@ class FindPythonFilesTest(CoverageTest): "sub/ssub/__init__.py", "sub/ssub/s.py", "sub/windows.pyw", ]) + + +class GetZipBytesTest(CoverageTest): + """Tests of `get_zip_bytes`.""" + + run_in_temp_dir = False + + def test_get_encoded_zip_files(self): + # See igor.py, do_zipmods, for the text of these files. + zip_file = "tests/zipmods.zip" + sys.path.append(zip_file) # So we can import the files. + for encoding in ["utf8", "gb2312", "hebrew", "shift_jis"]: + filename = zip_file + "/encoded_" + encoding + ".py" + zip_data = get_zip_bytes(filename) + zip_text = zip_data.decode(encoding) + self.assertIn('All OK', zip_text) + # Run the code to see that we really got it encoded properly. + __import__("encoded_"+encoding) |