WIP refactor: add remote_poc_test

This commit is contained in:
CJ van den Berg 2026-03-06 21:24:08 +01:00
parent ddc06d67d6
commit 47f4202b94
Signed by: neurocyte
GPG key ID: 8EB1E1BB660E3FB9
6 changed files with 212 additions and 0 deletions

View file

@ -0,0 +1,13 @@
/// Child process for the remoting POC test.
/// Writes one framed CBOR send_named message to stdout and exits.
const std = @import("std");
const cbor = @import("cbor");
const framing = @import("framing");
pub fn main() !void {
var msg_buf: [256]u8 = undefined;
var stream: std.Io.Writer = .fixed(&msg_buf);
try cbor.writeValue(&stream, .{ "send_named", @as(u64, 1), "test_actor", .{ "hello", "from_child" } });
const payload = stream.buffered();
try framing.write_frame(std.fs.File.stdout(), payload);
}

110
test/remote_poc_test.zig Normal file
View file

@ -0,0 +1,110 @@
const std = @import("std");
const thespian = @import("thespian");
const cbor = @import("cbor");
const framing = @import("framing");
const protocol = @import("protocol");
const build_options = @import("build_options");
const Allocator = std.mem.Allocator;
const result = thespian.result;
const unexpected = thespian.unexpected;
const pid_ref = thespian.pid_ref;
const Receiver = thespian.Receiver;
const message = thespian.message;
const extract = thespian.extract;
const subprocess = thespian.subprocess;
const tag = "subprocess";
const Parent = struct {
allocator: Allocator,
proc: subprocess,
accumulator: framing.Accumulator,
receiver: Receiver(*@This()),
frame_received: bool = false,
const Args = struct { allocator: Allocator };
fn start(args: Args) result {
return init(args) catch |e| return thespian.exit_error(e, @errorReturnTrace());
}
fn init(args: Args) !void {
const proc = try subprocess.init(
args.allocator,
message.fmt(.{build_options.remote_child_path}),
tag,
.Ignore,
);
const self = try args.allocator.create(@This());
self.* = .{
.allocator = args.allocator,
.proc = proc,
.accumulator = .{},
.receiver = .init(receive_fn, deinit, self),
};
errdefer self.deinit();
thespian.receive(&self.receiver);
}
fn deinit(self: *@This()) void {
self.proc.deinit();
self.allocator.destroy(self);
}
fn receive_fn(self: *@This(), from: pid_ref, m: message) result {
return self.receive(from, m) catch |e| return thespian.exit_error(e, @errorReturnTrace());
}
fn receive(self: *@This(), _: pid_ref, m: message) !void {
var bytes: []const u8 = "";
var exit_code: i64 = 0;
if (try m.match(.{ tag, "stdout", extract(&bytes) })) {
if (self.accumulator.feed(bytes)) |frame| {
const msg = try protocol.decode(frame);
switch (msg) {
.send_named => |s| {
try std.testing.expectEqual(@as(u64, 1), s.from_id);
try std.testing.expectEqualStrings("test_actor", s.to_name);
try std.testing.expect(try cbor.match(s.payload, .{ "hello", "from_child" }));
},
}
self.frame_received = true;
}
} else if (try m.match(.{ tag, "term", "exited", extract(&exit_code) })) {
try std.testing.expect(self.frame_received);
try std.testing.expectEqual(@as(i64, 0), exit_code);
return thespian.exit("success");
} else if (try m.match(.{ tag, "stderr", extract(&bytes) })) {
// ignore stderr from child
} else {
return unexpected(m);
}
}
};
test "remote: child process sends a message to parent" {
const allocator = std.testing.allocator;
var ctx = try thespian.context.init(allocator);
defer ctx.deinit();
var success = false;
var exit_handler = thespian.make_exit_handler(&success, struct {
fn handle(ok: *bool, status: []const u8) void {
ok.* = std.mem.eql(u8, status, "success");
}
}.handle);
_ = try ctx.spawn_link(
Parent.Args{ .allocator = allocator },
Parent.start,
"remote_poc",
&exit_handler,
null,
);
ctx.run();
if (!success) return error.TestFailed;
}

View file

@ -2,6 +2,7 @@ const std = @import("std");
pub const cpp = @import("tests_cpp.zig");
pub const thespian = @import("tests_thespian.zig");
pub const ip_tcp_client_server = @import("ip_tcp_client_server.zig");
pub const remote_poc = @import("remote_poc_test.zig");
test {
std.testing.refAllDecls(@This());