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
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
|
from dataclasses import dataclass
import pytest
from cyclopts.argument import ArgumentCollection
from cyclopts.config import Env
from cyclopts.token import Token
@pytest.fixture
def apps():
"""App is only used as a dictionary in these tests."""
return [{"function1": None}]
def test_config_env_default(apps, monkeypatch):
def foo(bar: int):
pass
argument_collection = ArgumentCollection._from_callable(foo)
monkeypatch.setenv("CYCLOPTS_TEST_APP_BAR", "100")
monkeypatch.setenv("CYCLOPTS_TEST_APP_SOMETHING_ELSE", "100")
Env("CYCLOPTS_TEST_APP_", command=False)(apps, (), argument_collection)
assert len(argument_collection[0].tokens) == 1
assert argument_collection[0].tokens[0].keyword == "CYCLOPTS_TEST_APP_BAR"
assert argument_collection[0].tokens[0].value == "100"
assert argument_collection[0].tokens[0].source == "env"
assert argument_collection[0].tokens[0].index == 0
assert argument_collection[0].tokens[0].keys == ()
def test_config_env_default_already_populated(apps, monkeypatch):
def foo(bar: int):
pass
argument_collection = ArgumentCollection._from_callable(foo)
argument_collection[0].append(Token(keyword="--bar", value="500", source="cli"))
monkeypatch.setenv("CYCLOPTS_TEST_APP_BAR", "100")
monkeypatch.setenv("CYCLOPTS_TEST_APP_SOMETHING_ELSE", "100")
Env("CYCLOPTS_TEST_APP_", command=False)(apps, (), argument_collection)
assert len(argument_collection[0].tokens) == 1
assert argument_collection[0].tokens[0].keyword == "--bar"
assert argument_collection[0].tokens[0].value == "500"
assert argument_collection[0].tokens[0].source == "cli"
assert argument_collection[0].tokens[0].index == 0
assert argument_collection[0].tokens[0].keys == ()
def test_config_env_command_true(apps, monkeypatch):
def foo(bar: int):
pass
argument_collection = ArgumentCollection._from_callable(foo)
monkeypatch.setenv("CYCLOPTS_TEST_APP_FOO_BAR", "100")
Env("CYCLOPTS_TEST_APP_", command=True)(apps, ("foo",), argument_collection)
assert len(argument_collection[0].tokens) == 1
assert argument_collection[0].tokens[0].keyword == "CYCLOPTS_TEST_APP_FOO_BAR"
assert argument_collection[0].tokens[0].value == "100"
assert argument_collection[0].tokens[0].source == "env"
assert argument_collection[0].tokens[0].index == 0
assert argument_collection[0].tokens[0].keys == ()
def test_config_env_dict(apps, monkeypatch):
def foo(bar_bar: dict):
pass
ac = ArgumentCollection._from_callable(foo)
monkeypatch.setenv("CYCLOPTS_TEST_APP_BAR_BAR_BUZZ", "100")
monkeypatch.setenv("CYCLOPTS_TEST_APP_BAR_BAR_FIZZ", "200")
Env("CYCLOPTS_TEST_APP_", command=False)(apps, (), ac)
assert len(ac[0].tokens) == 2
assert ac[0].tokens[0].keyword == "CYCLOPTS_TEST_APP_BAR_BAR_BUZZ"
assert ac[0].tokens[0].value == "100"
assert ac[0].tokens[0].source == "env"
assert ac[0].tokens[0].index == 0
assert ac[0].tokens[0].keys == ("buzz",)
assert ac[0].tokens[1].keyword == "CYCLOPTS_TEST_APP_BAR_BAR_FIZZ"
assert ac[0].tokens[1].value == "200"
assert ac[0].tokens[1].source == "env"
assert ac[0].tokens[1].index == 0
assert ac[0].tokens[1].keys == ("fizz",)
def test_config_env_dataclass(apps, monkeypatch):
@dataclass
class User:
fizz_fizz: int
buzz_buzz: int
def foo(bar_bar: User):
pass
ac = ArgumentCollection._from_callable(foo)
monkeypatch.setenv("CYCLOPTS_TEST_APP_BAR_BAR_BUZZ_BUZZ", "100")
monkeypatch.setenv("CYCLOPTS_TEST_APP_BAR_BAR_FIZZ_FIZZ", "200")
Env("CYCLOPTS_TEST_APP_", command=False)(apps, (), ac)
assert len(ac) == 3
assert len(ac[1].tokens) == 1
assert len(ac[2].tokens) == 1
assert ac[1].tokens[0].keyword == "CYCLOPTS_TEST_APP_BAR_BAR_FIZZ_FIZZ"
assert ac[1].tokens[0].value == "200"
assert ac[1].tokens[0].source == "env"
assert ac[1].tokens[0].index == 0
assert ac[1].tokens[0].keys == ()
assert ac[2].tokens[0].keyword == "CYCLOPTS_TEST_APP_BAR_BAR_BUZZ_BUZZ"
assert ac[2].tokens[0].value == "100"
assert ac[2].tokens[0].source == "env"
assert ac[2].tokens[0].index == 0
assert ac[2].tokens[0].keys == ()
|
