set return value = None when name passed for print

tmbo · Aug 15, 2022 · 508e800 · 508e800
1 parent ccf5382
commit 508e800
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 0 deletions.
diff --git a/questionary/prompt.py b/questionary/prompt.py
@@ -186,6 +186,8 @@ def unsafe_prompt(
             _kwargs.pop("input", None)
 
             print_formatted_text(message, **_kwargs)
+            if name:
+                answers[name] = None
             continue
 
         choices = question_config.get("choices")

diff --git a/tests/test_examples.py b/tests/test_examples.py
@@ -121,6 +121,7 @@ def test_advanced_workflow_example():
     result_dict = ask_with_patched_input(ask_dictstyle, text)
 
     assert result_dict == {
+        "intro": None,
         "conditional_step": True,
         "next_question": "questionary",
         "second_question": "Hello World",

diff --git a/tests/test_prompt.py b/tests/test_prompt.py
@@ -69,3 +69,10 @@ def test_print_no_name():
     questions = [{"type": "print", "message": "Hello World"}]
     result = patched_prompt(questions, "")
     assert result == {}
+
+
+def test_print_with_name():
+    """'print' type should return {name: None} when name is provided"""
+    questions = [{"name": "hello", "type": "print", "message": "Hello World"}]
+    result = patched_prompt(questions, "")
+    assert result == {"hello": None}