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
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
|
import warnings
import pytest
import cyclopts.core
from cyclopts import App
from cyclopts.core import _log_framework_warning
@pytest.fixture(autouse=True)
def clear_cache():
# Setup
_log_framework_warning.cache_clear()
yield
# Teardown
_log_framework_warning.cache_clear()
def test_app_iter(app):
"""Like a dictionary, __iter__ of an App should yield keys (command names)."""
@app.command
def foo():
pass
@app.command
def bar():
pass
actual = list(app)
assert actual == ["--help", "-h", "--version", "foo", "bar"]
def test_app_iter_with_meta(app):
@app.command
def foo():
pass
@app.command
def bar():
pass
@app.meta.command
def fizz():
pass
actual = list(app)
assert actual == ["--help", "-h", "--version", "foo", "bar"]
actual = list(app.meta)
assert actual == ["--help", "-h", "--version", "fizz", "foo", "bar"]
def test_app_update():
app1 = App()
app2 = App()
@app1.command
def foo():
pass
@app2.command
def bar():
pass
app1.update(app2)
assert list(app1) == ["--help", "-h", "--version", "foo", "bar"]
def test_log_framework_warning_unknown():
# Should not generate a warning for UNKNOWN framework
with warnings.catch_warnings():
warnings.simplefilter("error") # Convert warnings to errors
_log_framework_warning(cyclopts.core.TestFramework.UNKNOWN) # Should not raise
def test_log_framework_warning_pytest(app):
# Should generate a warning when called from non-cyclopts module
with pytest.warns(UserWarning) as warning_records:
try:
app(exit_on_error=False)
except Exception:
pass
assert len(warning_records) == 1
warning_msg = str(warning_records[0].message)
assert (
warning_msg
== 'Cyclopts application invoked without tokens under unit-test framework "pytest". Did you mean "app([])"?'
)
|
