tiny.profiling.driver.environment

1const std = @import("std");2const sys = @import("sys");3const catalog = @import("../root.zig").catalog;4const execute = @import("../root.zig").execute;5const PreparedWorkload = @import("root.zig").model.PreparedWorkload;6const WorkloadContext = @import("root.zig").model.WorkloadContext;7const makeResetTestPaths = @import("fixture/root.zig").makeResetTestPaths;8const resetTestContext = @import("fixture/root.zig").resetTestContext;9const resetTestPrepared = @import("fixture/root.zig").resetTestPrepared;10const resetTestWorkload = @import("fixture/root.zig").resetTestWorkload;1112pub fn effectiveChildEnvironment(13    context: *const WorkloadContext,14    prepared: *const PreparedWorkload,15) ?*const sys.process.Environ.Map {16    if (prepared.child_environment) |*environment| {17        std.debug.assert(prepared.execution_plan.direct);18        return environment;19    }20    return context.environ_map;21}2223pub fn warmupBenchControl(control: execute.BenchControl) execute.BenchControl {24    return .{25        .min_time_ns = control.min_time_ns,26        .filter = control.filter,27    };28}2930pub fn workloadBenchControl(31    control: execute.BenchControl,32    environment: []const catalog.EnvironmentOverride,33) execute.BenchControl {34    if (control.filter != null) return control;35    var result = control;36    for (environment) |entry| {37        if (std.mem.eql(u8, entry.name, "BENCH_FILTER")) {38            result.filter = entry.value;39            break;40        }41    }42    return result;43}4445test "profiling driver warmups preserve workload controls but disable causal experiments" {46    const control = warmupBenchControl(.{47        .causal = true,48        .min_time_ns = 30 * std.time.ns_per_s,49        .filter = "small alloc",50    });51    try std.testing.expect(!control.causal);52    try std.testing.expectEqual(@as(?u64, 30 * std.time.ns_per_s), control.min_time_ns);53    try std.testing.expectEqualStrings("small alloc", control.filter.?);54}5556test "profiling driver composes catalog and CLI benchmark filters" {57    const environment = [_]catalog.EnvironmentOverride{58        .{ .name = "HOME", .value = "/fixture" },59        .{ .name = "BENCH_FILTER", .value = "catalog lane" },60    };61    const catalog_control = workloadBenchControl(.{}, &environment);62    try std.testing.expectEqualStrings("catalog lane", catalog_control.filter.?);63    const cli_control = workloadBenchControl(.{ .filter = "CLI lane" }, &environment);64    try std.testing.expectEqualStrings("CLI lane", cli_control.filter.?);65    try std.testing.expectEqualStrings("/fixture", environment[0].value.?);66    try std.testing.expect(workloadBenchControl(.{}, environment[0..1]).filter == null);67}6869test "profiling driver preserves inherited environment without child overrides" {70    var tmp = std.testing.tmpDir(.{});71    defer tmp.cleanup();72    var arena_state = std.heap.ArenaAllocator.init(std.testing.allocator);73    defer arena_state.deinit();74    const allocator = arena_state.allocator();75    const root = try tmp.parent_dir.realPathFileAlloc(76        std.testing.io,77        tmp.sub_path[0..],78        allocator,79    );80    const paths = try makeResetTestPaths(allocator, root);81    var base_environment = sys.process.Environ.Map.init(allocator);82    try base_environment.put("PATH", "/usr/bin:/bin");83    const environment = [_]catalog.EnvironmentOverride{.{84        .name = "HOME",85        .value = root,86    }};87    const workload = resetTestWorkload(&.{"/bin/true"}, &environment);88    var prepared = try resetTestPrepared(89        allocator,90        &base_environment,91        workload,92        paths.workload,93        &.{"/bin/true"},94    );95    var context = resetTestContext(allocator, &base_environment, paths.run, 0, 1);96    try std.testing.expectEqualStrings(97        root,98        effectiveChildEnvironment(&context, &prepared).?.get("HOME").?,99    );100    prepared.child_environment = null;101    try std.testing.expectEqualStrings(102        "/usr/bin:/bin",103        effectiveChildEnvironment(&context, &prepared).?.get("PATH").?,104    );105    context.environ_map = null;106    try std.testing.expect(effectiveChildEnvironment(&context, &prepared) == null);107}

1pub const environment = @import("environment.zig");