(parser_type, prefix)
| 2201 | @pytest.mark.parametrize('parser_type', [pytest.Parser, argparse.ArgumentParser, CliDummyParser]) |
| 2202 | @pytest.mark.parametrize('prefix', ['', 'cfg']) |
| 2203 | def test_cli_user_settings_source(parser_type, prefix): |
| 2204 | class Cfg(BaseSettings): |
| 2205 | pet: Literal['dog', 'cat', 'bird'] = 'bird' |
| 2206 | |
| 2207 | if parser_type is pytest.Parser: |
| 2208 | parser = pytest.Parser(_ispytest=True) |
| 2209 | parse_args = parser.parse |
| 2210 | add_arg = parser.addoption |
| 2211 | cli_cfg_settings = CliSettingsSource( |
| 2212 | Cfg, |
| 2213 | cli_prefix=prefix, |
| 2214 | root_parser=parser, |
| 2215 | parse_args_method=pytest.Parser.parse, |
| 2216 | add_argument_method=pytest.Parser.addoption, |
| 2217 | add_argument_group_method=pytest.Parser.getgroup, |
| 2218 | add_parser_method=None, |
| 2219 | add_subparsers_method=None, |
| 2220 | formatter_class=None, |
| 2221 | ) |
| 2222 | elif parser_type is CliDummyParser: |
| 2223 | parser = CliDummyParser() |
| 2224 | parse_args = parser.parse_args |
| 2225 | add_arg = parser.add_argument |
| 2226 | cli_cfg_settings = CliSettingsSource( |
| 2227 | Cfg, |
| 2228 | cli_prefix=prefix, |
| 2229 | root_parser=parser, |
| 2230 | parse_args_method=CliDummyParser.parse_args, |
| 2231 | add_argument_method=CliDummyParser.add_argument, |
| 2232 | add_argument_group_method=CliDummyParser.add_argument_group, |
| 2233 | add_parser_method=CliDummySubParsers.add_parser, |
| 2234 | add_subparsers_method=CliDummyParser.add_subparsers, |
| 2235 | ) |
| 2236 | else: |
| 2237 | parser = argparse.ArgumentParser() |
| 2238 | parse_args = parser.parse_args |
| 2239 | add_arg = parser.add_argument |
| 2240 | cli_cfg_settings = CliSettingsSource(Cfg, cli_prefix=prefix, root_parser=parser) |
| 2241 | |
| 2242 | add_arg('--fruit', choices=['pear', 'kiwi', 'lime']) |
| 2243 | add_arg('--num-list', action='append', type=int) |
| 2244 | add_arg('--num', type=int) |
| 2245 | |
| 2246 | args = ['--fruit', 'pear', '--num', '0', '--num-list', '1', '--num-list', '2', '--num-list', '3'] |
| 2247 | parsed_args = parse_args(args) |
| 2248 | assert CliApp.run(Cfg, cli_args=parsed_args, cli_settings_source=cli_cfg_settings).model_dump() == {'pet': 'bird'} |
| 2249 | assert CliApp.run(Cfg, cli_args=args, cli_settings_source=cli_cfg_settings).model_dump() == {'pet': 'bird'} |
| 2250 | assert Cfg(_cli_settings_source=cli_cfg_settings(parsed_args=parsed_args)).model_dump() == {'pet': 'bird'} |
| 2251 | assert Cfg(_cli_settings_source=cli_cfg_settings(args=args)).model_dump() == {'pet': 'bird'} |
| 2252 | assert Cfg(_cli_settings_source=cli_cfg_settings(args=False)).model_dump() == {'pet': 'bird'} |
| 2253 | |
| 2254 | arg_prefix = f'{prefix}.' if prefix else '' |
| 2255 | args = [ |
| 2256 | '--fruit', |
| 2257 | 'kiwi', |
| 2258 | '--num', |
| 2259 | '0', |
| 2260 | '--num-list', |
nothing calls this directly
no test coverage detected
searching dependent graphs…