Handle 404

Author: patrick91

src/fastapi_cloud_cli/commands/logs.py

@@ -164,11 +164,17 @@ def _process_log_stream(
                 toolkit.print(
                     "The specified token is not valid. Use [blue]`fastapi login`[/] to generate a new token.",
                 )
+            if isinstance(e, HTTPStatusError) and e.response.status_code == 404:
+                toolkit.print(
+                    "App not found. Make sure to use the correct account.",
+                )
             elif isinstance(e, ReadTimeout):
                 toolkit.print(
                     "The request timed out. Please try again later.",
                 )
             else:
+                logger.exception("Failed to fetch logs")
+
                 toolkit.print(
                     "Failed to fetch logs. Please try again later.",
                 )

tests/test_logs.py

@@ -169,6 +169,21 @@ def test_handles_401_unauthorized(
     assert "token is not valid" in result.output
 
 
+@pytest.mark.respx(base_url=settings.base_api_url)
+def test_handles_404(
+    logged_in_cli: None, respx_mock: respx.MockRouter, configured_app: ConfiguredApp
+) -> None:
+    respx_mock.get(url__regex=rf"/apps/{configured_app.app_id}/logs/stream.*").mock(
+        return_value=httpx.Response(404)
+    )
+
+    with changing_dir(configured_app.path):
+        result = runner.invoke(app, ["logs", "--no-follow"])
+
+    assert result.exit_code == 1
+    assert "App not found" in result.output
+
+
 @pytest.mark.respx(base_url=settings.base_api_url)
 def test_handles_500_server_error(
     logged_in_cli: None, respx_mock: respx.MockRouter, configured_app: ConfiguredApp