testing/test_nodes: fix fake session to be more accurate

The type of _initialpaths is `FrozenSet[Path]`.
This commit is contained in:
Ran Benita 2020-12-19 14:06:17 +02:00
parent 042d12fae6
commit 7aa2240832
1 changed files with 7 additions and 6 deletions

View File

@ -1,3 +1,4 @@
from pathlib import Path
from typing import cast from typing import cast
from typing import List from typing import List
from typing import Type from typing import Type
@ -69,23 +70,23 @@ def test_node_warning_enforces_warning_types(pytester: Pytester) -> None:
def test__check_initialpaths_for_relpath() -> None: def test__check_initialpaths_for_relpath() -> None:
"""Ensure that it handles dirs, and does not always use dirname.""" """Ensure that it handles dirs, and does not always use dirname."""
cwd = py.path.local() cwd = Path.cwd()
class FakeSession1: class FakeSession1:
_initialpaths = [cwd] _initialpaths = frozenset({cwd})
session = cast(pytest.Session, FakeSession1) session = cast(pytest.Session, FakeSession1)
assert nodes._check_initialpaths_for_relpath(session, cwd) == "" assert nodes._check_initialpaths_for_relpath(session, py.path.local(cwd)) == ""
sub = cwd.join("file") sub = cwd / "file"
class FakeSession2: class FakeSession2:
_initialpaths = [cwd] _initialpaths = frozenset({cwd})
session = cast(pytest.Session, FakeSession2) session = cast(pytest.Session, FakeSession2)
assert nodes._check_initialpaths_for_relpath(session, sub) == "file" assert nodes._check_initialpaths_for_relpath(session, py.path.local(sub)) == "file"
outside = py.path.local("/outside") outside = py.path.local("/outside")
assert nodes._check_initialpaths_for_relpath(session, outside) is None assert nodes._check_initialpaths_for_relpath(session, outside) is None