move Context into Context.zig
This commit is contained in:
Techatrix
parent
ae5da361bf
commit
e562d4cf4c
87
tests/context.zig
Normal file
87
tests/context.zig
Normal file
@ -0,0 +1,87 @@
|
|||||||
|
const std = @import("std");
|
||||||
|
const headerPkg = @import("header");
|
||||||
|
const Server = @import("server");
|
||||||
|
|
||||||
|
const initialize_msg =
|
||||||
|
\\{"processId":6896,"clientInfo":{"name":"vscode","version":"1.46.1"},"rootPath":null,"rootUri":null,"capabilities":{"workspace":{"applyEdit":true,"workspaceEdit":{"documentChanges":true,"resourceOperations":["create","rename","delete"],"failureHandling":"textOnlyTransactional"},"didChangeConfiguration":{"dynamicRegistration":true},"didChangeWatchedFiles":{"dynamicRegistration":true},"symbol":{"dynamicRegistration":true,"symbolKind":{"valueSet":[1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,16,17,18,19,20,21,22,23,24,25,26]},"tagSupport":{"valueSet":[1]}},"executeCommand":{"dynamicRegistration":true},"configuration":true,"workspaceFolders":true},"textDocument":{"publishDiagnostics":{"relatedInformation":true,"versionSupport":false,"tagSupport":{"valueSet":[1,2]},"complexDiagnosticCodeSupport":true},"synchronization":{"dynamicRegistration":true,"willSave":true,"willSaveWaitUntil":true,"didSave":true},"completion":{"dynamicRegistration":true,"contextSupport":true,"completionItem":{"snippetSupport":true,"commitCharactersSupport":true,"documentationFormat":["markdown","plaintext"],"deprecatedSupport":true,"preselectSupport":true,"tagSupport":{"valueSet":[1]},"insertReplaceSupport":true},"completionItemKind":{"valueSet":[1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,16,17,18,19,20,21,22,23,24,25]}},"hover":{"dynamicRegistration":true,"contentFormat":["markdown","plaintext"]},"signatureHelp":{"dynamicRegistration":true,"signatureInformation":{"documentationFormat":["markdown","plaintext"],"parameterInformation":{"labelOffsetSupport":true}},"contextSupport":true},"definition":{"dynamicRegistration":true,"linkSupport":true},"references":{"dynamicRegistration":true},"documentHighlight":{"dynamicRegistration":true},"documentSymbol":{"dynamicRegistration":true,"symbolKind":{"valueSet":[1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,16,17,18,19,20,21,22,23,24,25,26]},"hierarchicalDocumentSymbolSupport":true,"tagSupport":{"valueSet":[1]}},"codeAction":{"dynamicRegistration":true,"isPreferredSupport":true,"codeActionLiteralSupport":{"codeActionKind":{"valueSet":["","quickfix","refactor","refactor.extract","refactor.inline","refactor.rewrite","source","source.organizeImports"]}}},"codeLens":{"dynamicRegistration":true},"formatting":{"dynamicRegistration":true},"rangeFormatting":{"dynamicRegistration":true},"onTypeFormatting":{"dynamicRegistration":true},"rename":{"dynamicRegistration":true,"prepareSupport":true},"documentLink":{"dynamicRegistration":true,"tooltipSupport":true},"typeDefinition":{"dynamicRegistration":true,"linkSupport":true},"implementation":{"dynamicRegistration":true,"linkSupport":true},"colorProvider":{"dynamicRegistration":true},"foldingRange":{"dynamicRegistration":true,"rangeLimit":5000,"lineFoldingOnly":true},"declaration":{"dynamicRegistration":true,"linkSupport":true},"selectionRange":{"dynamicRegistration":true},"semanticTokens":{"dynamicRegistration":true,"tokenTypes":["comment","keyword","number","regexp","operator","namespace","type","struct","class","interface","enum","typeParameter","function","member","macro","variable","parameter","property","label"],"tokenModifiers":["declaration","documentation","static","abstract","deprecated","readonly"]}},"window":{"workDoneProgress":true}},"trace":"off","workspaceFolders":[{"uri":"file://./tests", "name":"root"}]}
|
||||||
|
;
|
||||||
|
|
||||||
|
const allocator = std.testing.allocator;
|
||||||
|
|
||||||
|
pub const Context = struct {
|
||||||
|
server: Server,
|
||||||
|
request_id: u32 = 1,
|
||||||
|
|
||||||
|
pub fn init() !Context {
|
||||||
|
var context = Context{
|
||||||
|
.server = try Server.init(
|
||||||
|
allocator,
|
||||||
|
.{},
|
||||||
|
null,
|
||||||
|
.debug,
|
||||||
|
),
|
||||||
|
};
|
||||||
|
|
||||||
|
try context.request("initialize", initialize_msg, null);
|
||||||
|
try context.request("initialized", "{}", null);
|
||||||
|
return context;
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn deinit(self: *Context) void {
|
||||||
|
self.request("shutdown", "{}", null) catch {};
|
||||||
|
self.server.deinit();
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn request(
|
||||||
|
self: *Context,
|
||||||
|
method: []const u8,
|
||||||
|
params: []const u8,
|
||||||
|
expect: ?[]const u8,
|
||||||
|
) !void {
|
||||||
|
var output = std.ArrayList(u8).init(allocator);
|
||||||
|
defer output.deinit();
|
||||||
|
|
||||||
|
// create the request
|
||||||
|
self.request_id += 1;
|
||||||
|
const req = try std.fmt.allocPrint(allocator,
|
||||||
|
\\{{"jsonrpc":"2.0","id":{},"method":"{s}","params":{s}}}
|
||||||
|
, .{ self.request_id, method, params });
|
||||||
|
defer allocator.free(req);
|
||||||
|
|
||||||
|
// send the request to the server
|
||||||
|
try self.server.processJsonRpc(output.writer(), req);
|
||||||
|
|
||||||
|
// if we don't expect a response ignore it
|
||||||
|
const expected = expect orelse return;
|
||||||
|
|
||||||
|
// get the output from the server
|
||||||
|
var buffer_stream = std.io.fixedBufferStream(output.items);
|
||||||
|
const header = try headerPkg.readRequestHeader(allocator, buffer_stream.reader());
|
||||||
|
defer header.deinit(allocator);
|
||||||
|
|
||||||
|
var response_bytes = try allocator.alloc(u8, header.content_length);
|
||||||
|
defer allocator.free(response_bytes);
|
||||||
|
|
||||||
|
const content_length = try buffer_stream.reader().readAll(response_bytes);
|
||||||
|
try std.testing.expectEqual(content_length, header.content_length);
|
||||||
|
|
||||||
|
// parse the response
|
||||||
|
var parser = std.json.Parser.init(allocator, false);
|
||||||
|
defer parser.deinit();
|
||||||
|
|
||||||
|
var tree = try parser.parse(response_bytes);
|
||||||
|
defer tree.deinit();
|
||||||
|
|
||||||
|
const response = tree.root.Object;
|
||||||
|
|
||||||
|
// assertions
|
||||||
|
try std.testing.expectEqualStrings("2.0", response.get("jsonrpc").?.String);
|
||||||
|
try std.testing.expectEqual(self.request_id, @intCast(u32, response.get("id").?.Integer));
|
||||||
|
try std.testing.expect(!response.contains("error"));
|
||||||
|
|
||||||
|
const result_json = try std.json.stringifyAlloc(allocator, response.get("result").?, .{});
|
||||||
|
defer allocator.free(result_json);
|
||||||
|
|
||||||
|
try std.testing.expectEqualStrings(expected, result_json);
|
||||||
|
}
|
||||||
|
};
|
||||||
@ -1,91 +1,8 @@
|
|||||||
const std = @import("std");
|
const std = @import("std");
|
||||||
const headerPkg = @import("header");
|
const Context = @import("context.zig").Context;
|
||||||
const Server = @import("server");
|
|
||||||
|
|
||||||
const allocator = std.testing.allocator;
|
const allocator = std.testing.allocator;
|
||||||
|
|
||||||
const initialize_msg =
|
|
||||||
\\{"processId":6896,"clientInfo":{"name":"vscode","version":"1.46.1"},"rootPath":null,"rootUri":null,"capabilities":{"workspace":{"applyEdit":true,"workspaceEdit":{"documentChanges":true,"resourceOperations":["create","rename","delete"],"failureHandling":"textOnlyTransactional"},"didChangeConfiguration":{"dynamicRegistration":true},"didChangeWatchedFiles":{"dynamicRegistration":true},"symbol":{"dynamicRegistration":true,"symbolKind":{"valueSet":[1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,16,17,18,19,20,21,22,23,24,25,26]},"tagSupport":{"valueSet":[1]}},"executeCommand":{"dynamicRegistration":true},"configuration":true,"workspaceFolders":true},"textDocument":{"publishDiagnostics":{"relatedInformation":true,"versionSupport":false,"tagSupport":{"valueSet":[1,2]},"complexDiagnosticCodeSupport":true},"synchronization":{"dynamicRegistration":true,"willSave":true,"willSaveWaitUntil":true,"didSave":true},"completion":{"dynamicRegistration":true,"contextSupport":true,"completionItem":{"snippetSupport":true,"commitCharactersSupport":true,"documentationFormat":["markdown","plaintext"],"deprecatedSupport":true,"preselectSupport":true,"tagSupport":{"valueSet":[1]},"insertReplaceSupport":true},"completionItemKind":{"valueSet":[1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,16,17,18,19,20,21,22,23,24,25]}},"hover":{"dynamicRegistration":true,"contentFormat":["markdown","plaintext"]},"signatureHelp":{"dynamicRegistration":true,"signatureInformation":{"documentationFormat":["markdown","plaintext"],"parameterInformation":{"labelOffsetSupport":true}},"contextSupport":true},"definition":{"dynamicRegistration":true,"linkSupport":true},"references":{"dynamicRegistration":true},"documentHighlight":{"dynamicRegistration":true},"documentSymbol":{"dynamicRegistration":true,"symbolKind":{"valueSet":[1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,16,17,18,19,20,21,22,23,24,25,26]},"hierarchicalDocumentSymbolSupport":true,"tagSupport":{"valueSet":[1]}},"codeAction":{"dynamicRegistration":true,"isPreferredSupport":true,"codeActionLiteralSupport":{"codeActionKind":{"valueSet":["","quickfix","refactor","refactor.extract","refactor.inline","refactor.rewrite","source","source.organizeImports"]}}},"codeLens":{"dynamicRegistration":true},"formatting":{"dynamicRegistration":true},"rangeFormatting":{"dynamicRegistration":true},"onTypeFormatting":{"dynamicRegistration":true},"rename":{"dynamicRegistration":true,"prepareSupport":true},"documentLink":{"dynamicRegistration":true,"tooltipSupport":true},"typeDefinition":{"dynamicRegistration":true,"linkSupport":true},"implementation":{"dynamicRegistration":true,"linkSupport":true},"colorProvider":{"dynamicRegistration":true},"foldingRange":{"dynamicRegistration":true,"rangeLimit":5000,"lineFoldingOnly":true},"declaration":{"dynamicRegistration":true,"linkSupport":true},"selectionRange":{"dynamicRegistration":true},"semanticTokens":{"dynamicRegistration":true,"tokenTypes":["comment","keyword","number","regexp","operator","namespace","type","struct","class","interface","enum","typeParameter","function","member","macro","variable","parameter","property","label"],"tokenModifiers":["declaration","documentation","static","abstract","deprecated","readonly"]}},"window":{"workDoneProgress":true}},"trace":"off","workspaceFolders":[{"uri":"file://./tests", "name":"root"}]}
|
|
||||||
;
|
|
||||||
|
|
||||||
const Context = struct {
|
|
||||||
server: Server,
|
|
||||||
request_id: u32 = 1,
|
|
||||||
|
|
||||||
pub fn init() !Context {
|
|
||||||
var context = Context{
|
|
||||||
.server = try Server.init(
|
|
||||||
allocator,
|
|
||||||
.{},
|
|
||||||
null,
|
|
||||||
.debug,
|
|
||||||
),
|
|
||||||
};
|
|
||||||
|
|
||||||
try context.request("initialize", initialize_msg, null);
|
|
||||||
try context.request("initialized", "{}", null);
|
|
||||||
return context;
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn deinit(self: *Context) void {
|
|
||||||
self.request("shutdown", "{}", null) catch {};
|
|
||||||
self.server.deinit();
|
|
||||||
}
|
|
||||||
|
|
||||||
fn request(
|
|
||||||
self: *Context,
|
|
||||||
method: []const u8,
|
|
||||||
params: []const u8,
|
|
||||||
expect: ?[]const u8,
|
|
||||||
) !void {
|
|
||||||
var output = std.ArrayList(u8).init(allocator);
|
|
||||||
defer output.deinit();
|
|
||||||
|
|
||||||
// create the request
|
|
||||||
self.request_id += 1;
|
|
||||||
const req = try std.fmt.allocPrint(allocator,
|
|
||||||
\\{{"jsonrpc":"2.0","id":{},"method":"{s}","params":{s}}}
|
|
||||||
, .{ self.request_id, method, params });
|
|
||||||
defer allocator.free(req);
|
|
||||||
|
|
||||||
// send the request to the server
|
|
||||||
try self.server.processJsonRpc(output.writer(), req);
|
|
||||||
|
|
||||||
// if we don't expect a response ignore it
|
|
||||||
const expected = expect orelse return;
|
|
||||||
|
|
||||||
// get the output from the server
|
|
||||||
var buffer_stream = std.io.fixedBufferStream(output.items);
|
|
||||||
const header = try headerPkg.readRequestHeader(allocator, buffer_stream.reader());
|
|
||||||
defer header.deinit(allocator);
|
|
||||||
|
|
||||||
var response_bytes = try allocator.alloc(u8, header.content_length);
|
|
||||||
defer allocator.free(response_bytes);
|
|
||||||
|
|
||||||
const content_length = try buffer_stream.reader().readAll(response_bytes);
|
|
||||||
try std.testing.expectEqual(content_length, header.content_length);
|
|
||||||
|
|
||||||
// parse the response
|
|
||||||
var parser = std.json.Parser.init(allocator, false);
|
|
||||||
defer parser.deinit();
|
|
||||||
|
|
||||||
var tree = try parser.parse(response_bytes);
|
|
||||||
defer tree.deinit();
|
|
||||||
|
|
||||||
const response = tree.root.Object;
|
|
||||||
|
|
||||||
// assertions
|
|
||||||
try std.testing.expectEqualStrings("2.0", response.get("jsonrpc").?.String);
|
|
||||||
try std.testing.expectEqual(self.request_id, @intCast(u32, response.get("id").?.Integer));
|
|
||||||
try std.testing.expect(!response.contains("error"));
|
|
||||||
|
|
||||||
const result_json = try std.json.stringifyAlloc(allocator, response.get("result").?, .{});
|
|
||||||
defer allocator.free(result_json);
|
|
||||||
|
|
||||||
try std.testing.expectEqualStrings(expected, result_json);
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
test "Open file, ask for semantic tokens" {
|
test "Open file, ask for semantic tokens" {
|
||||||
var ctx = try Context.init();
|
var ctx = try Context.init();
|
||||||
defer ctx.deinit();
|
defer ctx.deinit();
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user