| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101 |
- """Unit tests for file system cache."""
- from __future__ import annotations
- import os
- import shutil
- import sys
- import tempfile
- import unittest
- from mypy.fscache import FileSystemCache
- class TestFileSystemCache(unittest.TestCase):
- def setUp(self) -> None:
- self.tempdir = tempfile.mkdtemp()
- self.oldcwd = os.getcwd()
- os.chdir(self.tempdir)
- self.fscache = FileSystemCache()
- def tearDown(self) -> None:
- os.chdir(self.oldcwd)
- shutil.rmtree(self.tempdir)
- def test_isfile_case_1(self) -> None:
- self.make_file("bar.py")
- self.make_file("pkg/sub_package/__init__.py")
- self.make_file("pkg/sub_package/foo.py")
- # Run twice to test both cached and non-cached code paths.
- for i in range(2):
- assert self.isfile_case("bar.py")
- assert self.isfile_case("pkg/sub_package/__init__.py")
- assert self.isfile_case("pkg/sub_package/foo.py")
- assert not self.isfile_case("non_existent.py")
- assert not self.isfile_case("pkg/non_existent.py")
- assert not self.isfile_case("pkg/")
- assert not self.isfile_case("bar.py/")
- for i in range(2):
- assert not self.isfile_case("Bar.py")
- assert not self.isfile_case("pkg/sub_package/__init__.PY")
- assert not self.isfile_case("pkg/Sub_Package/foo.py")
- assert not self.isfile_case("Pkg/sub_package/foo.py")
- def test_isfile_case_2(self) -> None:
- self.make_file("bar.py")
- self.make_file("pkg/sub_package/__init__.py")
- self.make_file("pkg/sub_package/foo.py")
- # Run twice to test both cached and non-cached code paths.
- # This reverses the order of checks from test_isfile_case_1.
- for i in range(2):
- assert not self.isfile_case("Bar.py")
- assert not self.isfile_case("pkg/sub_package/__init__.PY")
- assert not self.isfile_case("pkg/Sub_Package/foo.py")
- assert not self.isfile_case("Pkg/sub_package/foo.py")
- for i in range(2):
- assert self.isfile_case("bar.py")
- assert self.isfile_case("pkg/sub_package/__init__.py")
- assert self.isfile_case("pkg/sub_package/foo.py")
- assert not self.isfile_case("non_existent.py")
- assert not self.isfile_case("pkg/non_existent.py")
- def test_isfile_case_3(self) -> None:
- self.make_file("bar.py")
- self.make_file("pkg/sub_package/__init__.py")
- self.make_file("pkg/sub_package/foo.py")
- # Run twice to test both cached and non-cached code paths.
- for i in range(2):
- assert self.isfile_case("bar.py")
- assert not self.isfile_case("non_existent.py")
- assert not self.isfile_case("pkg/non_existent.py")
- assert not self.isfile_case("Bar.py")
- assert not self.isfile_case("pkg/sub_package/__init__.PY")
- assert not self.isfile_case("pkg/Sub_Package/foo.py")
- assert not self.isfile_case("Pkg/sub_package/foo.py")
- assert self.isfile_case("pkg/sub_package/__init__.py")
- assert self.isfile_case("pkg/sub_package/foo.py")
- def test_isfile_case_other_directory(self) -> None:
- self.make_file("bar.py")
- with tempfile.TemporaryDirectory() as other:
- self.make_file("other_dir.py", base=other)
- self.make_file("pkg/other_dir.py", base=other)
- assert self.isfile_case(os.path.join(other, "other_dir.py"))
- assert not self.isfile_case(os.path.join(other, "Other_Dir.py"))
- assert not self.isfile_case(os.path.join(other, "bar.py"))
- if sys.platform in ("win32", "darwin"):
- # We only check case for directories under our prefix, and since
- # this path is not under the prefix, case difference is fine.
- assert self.isfile_case(os.path.join(other, "PKG/other_dir.py"))
- def make_file(self, path: str, base: str | None = None) -> None:
- if base is None:
- base = self.tempdir
- fullpath = os.path.join(base, path)
- os.makedirs(os.path.dirname(fullpath), exist_ok=True)
- if not path.endswith("/"):
- with open(fullpath, "w") as f:
- f.write("# test file")
- def isfile_case(self, path: str) -> bool:
- return self.fscache.isfile_case(os.path.join(self.tempdir, path), self.tempdir)
|
