From cc8fb14906f0d768a7de640c70dd05c3b4cc305e Mon Sep 17 00:00:00 2001
From: Techatrix <19954306+Techatrix@users.noreply.github.com>
Date: Tue, 30 May 2023 00:48:46 +0200
Subject: [PATCH] add tests for autofix code actions

---
 src/zls.zig                         |   1 +
 tests/lsp_features/code_actions.zig | 158 ++++++++++++++++++++++++++++
 tests/tests.zig                     |   1 +
 3 files changed, 160 insertions(+)
 create mode 100644 tests/lsp_features/code_actions.zig

diff --git a/src/zls.zig b/src/zls.zig
index 407e130..ab68744 100644
--- a/src/zls.zig
+++ b/src/zls.zig
@@ -29,6 +29,7 @@ pub const completions = @import("features/completions.zig");
 pub const goto = @import("features/goto.zig");
 pub const hover_handler = @import("features/hover.zig");
 pub const selection_range = @import("features/selection_range.zig");
+pub const diagnostics = @import("features/diagnostics.zig");
 
 comptime {
     const std = @import("std");
diff --git a/tests/lsp_features/code_actions.zig b/tests/lsp_features/code_actions.zig
new file mode 100644
index 0000000..e458987
--- /dev/null
+++ b/tests/lsp_features/code_actions.zig
@@ -0,0 +1,158 @@
+const std = @import("std");
+const zls = @import("zls");
+
+const Context = @import("../context.zig").Context;
+
+const types = zls.types;
+const offsets = zls.offsets;
+
+const allocator: std.mem.Allocator = std.testing.allocator;
+
+test "code actions - discard value" {
+    try testAutofix(
+        \\test {
+        \\    var foo = {};
+        \\    const bar = {};
+        \\}
+        \\
+    ,
+        \\test {
+        \\    var foo = {};
+        \\    _ = foo;
+        \\    const bar = {};
+        \\    _ = bar;
+        \\}
+        \\
+    );
+}
+
+test "code actions - discard function parameter" {
+    try testAutofix(
+        \\fn foo(a: void, b: void, c: void) void {}
+        \\
+    ,
+        \\fn foo(a: void, b: void, c: void) void {
+        \\    _ = c;
+        \\    _ = b;
+        \\    _ = a;
+        \\}
+        \\
+    );
+}
+
+test "code actions - discard captures" {
+    try testAutofix(
+        \\test {
+        \\    for (0..10, 0..10, 0..10) |i, j, k| {}
+        \\    switch (union(enum) {}{}) {
+        \\        inline .a => |cap, tag| {},
+        \\    }
+        \\    if (null) |x| {}
+        \\    if (null) |v| {} else |e| {}
+        \\    _ = null catch |e| {};
+        \\}
+        \\
+    ,
+        \\test {
+        \\    for (0..10, 0..10, 0..10) |i, j, k| {
+        \\        _ = k;
+        \\        _ = j;
+        \\        _ = i;
+        \\    }
+        \\    switch (union(enum) {}{}) {
+        \\        inline .a => |cap, tag| {
+        \\            _ = tag;
+        \\            _ = cap;
+        \\        },
+        \\    }
+        \\    if (null) |x| {
+        \\        _ = x;
+        \\    }
+        \\    if (null) |v| {
+        \\        _ = v;
+        \\    } else |e| {
+        \\        _ = e;
+        \\    }
+        \\    _ = null catch |e| {
+        \\        _ = e;
+        \\    };
+        \\}
+        \\
+    );
+}
+
+test "code actions - remove pointless discard" {
+    try testAutofix(
+        \\fn foo(a: u32) u32 {
+        \\    _ = a;
+        \\    var b: ?u32 = a;
+        \\    _ = b;
+        \\    const c = b;
+        \\    _ = c;
+        \\    if (c) |d| {
+        \\        _ = d;
+        \\        return d;
+        \\    }
+        \\    return 0;
+        \\}
+        \\
+    ,
+        \\fn foo(a: u32) u32 {
+        \\    var b: ?u32 = a;
+        \\    const c = b;
+        \\    if (c) |d| {
+        \\        return d;
+        \\    }
+        \\    return 0;
+        \\}
+        \\
+    );
+}
+
+/// does not check for correct formatting
+fn testAutofix(before: []const u8, after: []const u8) !void {
+    var ctx = try Context.init();
+    defer ctx.deinit();
+    ctx.config.enable_ast_check_diagnostics = true;
+
+    const uri = try ctx.addDocument(before);
+    const handle = ctx.server.document_store.getHandle(uri).?;
+
+    var diagnostics: std.ArrayListUnmanaged(types.Diagnostic) = .{};
+    try zls.diagnostics.getAstCheckDiagnostics(ctx.server, handle.*, &diagnostics);
+
+    const params = types.CodeActionParams{
+        .textDocument = .{ .uri = uri },
+        .range = .{
+            .start = .{ .line = 0, .character = 0 },
+            .end = offsets.indexToPosition(before, before.len, ctx.server.offset_encoding),
+        },
+        .context = .{ .diagnostics = diagnostics.items },
+    };
+
+    @setEvalBranchQuota(5000);
+    const response = try ctx.requestGetResponse([]types.CodeAction, "textDocument/codeAction", params);
+
+    var text_edits: std.ArrayListUnmanaged(types.TextEdit) = .{};
+    defer text_edits.deinit(allocator);
+
+    for (response.result) |code_action| {
+        if (code_action.kind.? != .@"source.fixAll") continue;
+        const workspace_edit = code_action.edit.?;
+        const changes = workspace_edit.changes.?;
+        try std.testing.expectEqual(@as(usize, 1), changes.count());
+        try std.testing.expect(changes.contains(uri));
+
+        try text_edits.appendSlice(allocator, changes.get(uri).?);
+    }
+
+    const actual = try zls.diff.applyTextEdits(allocator, before, text_edits.items, ctx.server.offset_encoding);
+    defer allocator.free(actual);
+    try ctx.server.document_store.refreshDocument(uri, try allocator.dupeZ(u8, actual));
+
+    try std.testing.expect(handle.tree.errors.len == 0);
+    const formatted_actual = try handle.tree.render(allocator);
+    defer allocator.free(formatted_actual);
+
+    try std.testing.expectEqualStrings(after, formatted_actual);
+}
diff --git a/tests/tests.zig b/tests/tests.zig
index ebe7bce..06984f1 100644
--- a/tests/tests.zig
+++ b/tests/tests.zig
@@ -12,6 +12,7 @@ comptime {
     // TODO Document Synchronization
 
     // LSP features
+    _ = @import("lsp_features/code_actions.zig");
     _ = @import("lsp_features/completion.zig");
     _ = @import("lsp_features/definition.zig");
     _ = @import("lsp_features/document_symbol.zig");