Fix default values

fastapi · Sep 11, 2024 · 8312bb8 · 8312bb8
1 parent b77a29f
commit 8312bb8
Show file tree

Hide file tree

Showing 4 changed files with 15 additions and 3 deletions.
diff --git a/docs_src/parameter_types/enum/tutorial004.py b/docs_src/parameter_types/enum/tutorial004.py
@@ -10,7 +10,7 @@ class LogLevel(enum.Enum):
     warning = logging.WARNING
 
 
-def main(log_level: LogLevel = typer.Option(LogLevel.warning, enum_by_name=True)):
+def main(log_level: LogLevel = typer.Option("warning", enum_by_name=True)):
     typer.echo(f"Log level set to: {logging.getLevelName(log_level.value)}")
 
 

diff --git a/docs_src/parameter_types/enum/tutorial005.py b/docs_src/parameter_types/enum/tutorial005.py
@@ -10,7 +10,7 @@ class Access(enum.IntEnum):
     open = 4
 
 
-def main(access: Access = typer.Option(Access.private, enum_by_name=True)):
+def main(access: Access = typer.Option("private", enum_by_name=True)):
     typer.echo(f"Access level: {access.name} ({access.value})")
 
 

diff --git a/tests/test_tutorial/test_parameter_types/test_enum/test_tutorial004.py b/tests/test_tutorial/test_parameter_types/test_enum/test_tutorial004.py
@@ -11,6 +11,12 @@
 app.command()(mod.main)
 
 
+def test_enum_names_default():
+    result = runner.invoke(app)
+    assert result.exit_code == 0
+    assert "Log level set to: WARNING" in result.output
+
+
 def test_enum_names():
     result = runner.invoke(app, ["--log-level", "debug"])
     assert result.exit_code == 0

diff --git a/tests/test_tutorial/test_parameter_types/test_enum/test_tutorial005.py b/tests/test_tutorial/test_parameter_types/test_enum/test_tutorial005.py
@@ -11,10 +11,16 @@
 app.command()(mod.main)
 
 
+def test_int_enum_default():
+    result = runner.invoke(app)
+    assert result.exit_code == 0
+    assert "Access level: private (1)" in result.output
+
+
 def test_int_enum():
     result = runner.invoke(app, ["--access", "open"])
     assert result.exit_code == 0
-    assert "Access level: open" in result.output
+    assert "Access level: open (4)" in result.output
 
 
 def test_script():