Add unit tests for multiline prompts

tests/option_list/test_option_prompt_replacement.py

@@ -12,21 +12,24 @@ class OptionListApp(App[None]):
     def compose(self) -> ComposeResult:
         yield OptionList(
             Option("0", id="0"),
-            Option("1"),
+            Option("line1\nline2"),
         )
 
+
 async def test_replace_option_prompt_with_invalid_id() -> None:
     """Attempting to replace the prompt of an option ID that doesn't exist should raise an exception."""
     async with OptionListApp().run_test() as pilot:
         with pytest.raises(OptionDoesNotExist):
             pilot.app.query_one(OptionList).replace_option_prompt("does-not-exist", "new-prompt")
 
+
 async def test_replace_option_prompt_with_invalid_index() -> None:
     """Attempting to replace the prompt of an option index that doesn't exist should raise an exception."""
     async with OptionListApp().run_test() as pilot:
         with pytest.raises(OptionDoesNotExist):
             pilot.app.query_one(OptionList).replace_option_prompt_at_index(23, "new-prompt")
 
+
 async def test_replace_option_prompt_with_valid_id() -> None:
     """It should be possible to replace the prompt of an option ID that does exist."""
     async with OptionListApp().run_test() as pilot:
@@ -34,8 +37,36 @@ async def test_replace_option_prompt_with_valid_id() -> None:
         option_list.replace_option_prompt("0", "new-prompt")
         assert option_list.get_option("0").prompt == "new-prompt"
 
+
 async def test_replace_option_prompt_with_valid_index() -> None:
     """It should be possible to replace the prompt of an option index that does exist."""
     async with OptionListApp().run_test() as pilot:
         option_list = pilot.app.query_one(OptionList).replace_option_prompt_at_index(1, "new-prompt")
         assert option_list.get_option_at_index(1).prompt == "new-prompt"
+
+
+async def test_replace_single_line_option_prompt_with_multiple() -> None:
+    """It should be possible to replace single line prompt with multiple lines """
+    new_prompt = "new-prompt\nsecond line"
+    async with OptionListApp().run_test() as pilot:
+        option_list = pilot.app.query_one(OptionList)
+        option_list.replace_option_prompt("0", new_prompt)
+        assert option_list.get_option("0").prompt == new_prompt
+
+
+async def test_replace_multiple_line_option_prompt_with_single() -> None:
+    """It should be possible to replace multiple line prompt with a single line"""
+    new_prompt = "new-prompt"
+    async with OptionListApp().run_test() as pilot:
+        option_list = pilot.app.query_one(OptionList)
+        option_list.replace_option_prompt("0", new_prompt)
+        assert option_list.get_option("0").prompt == new_prompt
+
+
+async def test_replace_multiple_line_option_prompt_with_multiple() -> None:
+    """It should be possible to replace multiple line prompt with multiple lines"""
+    new_prompt = "new-prompt\nsecond line"
+    async with OptionListApp().run_test() as pilot:
+        option_list = pilot.app.query_one(OptionList)
+        option_list.replace_option_prompt_at_index(1, new_prompt)
+        assert option_list.get_option_at_index(1).prompt == new_prompt