wehub-resource-sync
137 lines
4.0 KiB
Python
137 lines
4.0 KiB
Python
from pathlib import Path
|
|
|
|
from clint.config import Config
|
|
from clint.index import SymbolIndex
|
|
from clint.linter import Position, Range, lint_file
|
|
from clint.rules.mock_patch_dict_environ import MockPatchDictEnviron
|
|
|
|
|
|
def test_mock_patch_dict_environ_with_string_literal(index: SymbolIndex) -> None:
|
|
code = """
|
|
import os
|
|
from unittest import mock
|
|
|
|
# Bad - string literal
|
|
def test_func():
|
|
with mock.patch.dict("os.environ", {"FOO": "True"}):
|
|
pass
|
|
"""
|
|
config = Config(select={MockPatchDictEnviron.name})
|
|
violations = lint_file(Path("test_file.py"), code, config, index)
|
|
assert len(violations) == 1
|
|
assert all(isinstance(v.rule, MockPatchDictEnviron) for v in violations)
|
|
assert violations[0].range == Range(Position(6, 9))
|
|
|
|
|
|
def test_mock_patch_dict_environ_with_expression(index: SymbolIndex) -> None:
|
|
code = """
|
|
import os
|
|
from unittest import mock
|
|
|
|
# Bad - os.environ as expression
|
|
def test_func():
|
|
with mock.patch.dict(os.environ, {"FOO": "bar"}):
|
|
pass
|
|
"""
|
|
config = Config(select={MockPatchDictEnviron.name})
|
|
violations = lint_file(Path("test_file.py"), code, config, index)
|
|
assert len(violations) == 1
|
|
assert all(isinstance(v.rule, MockPatchDictEnviron) for v in violations)
|
|
assert violations[0].range == Range(Position(6, 9))
|
|
|
|
|
|
def test_mock_patch_dict_environ_as_decorator(index: SymbolIndex) -> None:
|
|
code = """
|
|
import os
|
|
from unittest import mock
|
|
|
|
# Bad - as decorator
|
|
@mock.patch.dict("os.environ", {"FOO": "value"})
|
|
def test_func():
|
|
pass
|
|
"""
|
|
config = Config(select={MockPatchDictEnviron.name})
|
|
violations = lint_file(Path("test_file.py"), code, config, index)
|
|
assert len(violations) == 1
|
|
assert all(isinstance(v.rule, MockPatchDictEnviron) for v in violations)
|
|
assert violations[0].range == Range(Position(5, 1))
|
|
|
|
|
|
def test_mock_patch_dict_environ_with_clear(index: SymbolIndex) -> None:
|
|
code = """
|
|
import os
|
|
from unittest import mock
|
|
|
|
# Bad - with clear=True
|
|
def test_func():
|
|
with mock.patch.dict(os.environ, {}, clear=True):
|
|
pass
|
|
"""
|
|
config = Config(select={MockPatchDictEnviron.name})
|
|
violations = lint_file(Path("test_file.py"), code, config, index)
|
|
assert len(violations) == 1
|
|
assert all(isinstance(v.rule, MockPatchDictEnviron) for v in violations)
|
|
assert violations[0].range == Range(Position(6, 9))
|
|
|
|
|
|
def test_mock_patch_dict_non_environ(index: SymbolIndex) -> None:
|
|
code = """
|
|
from unittest import mock
|
|
|
|
# Good - not os.environ
|
|
def test_func():
|
|
with mock.patch.dict("some.other.dict", {"key": "value"}):
|
|
pass
|
|
"""
|
|
config = Config(select={MockPatchDictEnviron.name})
|
|
violations = lint_file(Path("test_file.py"), code, config, index)
|
|
assert len(violations) == 0
|
|
|
|
|
|
def test_mock_patch_dict_environ_non_test_file(index: SymbolIndex) -> None:
|
|
code = """
|
|
import os
|
|
from unittest import mock
|
|
|
|
# Good - not in test file
|
|
def normal_func():
|
|
with mock.patch.dict("os.environ", {"FOO": "True"}):
|
|
pass
|
|
"""
|
|
config = Config(select={MockPatchDictEnviron.name})
|
|
violations = lint_file(Path("normal_file.py"), code, config, index)
|
|
assert len(violations) == 0
|
|
|
|
|
|
def test_mock_patch_dict_environ_with_mock_alias(index: SymbolIndex) -> None:
|
|
code = """
|
|
import os
|
|
from unittest import mock as mock_lib
|
|
|
|
# Bad - with alias
|
|
def test_func():
|
|
with mock_lib.patch.dict("os.environ", {"FOO": "bar"}):
|
|
pass
|
|
"""
|
|
config = Config(select={MockPatchDictEnviron.name})
|
|
violations = lint_file(Path("test_file.py"), code, config, index)
|
|
assert len(violations) == 1
|
|
assert all(isinstance(v.rule, MockPatchDictEnviron) for v in violations)
|
|
assert violations[0].range == Range(Position(6, 9))
|
|
|
|
|
|
def test_mock_patch_dict_environ_nested_function_not_caught(index: SymbolIndex) -> None:
|
|
code = """
|
|
import os
|
|
from unittest import mock
|
|
|
|
def test_outer():
|
|
def inner_function():
|
|
with mock.patch.dict("os.environ", {"FOO": "True"}):
|
|
pass
|
|
inner_function()
|
|
"""
|
|
config = Config(select={MockPatchDictEnviron.name})
|
|
violations = lint_file(Path("test_file.py"), code, config, index)
|
|
assert len(violations) == 0
|