1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
|
import click
import pytest
from check_jsonschema.cli.warnings import deprecation_warning_callback
@click.command("foo")
@click.option(
"--bar",
is_flag=True,
callback=deprecation_warning_callback("--bar", is_flag=True),
)
@click.option(
"--baz",
callback=deprecation_warning_callback(
"--baz", append_message="Use --frob instead!"
),
)
def mycli(bar, baz):
print(bar)
if baz:
print(baz)
def test_deprecation_warning_callback_on_missing_opts(cli_runner):
result = cli_runner.invoke(mycli, [])
assert result.exit_code == 0
assert result.stdout == "False\n"
def test_deprecation_warning_callback_on_flag(cli_runner):
with pytest.warns(
UserWarning,
match="'--bar' is deprecated and will be removed in a future release",
):
result = cli_runner.invoke(mycli, ["--bar"], catch_exceptions=False)
assert result.exit_code == 0, result.stdout
assert result.stdout == "True\n"
def test_deprecation_warning_callback_added_message(cli_runner):
with pytest.warns(
UserWarning,
match=(
"'--baz' is deprecated and will be removed in a future release. "
"Use --frob instead!"
),
):
result = cli_runner.invoke(mycli, ["--baz", "ok"], catch_exceptions=False)
assert result.exit_code == 0, result.stdout
assert result.stdout == "False\nok\n"
|
