Coverage for tests/test_new.py: 100%

1import shutil 8 ctx1abcdefgh

2import subprocess 8 ctx1abcdefgh

3from pathlib import Path 8 ctx1abcdefgh

4from typing import Any 8 ctx1abcdefgh

5 

6import pytest 8 ctx1abcdefgh

7from typer.testing import CliRunner 8 ctx1abcdefgh

8 

9from fastapi_new.cli import app 8 ctx1abcdefgh

10 

11runner = CliRunner() 8 ctx1abcdefgh

12 

13 

14@pytest.fixture 8 ctx1abcdefgh

15def temp_project_dir(tmp_path: Path, monkeypatch: Any) -> Path: 8 ctx1abcdefgh

16 """Create a temporary directory and cd into it.""" 

17 monkeypatch.chdir(tmp_path) 8 ctx1abcdefgh

18 return tmp_path 8 ctx1abcdefgh

19 

20 

21def _assert_project_created(project_path: Path) -> None: 8 ctx1abcdefgh

22 assert (project_path / "main.py").exists() 48 ctx1yiWXzYAjZ0B1Ck23D4El56F7Gm89H!In#$J%Ko'(L)Mp*+N,

23 assert (project_path / "README.md").exists() 48 ctx1yiWXzYAjZ0B1Ck23D4El56F7Gm89H!In#$J%Ko'(L)Mp*+N,

24 assert (project_path / "pyproject.toml").exists() 48 ctx1yiWXzYAjZ0B1Ck23D4El56F7Gm89H!In#$J%Ko'(L)Mp*+N,

25 

26 

27def test_creates_project_successfully(temp_project_dir: Path) -> None: 8 ctx1abcdefgh

28 result = runner.invoke(app, ["my_fastapi_project"]) 8 ctx1yACEGIKM

29 

30 assert result.exit_code == 0 8 ctx1yACEGIKM

31 project_path = temp_project_dir / "my_fastapi_project" 8 ctx1yACEGIKM

32 _assert_project_created(project_path) 8 ctx1yACEGIKM

33 assert "Success!" in result.output 8 ctx1yACEGIKM

34 assert "my_fastapi_project" in result.output 8 ctx1yACEGIKM

35 

36 

37def test_creates_project_with_python_version(temp_project_dir: Path) -> None: 8 ctx1abcdefgh

38 # Test long form 

39 result = runner.invoke(app, ["project_long", "--python", "3.12"]) 8 ctx1ijklmnop

40 assert result.exit_code == 0 8 ctx1ijklmnop

41 project_path = temp_project_dir / "project_long" 8 ctx1ijklmnop

42 _assert_project_created(project_path) 8 ctx1ijklmnop

43 assert "3.12" in (project_path / "pyproject.toml").read_text() 8 ctx1ijklmnop

44 

45 # Test short form 

46 result = runner.invoke(app, ["project_short", "-p", "3.11"]) 8 ctx1ijklmnop

47 assert result.exit_code == 0 8 ctx1ijklmnop

48 project_path = temp_project_dir / "project_short" 8 ctx1ijklmnop

49 assert "3.11" in (project_path / "pyproject.toml").read_text() 8 ctx1ijklmnop

50 

51 

52def test_validates_template_file_contents(temp_project_dir: Path) -> None: 8 ctx1abcdefgh

53 result = runner.invoke(app, ["sample_project"]) 8 ctx1qrstuvwx

54 assert result.exit_code == 0 8 ctx1qrstuvwx

55 

56 project_path = temp_project_dir / "sample_project" 8 ctx1qrstuvwx

57 

58 main_py_content = (project_path / "main.py").read_text() 8 ctx1qrstuvwx

59 assert "from fastapi import FastAPI" in main_py_content 8 ctx1qrstuvwx

60 assert "app = FastAPI()" in main_py_content 8 ctx1qrstuvwx

61 

62 # Check README.md 

63 readme_content = (project_path / "README.md").read_text() 8 ctx1qrstuvwx

64 assert "# sample_project" in readme_content 8 ctx1qrstuvwx

65 assert "A project created with FastAPI" in readme_content 8 ctx1qrstuvwx

66 

67 # Check pyproject.toml 

68 pyproject_content = (project_path / "pyproject.toml").read_text() 8 ctx1qrstuvwx

69 assert 'name = "sample-project"' in pyproject_content 8 ctx1qrstuvwx

70 assert "fastapi[standard]" in pyproject_content 8 ctx1qrstuvwx

71 

72 

73def test_initializes_in_current_directory(temp_project_dir: Path) -> None: 8 ctx1abcdefgh

74 result = runner.invoke(app, []) 8 ctx1X0369$(+

75 

76 assert result.exit_code == 0 8 ctx1X0369$(+

77 assert "Initializing in current directory" in result.output 8 ctx1X0369$(+

78 _assert_project_created(temp_project_dir) 8 ctx1X0369$(+

79 

80 

81def test_initializes_in_current_directory_with_dot(temp_project_dir: Path) -> None: 8 ctx1abcdefgh

82 result = runner.invoke(app, ["."]) 8 ctx1zBDFHJLN

83 

84 assert result.exit_code == 0 8 ctx1zBDFHJLN

85 assert "Initializing in current directory" in result.output 8 ctx1zBDFHJLN

86 _assert_project_created(temp_project_dir) 8 ctx1zBDFHJLN

87 

88 readme_content = (temp_project_dir / "README.md").read_text() 8 ctx1zBDFHJLN

89 assert f"# {temp_project_dir.name}" in readme_content 8 ctx1zBDFHJLN

90 

91 

92def test_rejects_existing_directory(temp_project_dir: Path) -> None: 8 ctx1abcdefgh

93 existing_dir = temp_project_dir / "existing_project" 8 ctx2~ abbbcbdbebfbgb

94 existing_dir.mkdir() 8 ctx2~ abbbcbdbebfbgb

95 

96 result = runner.invoke(app, ["existing_project"]) 8 ctx2~ abbbcbdbebfbgb

97 assert result.exit_code == 1 8 ctx2~ abbbcbdbebfbgb

98 assert "Directory 'existing_project' already exists." in result.output 8 ctx2~ abbbcbdbebfbgb

99 

100 

101def test_rejects_python_below_3_10(temp_project_dir: Path) -> None: 8 ctx1abcdefgh

102 result = runner.invoke(app, ["test_project", "--python", "3.9"]) 8 ctx2pbqbrbsbtbubvbwb

103 assert result.exit_code == 1 8 ctx2pbqbrbsbtbubvbwb

104 assert "Python 3.9 is not supported" in result.output 8 ctx2pbqbrbsbtbubvbwb

105 assert "FastAPI requires Python 3.10" in result.output 8 ctx2pbqbrbsbtbubvbwb

106 

107 

108def test_passes_single_digit_python_version_to_uv(temp_project_dir: Path) -> None: 8 ctx1abcdefgh

109 result = runner.invoke(app, ["test_project", "--python", "3"]) 8 ctx1Y147!%),

110 assert result.exit_code == 0 8 ctx1Y147!%),

111 project_path = temp_project_dir / "test_project" 8 ctx1Y147!%),

112 _assert_project_created(project_path) 8 ctx1Y147!%),

113 

114 

115def test_passes_malformed_python_version_to_uv(temp_project_dir: Path) -> None: 8 ctx1abcdefgh

116 result = runner.invoke(app, ["test_project", "--python", "abc.def"]) 8 ctx2xbybzbAbBbCbDbEb

117 # uv will reject this, we just verify we don't crash during validation 

118 assert result.exit_code == 1 8 ctx2xbybzbAbBbCbDbEb

119 

120 

121def test_creates_project_without_python_flag(temp_project_dir: Path) -> None: 8 ctx1abcdefgh

122 result = runner.invoke(app, ["test_project"]) 8 ctx1WZ258#'*

123 assert result.exit_code == 0 8 ctx1WZ258#'*

124 project_path = temp_project_dir / "test_project" 8 ctx1WZ258#'*

125 _assert_project_created(project_path) 8 ctx1WZ258#'*

126 

127 

128def test_failed_to_initialize_with_uv(monkeypatch: Any) -> None: 8 ctx1abcdefgh

129 def mock_run(*args: Any, **kwargs: Any) -> None: 8 ctx1-./:;=?@

130 # Let the first check for 'uv' succeed, but fail on 'uv init' 

131 if args[0][0] == "uv" and args[0][1] == "init": 8 ctx1-./:;=?@

132 raise subprocess.CalledProcessError( 8 ctx1-./:;=?@

133 1, args[0], stderr=b"uv init failed for some reason" 

134 ) 

135 

136 monkeypatch.setattr(subprocess, "run", mock_run) 8 ctx1-./:;=?@

137 

138 result = runner.invoke(app, ["failing_project"]) 8 ctx1-./:;=?@

139 assert result.exit_code == 1 8 ctx1-./:;=?@

140 assert "Failed to initialize project with uv" in result.output 8 ctx1-./:;=?@

141 

142 

143def test_failed_to_add_dependencies(temp_project_dir: Path, monkeypatch: Any) -> None: 8 ctx1abcdefgh

144 def mock_run(*args: Any, **kwargs: Any) -> None: 8 ctx1[]^_`{|}

145 # Let 'uv init' succeed, but fail on 'uv add' 

146 if args[0][0] == "uv" and args[0][1] == "add": 8 ctx1[]^_`{|}

147 raise subprocess.CalledProcessError( 8 ctx1[]^_`{|}

148 1, args[0], stderr=b"Failed to resolve dependencies" 

149 ) 

150 

151 monkeypatch.setattr(subprocess, "run", mock_run) 8 ctx1[]^_`{|}

152 

153 result = runner.invoke(app, ["failing_deps"]) 8 ctx1[]^_`{|}

154 assert result.exit_code == 1 8 ctx1[]^_`{|}

155 assert "Failed to install dependencies" in result.output 8 ctx1[]^_`{|}

156 

157 

158def test_file_write_failure(temp_project_dir: Path, monkeypatch: Any) -> None: 8 ctx1abcdefgh

159 original_write_text = Path.write_text 8 ctx1OPQRSTUV

160 

161 def mock_write_text(self: Path, *args: Any, **kwargs: Any) -> None: 8 ctx1OPQRSTUV

162 # Fail when trying to write README.md (let main.py succeed first) 

163 if self.name == "README.md": 8 ctx1OPQRSTUV

164 raise PermissionError("Permission denied") 8 ctx1OPQRSTUV

165 original_write_text(self, *args, **kwargs) 8 ctx1OPQRSTUV

166 

167 monkeypatch.setattr(Path, "write_text", mock_write_text) 8 ctx1OPQRSTUV

168 

169 result = runner.invoke(app, ["test_write_fail"]) 8 ctx1OPQRSTUV

170 assert result.exit_code == 1 8 ctx1OPQRSTUV

171 assert "Failed to write template files" in result.output 8 ctx1OPQRSTUV

172 

173 

174def test_uv_not_installed(temp_project_dir: Path, monkeypatch: Any) -> None: 8 ctx1abcdefgh

175 monkeypatch.setattr(shutil, "which", lambda _: None) 8 ctx2hbibjbkblbmbnbob

176 

177 result = runner.invoke(app, ["test_uv_missing_project"]) 8 ctx2hbibjbkblbmbnbob

178 assert result.exit_code == 1 8 ctx2hbibjbkblbmbnbob

179 assert "uv is required to create new projects" in result.output 8 ctx2hbibjbkblbmbnbob

180 assert "https://docs.astral.sh/uv/" in result.output 8 ctx2hbibjbkblbmbnbob