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
|
from typing import TYPE_CHECKING
import pytest
from litestar.cli._utils import _format_is_enabled
from litestar.cli.main import litestar_group as cli_command
from tests.unit.test_cli import CREATE_APP_FILE_CONTENT
from tests.unit.test_cli.conftest import CreateAppFileFixture
if TYPE_CHECKING:
from pathlib import Path
from click.testing import CliRunner
from pytest_mock import MockerFixture
def test_format_is_enabled() -> None:
assert _format_is_enabled(0) == "[red]Disabled[/]"
assert _format_is_enabled(False) == "[red]Disabled[/]"
assert _format_is_enabled("") == "[red]Disabled[/]"
assert _format_is_enabled(1) == "[green]Enabled[/]"
assert _format_is_enabled(True) == "[green]Enabled[/]"
assert _format_is_enabled("a") == "[green]Enabled[/]"
@pytest.mark.xdist_group("cli_autodiscovery")
def test_info_command(mocker: "MockerFixture", runner: "CliRunner", app_file: "Path") -> None:
mock = mocker.patch("litestar.cli.commands.core.show_app_info")
result = runner.invoke(cli_command, ["info"])
assert result.exception is None
mock.assert_called_once()
@pytest.mark.xdist_group("cli_autodiscovery")
def test_info_command_with_app_dir(
mocker: "MockerFixture", runner: "CliRunner", create_app_file: CreateAppFileFixture
) -> None:
app_file = "main.py"
app_file_without_extension = app_file.split(".", maxsplit=1)[0]
create_app_file(
file=app_file,
directory="src",
content=CREATE_APP_FILE_CONTENT,
subdir="info_with_app_dir",
init_content=f"from .{app_file_without_extension} import create_app",
)
mock = mocker.patch("litestar.cli.commands.core.show_app_info")
result = runner.invoke(cli_command, ["--app", "info_with_app_dir:create_app", "--app-dir", "src", "info"])
assert result.exception is None
mock.assert_called_once()
@pytest.mark.xdist_group("cli_autodiscovery")
@pytest.mark.parametrize("invalid_app", ["invalid", "info_with_app_dir"])
def test_incorrect_app_argument(
invalid_app: str, mocker: "MockerFixture", runner: "CliRunner", create_app_file: CreateAppFileFixture
) -> None:
app_file = "main.py"
app_file_without_extension = app_file.split(".", maxsplit=1)[0]
create_app_file(
file=app_file,
directory="src",
content=CREATE_APP_FILE_CONTENT,
subdir="info_with_app_dir",
init_content=f"from .{app_file_without_extension} import create_app",
)
mock = mocker.patch("litestar.cli.commands.core.show_app_info")
result = runner.invoke(cli_command, ["--app", invalid_app, "--app-dir", "src", "info"])
assert result.exit_code == 1
mock.assert_not_called()
@pytest.mark.xdist_group("cli_autodiscovery")
def test_invalid_import_in_app_argument(
runner: "CliRunner", create_app_file: CreateAppFileFixture, tmp_project_dir: "Path"
) -> None:
app_file = "main.py"
create_app_file(
file=app_file,
content="from something import bar\n" + CREATE_APP_FILE_CONTENT,
)
app_dir = str(tmp_project_dir.absolute())
result = runner.invoke(cli_command, ["--app", "main:create_app", "--app-dir", app_dir, "info"])
assert isinstance(result.exception, ModuleNotFoundError)
# https://github.com/litestar-org/litestar/issues/4331
@pytest.mark.xdist_group("cli_autodiscovery")
def test_help_option_with_app_dir(
mocker: "MockerFixture", runner: "CliRunner", create_app_file: CreateAppFileFixture
) -> None:
app_file = "main.py"
app_file_without_extension = app_file.split(".", maxsplit=1)[0]
create_app_file(
file=app_file,
directory="src",
content=CREATE_APP_FILE_CONTENT,
subdir="help_with_app_dir",
init_content=f"from .{app_file_without_extension} import create_app",
)
mock_format_help = mocker.patch.object(cli_command, "format_help")
result = runner.invoke(cli_command, ["--app-dir", "docker/neurogate", "--help"])
assert result.exit_code == 0
assert not result.exception
mock_format_help.assert_called_once()
|
