1 files changed, 55 insertions, 0 deletions
diff --git a/test/panes.zig b/test/panes.zig
index a452bcfb..16e40ee3 100644
--- a/test/panes.zig
+++ b/test/panes.zig
@@ -190,6 +190,60 @@ const TagNavigationTests = struct {
     }
 };
 
+const BodyHitTests = struct {
+    fn mouse(layer: *const pardes.BodyLayer, row: u16, col: u16, kind: pardes.Mouse.Kind) pardes.Mouse {
+        const width: f32 = if (row < layer.context_rows) 8 else 10;
+        const x = @as(f32, @floatFromInt(layer.viewport.x)) * 10 + (@as(f32, @floatFromInt(col)) + 0.5) * width;
+        const y = layer.rowTop(row, 20, 10) + layer.rowHeight(row, 20, 10) / 2;
+        return .{ .button = .left, .kind = kind, .col = @intFromFloat(x / 10), .row = @intFromFloat(y / 20), .body_hit = layer.hitAt(x, y, 10, 20, 8, 10).? };
+    }
+
+    test "compact TreeContext mouse follows source rows in reclaimed space" {
+        if (!syntax.enabled) return error.SkipZigTest;
+        const p = try Pardes.init(std.testing.allocator, .{ .tty_only = true, .cols = 64, .rows = 18 });
+        defer p.deinit();
+        p.row_metrics = .{ .body_w = 10, .body_h = 20, .tagline_w = 8, .tagline_h = 10 };
+        const pane = try TreeContextTests.setZigFile(p, "const Box = struct {\n    pub fn work(\n        value: usize,\n    ) void {\n" ++ "        // body\n" ** 60 ++ "    }\n};\n");
+        try std.testing.expect(p.executeBuiltinLine(0, "TreeContext"));
+        pane.scrollBy(10);
+        var frame = std.heap.ArenaAllocator.init(p.gpa);
+        defer frame.deinit();
+        _ = try p.render(frame.allocator());
+        const layer = &p.surface.body_layers[0];
+        try std.testing.expect(layer.context_rows >= 4);
+        try std.testing.expect(layer.rows > pane.rows);
+        const first = mouse(layer, 0, 2, .press);
+        const second = mouse(layer, 1, 2, .press);
+        try std.testing.expectEqual(first.row, second.row);
+        const top = pane.scroll();
+        for ([_]u16{ 0, 1 }) |row| {
+            const expected = pane.wrapAt(row).line;
+            p.update(.{ .mouse = mouse(layer, row, 2, .press) });
+            p.update(.{ .mouse = mouse(layer, row, 2, .release) });
+            try std.testing.expectEqual(expected, pane.cur_row);
+            try std.testing.expectEqual(top, pane.scroll());
+        }
+        const last = pane.body_visible_rows - 1;
+        try std.testing.expect(last >= pane.rows);
+        const expected = pane.wrapAt(last).line;
+        const prefix = panes.File.gutterWidth(pane);
+        p.update(.{ .mouse = mouse(layer, last - 1, prefix + 11, .press) });
+        // Release at a different source row without an intermediate drag event.
+        p.update(.{ .mouse = mouse(layer, last, prefix + 14, .release) });
+        try std.testing.expectEqualStrings("body\nbody", pardes.test_api.heldSelection(p, 0).?);
+        p.update(.{ .mouse = mouse(layer, last, 10, .press) });
+        p.update(.{ .mouse = mouse(layer, last, 10, .release) });
+        try std.testing.expectEqual(expected, pane.cur_row);
+        var stale = mouse(layer, 0, 2, .press);
+        stale.body_hit.?.serial +%= 1;
+        p.update(.{ .mouse = stale });
+        try std.testing.expectEqual(expected, pane.cur_row);
+        p.update(.{ .key = .{ .cp = 'i' } });
+        p.update(.{ .key = .{ .cp = 'X', .text = "X" } });
+        try std.testing.expect(std.mem.indexOfScalar(u8, modal.lineSlice(pane.file.?.content, @intCast(expected)), 'X') != null);
+    }
+};
+
 const TreeContextTests = struct {
     const source = "const Box = struct {\n" ++
         "    pub fn work(\n" ++
@@ -3556,6 +3610,7 @@ test {
     _ = TagNavigationTests;
     _ = TagNameTintTests;
     _ = TreeContextTests;
+    _ = BodyHitTests;
     _ = TtySelectionTests;
     _ = ClickCursorTests;
     _ = JumpSelectionTests;