From 9369a1a34be7d12645915f74ee65c86f523e1e18 Mon Sep 17 00:00:00 2001
From: Jon Parise <jon@indelible.org>
Date: Wed, 17 Jul 2024 20:39:43 -0400
Subject: [PATCH] terminal: test clearPrompt with a continuation

---
 src/terminal/Screen.zig | 28 ++++++++++++++++++++++++++++
 1 file changed, 28 insertions(+)

diff --git a/src/terminal/Screen.zig b/src/terminal/Screen.zig
index 11d6b85f4..12bdaa73c 100644
--- a/src/terminal/Screen.zig
+++ b/src/terminal/Screen.zig
@@ -2984,6 +2984,34 @@ test "Screen: clearPrompt" {
     }
 }
 
+test "Screen: clearPrompt continuation" {
+    const testing = std.testing;
+    const alloc = testing.allocator;
+
+    var s = try init(alloc, 5, 4, 0);
+    defer s.deinit();
+    const str = "1ABCD\n2EFGH\n3IJKL\n4MNOP";
+    try s.testWriteString(str);
+
+    // Set one of the rows to be a prompt followed by a continuation row
+    {
+        s.cursorAbsolute(0, 1);
+        s.cursor.page_row.semantic_prompt = .prompt;
+        s.cursorAbsolute(0, 2);
+        s.cursor.page_row.semantic_prompt = .prompt_continuation;
+        s.cursorAbsolute(0, 3);
+        s.cursor.page_row.semantic_prompt = .input;
+    }
+
+    s.clearPrompt();
+
+    {
+        const contents = try s.dumpStringAlloc(alloc, .{ .screen = .{} });
+        defer alloc.free(contents);
+        try testing.expectEqualStrings("1ABCD\n2EFGH", contents);
+    }
+}
+
 test "Screen: clearPrompt no prompt" {
     const testing = std.testing;
     const alloc = testing.allocator;