wehub-resource-sync
59 lines
1.7 KiB
Python
59 lines
1.7 KiB
Python
from pathlib import Path
|
|
|
|
import pytest
|
|
from clint.config import Config
|
|
from clint.index import SymbolIndex
|
|
from clint.linter import lint_file
|
|
from clint.rules.use_gh_token import UseGhToken
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"code",
|
|
[
|
|
pytest.param(
|
|
'import os\n\ntoken = os.getenv("GITHUB_TOKEN")',
|
|
id="os.getenv",
|
|
),
|
|
pytest.param(
|
|
'import os\n\ntoken = os.environ.get("GITHUB_TOKEN")',
|
|
id="os.environ.get",
|
|
),
|
|
pytest.param(
|
|
'import os\n\ntoken = os.getenv("GITHUB_TOKEN", "default")',
|
|
id="os.getenv with default",
|
|
),
|
|
pytest.param(
|
|
'import os\n\ntoken = os.environ.get("GITHUB_TOKEN", None)',
|
|
id="os.environ.get with default",
|
|
),
|
|
],
|
|
)
|
|
def test_violation(code: str, index: SymbolIndex) -> None:
|
|
config = Config(select={UseGhToken.name})
|
|
violations = lint_file(Path("file.py"), code, config, index)
|
|
assert len(violations) == 1
|
|
assert isinstance(violations[0].rule, UseGhToken)
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"code",
|
|
[
|
|
pytest.param(
|
|
'import os\n\ntoken = os.getenv("GH_TOKEN")',
|
|
id="os.getenv with GH_TOKEN",
|
|
),
|
|
pytest.param(
|
|
'import os\n\ntoken = os.environ.get("GH_TOKEN")',
|
|
id="os.environ.get with GH_TOKEN",
|
|
),
|
|
pytest.param(
|
|
'import os\n\ntoken = os.getenv("OTHER_VAR")',
|
|
id="os.getenv with other var",
|
|
),
|
|
],
|
|
)
|
|
def test_no_violation(code: str, index: SymbolIndex) -> None:
|
|
config = Config(select={UseGhToken.name})
|
|
violations = lint_file(Path("file.py"), code, config, index)
|
|
assert len(violations) == 0
|