Compare commits
4 Commits
64c1a1604d
...
60632f62ad
Author | SHA1 | Date | |
---|---|---|---|
60632f62ad | |||
e2bf69eb31 | |||
3ec8ab1b9c | |||
dc97e44c9e |
@ -19,6 +19,7 @@ pub fn build(b: *std.Build) void {
|
|||||||
});
|
});
|
||||||
|
|
||||||
const tests = b.addTest(.{
|
const tests = b.addTest(.{
|
||||||
|
.name = "nats-zig-unit-tests",
|
||||||
.root_source_file = .{ .path = "tests/main.zig" },
|
.root_source_file = .{ .path = "tests/main.zig" },
|
||||||
.target = target,
|
.target = target,
|
||||||
.optimize = optimize,
|
.optimize = optimize,
|
||||||
@ -29,6 +30,7 @@ pub fn build(b: *std.Build) void {
|
|||||||
|
|
||||||
const run_main_tests = b.addRunArtifact(tests);
|
const run_main_tests = b.addRunArtifact(tests);
|
||||||
const test_step = b.step("test", "Run tests");
|
const test_step = b.step("test", "Run tests");
|
||||||
|
test_step.dependOn(&b.addInstallArtifact(tests, .{}).step);
|
||||||
test_step.dependOn(&run_main_tests.step);
|
test_step.dependOn(&run_main_tests.step);
|
||||||
|
|
||||||
add_examples(b, .{
|
add_examples(b, .{
|
||||||
|
16
src/nats.zig
16
src/nats.zig
@ -114,7 +114,8 @@ pub fn sign(encoded_seed: [:0]const u8, input: [:0]const u8) Error![]const u8 {
|
|||||||
|
|
||||||
// Note: an "Inbox" is actually just a string. This API creates a random (unique)
|
// Note: an "Inbox" is actually just a string. This API creates a random (unique)
|
||||||
// string suitable for passing as the `reply` field to Message.create or
|
// string suitable for passing as the `reply` field to Message.create or
|
||||||
// Connection.publishRequest.
|
// Connection.publishRequest. The string is owned by the caller and should be freed
|
||||||
|
// using `destroyInbox`.
|
||||||
pub fn createInbox() Error![:0]u8 {
|
pub fn createInbox() Error![:0]u8 {
|
||||||
var self: [*c]u8 = undefined;
|
var self: [*c]u8 = undefined;
|
||||||
const status = Status.fromInt(nats_c.natsInbox_Create(@ptrCast(&self)));
|
const status = Status.fromInt(nats_c.natsInbox_Create(@ptrCast(&self)));
|
||||||
@ -122,12 +123,19 @@ pub fn createInbox() Error![:0]u8 {
|
|||||||
return status.toError() orelse std.mem.sliceTo(self, 0);
|
return status.toError() orelse std.mem.sliceTo(self, 0);
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn destroyInbox(inbox: [:0]const u8) void {
|
pub fn destroyInbox(inbox: [:0]u8) void {
|
||||||
nats_c.natsInbox_Destroy(@constCast(@ptrCast(inbox.ptr)));
|
nats_c.natsInbox_Destroy(@ptrCast(inbox.ptr));
|
||||||
}
|
}
|
||||||
|
|
||||||
// I think this is also a jetstream API. This function sure does not seem at all useful
|
// I think this is also a jetstream API. This function sure does not seem at all useful
|
||||||
// by itself.
|
// by itself. Note: for some reason, most of the jetstream data structures are all
|
||||||
|
// public, instead of following the opaque handle style that the rest of the library
|
||||||
|
// does.
|
||||||
|
|
||||||
|
// typedef struct natsMsgList {
|
||||||
|
// natsMsg **Msgs;
|
||||||
|
// int Count;
|
||||||
|
// } natsMsgList;
|
||||||
pub const MessageList = opaque {
|
pub const MessageList = opaque {
|
||||||
pub fn destroy(self: *MessageList) void {
|
pub fn destroy(self: *MessageList) void {
|
||||||
nats_c.natsMsgList_Destroy(@ptrCast(self));
|
nats_c.natsMsgList_Destroy(@ptrCast(self));
|
||||||
|
18
tests/connection.zig
Normal file
18
tests/connection.zig
Normal file
@ -0,0 +1,18 @@
|
|||||||
|
const std = @import("std");
|
||||||
|
|
||||||
|
const nats = @import("nats");
|
||||||
|
|
||||||
|
const util = @import("./util.zig");
|
||||||
|
|
||||||
|
test "nats.Connection.connectTo" {
|
||||||
|
var server = try util.TestServer.launch(.{});
|
||||||
|
defer server.stop();
|
||||||
|
|
||||||
|
{
|
||||||
|
try nats.init(nats.default_spin_count);
|
||||||
|
defer nats.deinit();
|
||||||
|
|
||||||
|
const connection = try nats.Connection.connectTo(nats.default_server_url);
|
||||||
|
defer connection.destroy();
|
||||||
|
}
|
||||||
|
}
|
@ -1,4 +1,5 @@
|
|||||||
test {
|
test {
|
||||||
_ = @import("./nats.zig");
|
_ = @import("./nats.zig");
|
||||||
|
_ = @import("./connection.zig");
|
||||||
_ = @import("./message.zig");
|
_ = @import("./message.zig");
|
||||||
}
|
}
|
||||||
|
@ -1,8 +1,6 @@
|
|||||||
const std = @import("std");
|
const std = @import("std");
|
||||||
const nats = @import("nats");
|
|
||||||
|
|
||||||
// const nats = @import("../src/nats.zig");
|
const nats = @import("nats");
|
||||||
// const message = @import("../src/message.zig");
|
|
||||||
|
|
||||||
test "message: create message" {
|
test "message: create message" {
|
||||||
const subject = "hello";
|
const subject = "hello";
|
||||||
@ -11,7 +9,7 @@ test "message: create message" {
|
|||||||
|
|
||||||
// have to initialize the library so the reference counter can correctly destroy
|
// have to initialize the library so the reference counter can correctly destroy
|
||||||
// objects, otherwise we segfault on trying to free the memory.
|
// objects, otherwise we segfault on trying to free the memory.
|
||||||
try nats.init(-1);
|
try nats.init(nats.default_spin_count);
|
||||||
defer nats.deinit();
|
defer nats.deinit();
|
||||||
|
|
||||||
const message = try nats.Message.create(subject, reply, data);
|
const message = try nats.Message.create(subject, reply, data);
|
||||||
@ -28,7 +26,7 @@ test "message: create message" {
|
|||||||
}
|
}
|
||||||
|
|
||||||
test "message: get subject" {
|
test "message: get subject" {
|
||||||
try nats.init(-1);
|
try nats.init(nats.default_spin_count);
|
||||||
defer nats.deinit();
|
defer nats.deinit();
|
||||||
|
|
||||||
const subject = "hello";
|
const subject = "hello";
|
||||||
@ -40,7 +38,7 @@ test "message: get subject" {
|
|||||||
}
|
}
|
||||||
|
|
||||||
test "message: get reply" {
|
test "message: get reply" {
|
||||||
try nats.init(-1);
|
try nats.init(nats.default_spin_count);
|
||||||
defer nats.deinit();
|
defer nats.deinit();
|
||||||
|
|
||||||
const subject = "hello";
|
const subject = "hello";
|
||||||
|
73
tests/util.zig
Normal file
73
tests/util.zig
Normal file
@ -0,0 +1,73 @@
|
|||||||
|
const std = @import("std");
|
||||||
|
|
||||||
|
const TestLaunchError = error{
|
||||||
|
NoLaunchStringFound,
|
||||||
|
};
|
||||||
|
|
||||||
|
pub const TestServer = struct {
|
||||||
|
process: std.ChildProcess,
|
||||||
|
|
||||||
|
pub const LaunchOptions = struct {
|
||||||
|
executable: []const u8 = "nats-server",
|
||||||
|
port: u16 = 4222,
|
||||||
|
auth: union(enum) {
|
||||||
|
none: void,
|
||||||
|
token: []const u8,
|
||||||
|
password: struct { user: []const u8, pass: []const u8 },
|
||||||
|
} = .none,
|
||||||
|
allocator: std.mem.Allocator = std.testing.allocator,
|
||||||
|
|
||||||
|
fn argLen(self: LaunchOptions) usize {
|
||||||
|
// executable, -a, 127.0.0.1, -p, 4222
|
||||||
|
const base_len: usize = 5;
|
||||||
|
return base_len + switch (self.auth) {
|
||||||
|
.none => @as(usize, 0),
|
||||||
|
.token => @as(usize, 2),
|
||||||
|
.password => @as(usize, 4),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
pub fn launch(options: LaunchOptions) !TestServer {
|
||||||
|
// const allocator = std.testing.allocator;
|
||||||
|
var portbuf = [_]u8{0} ** 5;
|
||||||
|
const strport = try std.fmt.bufPrint(&portbuf, "{d}", .{options.port});
|
||||||
|
|
||||||
|
const argsbuf: [9][]const u8 = blk: {
|
||||||
|
const executable: [1][]const u8 = .{options.executable};
|
||||||
|
const listen: [2][]const u8 = .{ "-a", "127.0.0.1" };
|
||||||
|
const port: [2][]const u8 = .{ "-p", strport };
|
||||||
|
const auth: [4][]const u8 = switch (options.auth) {
|
||||||
|
.none => .{""} ** 4,
|
||||||
|
.token => |tok| .{ "--auth", tok, "", "" },
|
||||||
|
.password => |auth| .{ "--user", auth.user, "--password", auth.pass },
|
||||||
|
};
|
||||||
|
|
||||||
|
break :blk executable ++ listen ++ port ++ auth;
|
||||||
|
};
|
||||||
|
|
||||||
|
const args = argsbuf[0..options.argLen()];
|
||||||
|
|
||||||
|
var child = std.ChildProcess.init(args, options.allocator);
|
||||||
|
child.stdin_behavior = .Ignore;
|
||||||
|
child.stdout_behavior = .Pipe;
|
||||||
|
child.stderr_behavior = .Pipe;
|
||||||
|
|
||||||
|
try child.spawn();
|
||||||
|
var poller = std.io.poll(options.allocator, enum { stderr }, .{ .stderr = child.stderr.? });
|
||||||
|
defer poller.deinit();
|
||||||
|
|
||||||
|
while (try poller.poll()) {
|
||||||
|
if (std.mem.indexOf(u8, poller.fifo(.stderr).buf, "[INF] Server is ready")) |_| {
|
||||||
|
return .{ .process = child };
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
_ = try child.kill();
|
||||||
|
return error.NoLaunchStringFound;
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn stop(self: *TestServer) void {
|
||||||
|
_ = self.process.kill() catch return;
|
||||||
|
}
|
||||||
|
};
|
Loading…
x
Reference in New Issue
Block a user