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
127
128
129
130
131
132
133
134
|
from __future__ import annotations
import os
import tempfile
from collections.abc import Generator
from pathlib import Path
from unittest.mock import Mock
import pytest
from _pytest.monkeypatch import MonkeyPatch
from click.testing import CliRunner
import quart.cli
from quart.app import Quart
from quart.cli import AppGroup
from quart.cli import cli
from quart.cli import load_dotenv
from quart.cli import ScriptInfo
@pytest.fixture(scope="module")
def reset_env() -> None:
os.environ.pop("QUART_DEBUG", None)
@pytest.fixture(name="app")
def loadable_app(monkeypatch: MonkeyPatch) -> Mock:
app = Mock(spec=Quart)
app.cli = AppGroup()
module = Mock()
module.app = app
monkeypatch.setattr(quart.cli, "import_module", lambda _: module)
return app
@pytest.fixture(name="dev_app")
def loadable_dev_app(app: Mock) -> Mock:
app.debug = True
return app
@pytest.fixture(name="debug_env")
def debug_env_patch(monkeypatch: MonkeyPatch) -> None:
monkeypatch.setenv("QUART_DEBUG", "true")
@pytest.fixture(name="no_debug_env")
def no_debug_env_patch(monkeypatch: MonkeyPatch) -> None:
monkeypatch.setenv("QUART_DEBUG", "false")
@pytest.fixture(name="empty_cwd")
def empty_cwd() -> Generator[Path, None, None]:
directory = tempfile.TemporaryDirectory()
cwd = os.getcwd()
os.chdir(directory.name)
yield Path(directory.name)
os.chdir(cwd)
directory.cleanup()
def test_script_info_load_app(app: Mock) -> None:
info = ScriptInfo("module:app")
assert info.load_app() == app
def test_version_command() -> None:
runner = CliRunner()
result = runner.invoke(cli, ["--version"])
assert "Quart" in result.output
def test_run_command(app: Mock) -> None:
runner = CliRunner()
runner.invoke(cli, ["--app", "module:app", "run"])
app.run.assert_called_once_with(
debug=False,
host="127.0.0.1",
port=5000,
certfile=None,
keyfile=None,
use_reloader=False,
)
def test_run_command_development_debug_disabled(
dev_app: Mock, no_debug_env: None
) -> None:
runner = CliRunner()
runner.invoke(cli, ["--app", "module:app", "run"])
dev_app.run.assert_called_once_with(
debug=False,
host="127.0.0.1",
port=5000,
certfile=None,
keyfile=None,
use_reloader=False,
)
def test_load_dotenv(empty_cwd: Path) -> None:
value = "dotenv"
with open(empty_cwd / ".env", "w", encoding="utf8") as env:
env.write(f"TEST_ENV_VAR={value}\n")
load_dotenv()
assert os.environ.pop("TEST_ENV_VAR", None) == value
def test_load_dotquartenv(empty_cwd: Path) -> None:
value = "dotquartenv"
with open(empty_cwd / ".quartenv", "w", encoding="utf8") as env:
env.write(f"TEST_ENV_VAR={value}\n")
load_dotenv()
assert os.environ.pop("TEST_ENV_VAR", None) == value
def test_load_dotenv_beats_dotquartenv(empty_cwd: Path) -> None:
env_value = "dotenv"
quartenv_value = "dotquartenv"
with open(empty_cwd / ".env", "w", encoding="utf8") as env:
env.write(f"TEST_ENV_VAR={env_value}\n")
with open(empty_cwd / ".quartenv", "w", encoding="utf8") as env:
env.write(f"TEST_ENV_VAR={quartenv_value}\n")
load_dotenv()
assert os.environ.pop("TEST_ENV_VAR", None) == env_value
|
