tests: conditional printing api docs urls

Author: FlavienRx

tests/assets/single_file_docs.py

@@ -0,0 +1,48 @@
+from fastapi import FastAPI
+
+no_openapi = FastAPI(openapi_url=None)
+
+
+@no_openapi.get("/")
+def no_openapi_root():
+    return {"message": "single file no_openapi"}
+
+
+none_docs = FastAPI(docs_url=None, redoc_url=None)
+
+
+@none_docs.get("/")
+def none_docs_root():
+    return {"message": "single file none_docs"}
+
+
+no_docs = FastAPI(docs_url=None)
+
+
+@no_docs.get("/")
+def no_docs_root():
+    return {"message": "single file no_docs"}
+
+
+no_redoc = FastAPI(redoc_url=None)
+
+
+@no_redoc.get("/")
+def no_redoc_root():
+    return {"message": "single file no_redoc"}
+
+
+full_docs = FastAPI()
+
+
+@full_docs.get("/")
+def full_docs_root():
+    return {"message": "single file full_docs"}
+
+
+custom_docs = FastAPI(docs_url="/custom-docs-url", redoc_url="/custom-redoc-url")
+
+
+@custom_docs.get("/")
+def custom_docs_root():
+    return {"message": "single file custom_docs"}

tests/test_cli.py

@@ -349,6 +349,84 @@ def test_run_env_vars_and_args() -> None:
         assert "Documentation at http://0.0.0.0:8080/docs" in result.output
 
 
+def test_no_openapi() -> None:
+    with changing_dir(assets_path):
+        with patch.object(uvicorn, "run") as mock_run:
+            result = runner.invoke(
+                app, ["dev", "single_file_docs.py", "--app", "no_openapi"]
+            )
+            assert result.exit_code == 0, result.output
+            assert mock_run.called
+
+        assert "http://127.0.0.1:8000/docs" not in result.output
+        assert "http://127.0.0.1:8000/redoc" not in result.output
+
+
+def test_none_docs() -> None:
+    with changing_dir(assets_path):
+        with patch.object(uvicorn, "run") as mock_run:
+            result = runner.invoke(
+                app, ["dev", "single_file_docs.py", "--app", "none_docs"]
+            )
+            assert result.exit_code == 0, result.output
+            assert mock_run.called
+
+        assert "http://127.0.0.1:8000/docs" not in result.output
+        assert "http://127.0.0.1:8000/redoc" not in result.output
+
+
+def test_no_docs() -> None:
+    with changing_dir(assets_path):
+        with patch.object(uvicorn, "run") as mock_run:
+            result = runner.invoke(
+                app, ["dev", "single_file_docs.py", "--app", "no_docs"]
+            )
+            assert result.exit_code == 0, result.output
+            assert mock_run.called
+
+        assert "http://127.0.0.1:8000/redoc" in result.output
+        assert "http://127.0.0.1:8000/docs" not in result.output
+
+
+def test_no_redoc() -> None:
+    with changing_dir(assets_path):
+        with patch.object(uvicorn, "run") as mock_run:
+            result = runner.invoke(
+                app, ["dev", "single_file_docs.py", "--app", "no_redoc"]
+            )
+            assert result.exit_code == 0, result.output
+            assert mock_run.called
+
+        assert "http://127.0.0.1:8000/docs" in result.output
+        assert "http://127.0.0.1:8000/redocs" not in result.output
+
+
+def test_full_docs() -> None:
+    with changing_dir(assets_path):
+        with patch.object(uvicorn, "run") as mock_run:
+            result = runner.invoke(
+                app, ["dev", "single_file_docs.py", "--app", "full_docs"]
+            )
+            assert result.exit_code == 0, result.output
+            assert mock_run.called
+
+        assert "http://127.0.0.1:8000/docs" in result.output
+        assert "http://127.0.0.1:8000/redoc" in result.output
+
+
+def test_custom_docs() -> None:
+    with changing_dir(assets_path):
+        with patch.object(uvicorn, "run") as mock_run:
+            result = runner.invoke(
+                app, ["dev", "single_file_docs.py", "--app", "custom_docs"]
+            )
+            assert result.exit_code == 0, result.output
+            assert mock_run.called
+
+        assert "http://127.0.0.1:8000/custom-docs-url" in result.output
+        assert "http://127.0.0.1:8000/custom-redoc-url" in result.output
+
+
 def test_run_error() -> None:
     with changing_dir(assets_path):
         result = runner.invoke(app, ["run", "non_existing_file.py"])

tests/test_discover.py

@@ -7,18 +7,20 @@
     get_import_data_from_import_string,
 )
 from fastapi_cli.exceptions import FastAPICLIException
+from tests.utils import changing_dir
 
 assets_path = Path(__file__).parent / "assets"
 
 
 def test_get_import_data_from_import_string_valid() -> None:
-    result = get_import_data_from_import_string("module.submodule:app")
+    with changing_dir(assets_path):
+        result = get_import_data_from_import_string("package.mod.app:app")
 
     assert isinstance(result, ImportData)
     assert result.app_name == "app"
-    assert result.import_string == "module.submodule:app"
-    assert result.module_data.module_import_str == "module.submodule"
-    assert result.module_data.extra_sys_path == Path(".").resolve()
+    assert result.import_string == "package.mod.app:app"
+    assert result.module_data.module_import_str == "package.mod.app"
+    assert result.module_data.extra_sys_path == Path(assets_path).resolve()
     assert result.module_data.module_paths == []