From e26cb46f4afcd31610cbc81c6391c2e82b707484 Mon Sep 17 00:00:00 2001
From: Mitchell Hashimoto <mitchell.hashimoto@gmail.com>
Date: Wed, 31 May 2023 16:20:30 -0700
Subject: [PATCH] terminal: tests for semantic prompt on resize

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

diff --git a/src/terminal/Screen.zig b/src/terminal/Screen.zig
index 7149d5f34..03dece7c6 100644
--- a/src/terminal/Screen.zig
+++ b/src/terminal/Screen.zig
@@ -4455,6 +4455,53 @@ test "Screen: resize more cols trailing background colors" {
     }
 }
 
+test "Screen: resize more cols no reflow preserves semantic prompt" {
+    const testing = std.testing;
+    const alloc = testing.allocator;
+
+    var s = try init(alloc, 3, 5, 0);
+    defer s.deinit();
+    const str = "1ABCD\n2EFGH\n3IJKL";
+    try s.testWriteString(str);
+
+    // Set one of the rows to be a prompt
+    {
+        const row = s.getRow(.{ .active = 1 });
+        row.setSemanticPrompt(.prompt);
+    }
+
+    const cursor = s.cursor;
+    try s.resize(3, 10);
+
+    // Cursor should not move
+    try testing.expectEqual(cursor, s.cursor);
+
+    {
+        var contents = try s.testString(alloc, .viewport);
+        defer alloc.free(contents);
+        try testing.expectEqualStrings(str, contents);
+    }
+    {
+        var contents = try s.testString(alloc, .screen);
+        defer alloc.free(contents);
+        try testing.expectEqualStrings(str, contents);
+    }
+
+    // Our one row should still be a semantic prompt, the others should not.
+    {
+        const row = s.getRow(.{ .active = 0 });
+        try testing.expect(row.getSemanticPrompt() == .unknown);
+    }
+    {
+        const row = s.getRow(.{ .active = 1 });
+        try testing.expect(row.getSemanticPrompt() == .prompt);
+    }
+    {
+        const row = s.getRow(.{ .active = 2 });
+        try testing.expect(row.getSemanticPrompt() == .unknown);
+    }
+}
+
 test "Screen: resize more cols grapheme map" {
     const testing = std.testing;
     const alloc = testing.allocator;
@@ -4978,6 +5025,55 @@ test "Screen: resize less cols with graphemes" {
     }
 }
 
+test "Screen: resize less cols no reflow preserves semantic prompt" {
+    const testing = std.testing;
+    const alloc = testing.allocator;
+
+    var s = try init(alloc, 3, 5, 0);
+    defer s.deinit();
+    const str = "1AB\n2EF\n3IJ";
+    try s.testWriteString(str);
+
+    // Set one of the rows to be a prompt
+    {
+        const row = s.getRow(.{ .active = 1 });
+        row.setSemanticPrompt(.prompt);
+    }
+
+    s.cursor.x = 0;
+    s.cursor.y = 0;
+    const cursor = s.cursor;
+    try s.resize(3, 3);
+
+    // Cursor should not move
+    try testing.expectEqual(cursor, s.cursor);
+
+    {
+        var contents = try s.testString(alloc, .viewport);
+        defer alloc.free(contents);
+        try testing.expectEqualStrings(str, contents);
+    }
+    {
+        var contents = try s.testString(alloc, .screen);
+        defer alloc.free(contents);
+        try testing.expectEqualStrings(str, contents);
+    }
+
+    // Our one row should still be a semantic prompt, the others should not.
+    {
+        const row = s.getRow(.{ .active = 0 });
+        try testing.expect(row.getSemanticPrompt() == .unknown);
+    }
+    {
+        const row = s.getRow(.{ .active = 1 });
+        try testing.expect(row.getSemanticPrompt() == .prompt);
+    }
+    {
+        const row = s.getRow(.{ .active = 2 });
+        try testing.expect(row.getSemanticPrompt() == .unknown);
+    }
+}
+
 test "Screen: resize less cols with reflow but row space" {
     const testing = std.testing;
     const alloc = testing.allocator;