From 9a4bd8750c3e76792eba81ba07cf9a3ce7065e7c Mon Sep 17 00:00:00 2001 From: Roland Walker Date: Sun, 15 Nov 2026 07:26:30 -0500 Subject: [PATCH] improve test coverage for batch_utils.py restoring 100% coverage --- changelog.md | 8 ++++++++ mycli_test/pytests/test_batch_utils.py | 10 ++++++++++ 2 files changed, 18 insertions(+) diff --git a/changelog.md b/changelog.md index c311431c..e3fd383f 100644 --- a/changelog.md +++ b/changelog.md @@ -1,3 +1,11 @@ +Upcoming (TBD) +============== + +Internal +-------- +* Improve test coverage for `batch_utils.py`. + + v2.31.0 (2026/11/14) ============== diff --git a/mycli_test/pytests/test_batch_utils.py b/mycli_test/pytests/test_batch_utils.py index 3af2984f..b20058c9 100644 --- a/mycli_test/pytests/test_batch_utils.py +++ b/mycli_test/pytests/test_batch_utils.py @@ -128,3 +128,13 @@ def test_statements_from_filehandle_ends_slash_use_at_newline() -> None: ('/use mysql', 1), ('select database();', 2), ] + + +@pytest.mark.parametrize('command', ['DELIMITER //', 'use mysql']) +def test_statements_from_filehandle_preserves_commands_inside_multiline_string(command: str) -> None: + query = f"select 'first\n{command}\nlast';" + + assert collect_statements(query + '\nselect 2;\n') == [ + (query, 0), + ('select 2;', 1), + ]