import { describe, expect, it } from "@effect/vitest";import * as Effect from "effect/Effect";import * as EffectAcpErrors from "effect-acp/errors"; import { applyGrokAcpModelSelection, buildGrokAcpSpawnInput, resolveGrokAcpBaseModelId,} from "./GrokAcpSupport.ts"; describe("resolveGrokAcpBaseModelId", () => { it("normalizes empty and custom Grok model ids", () => { expect(resolveGrokAcpBaseModelId(undefined)).toBe("grok-build"); expect(resolveGrokAcpBaseModelId(" ")).toBe("grok-build"); expect(resolveGrokAcpBaseModelId(" grok-test-custom-model ")).toBe("grok-test-custom-model"); });}); describe("buildGrokAcpSpawnInput", () => { it("passes the T3 Code referrer through Grok OAuth env", () => { const spawn = buildGrokAcpSpawnInput({ binaryPath: "/usr/local/bin/grok" }, "/tmp/project", { XAI_API_KEY: "secret", GROK_OAUTH2_REFERRER: "other-client", }); expect(spawn).toEqual({ command: "/usr/local/bin/grok", args: ["agent", "stdio"], cwd: "/tmp/project", env: { XAI_API_KEY: "secret", GROK_OAUTH2_REFERRER: "t3code", }, }); });}); describe("applyGrokAcpModelSelection", () => { const makeRecordingRuntime = (failure?: EffectAcpErrors.AcpError) => { const modelCalls: Array<string> = []; const runtime = { setSessionModel: (modelId: string) => Effect.gen(function* () { modelCalls.push(modelId); if (failure) return yield* failure; return {}; }), }; return { runtime, modelCalls }; }; it.effect("calls session/set_model when the requested model differs from current", () => Effect.gen(function* () { const { runtime, modelCalls } = makeRecordingRuntime(); const result = yield* applyGrokAcpModelSelection({ runtime, currentModelId: "grok-build", requestedModelId: "grok-mock-alt", mapError: (cause) => cause.message, }); expect(modelCalls).toEqual(["grok-mock-alt"]); expect(result).toBe("grok-mock-alt"); }), ); it.effect("skips set_model when requested matches current", () => Effect.gen(function* () { const { runtime, modelCalls } = makeRecordingRuntime(); const result = yield* applyGrokAcpModelSelection({ runtime, currentModelId: "grok-build", requestedModelId: "grok-build", mapError: (cause) => cause.message, }); expect(modelCalls).toEqual([]); expect(result).toBe("grok-build"); }), ); it.effect("skips set_model when no model is requested", () => Effect.gen(function* () { const { runtime, modelCalls } = makeRecordingRuntime(); const result = yield* applyGrokAcpModelSelection({ runtime, currentModelId: "grok-build", requestedModelId: undefined, mapError: (cause) => cause.message, }); expect(modelCalls).toEqual([]); expect(result).toBe("grok-build"); }), ); it.effect("propagates session/set_model failures via mapError", () => Effect.gen(function* () { const failure = EffectAcpErrors.AcpRequestError.invalidParams("session id not known"); const { runtime } = makeRecordingRuntime(failure); const error = yield* Effect.flip( applyGrokAcpModelSelection({ runtime, currentModelId: "grok-build", requestedModelId: "grok-mock-alt", mapError: (cause) => cause.message, }), ); expect(error).toBe(failure.message); }), );});
import { describe, expect, it } from "@effect/vitest";
import * as Effect from "effect/Effect";
import * as EffectAcpErrors from "effect-acp/errors";
import {
applyGrokAcpModelSelection,
buildGrokAcpSpawnInput,
resolveGrokAcpBaseModelId,
} from "./GrokAcpSupport.ts";
describe("resolveGrokAcpBaseModelId", () => {
it("normalizes empty and custom Grok model ids", () => {
expect(resolveGrokAcpBaseModelId(undefined)).toBe("grok-build");
expect(resolveGrokAcpBaseModelId(" ")).toBe("grok-build");
expect(resolveGrokAcpBaseModelId(" grok-test-custom-model ")).toBe("grok-test-custom-model");
});
});
describe("buildGrokAcpSpawnInput", () => {
it("passes the T3 Code referrer through Grok OAuth env", () => {
const spawn = buildGrokAcpSpawnInput({ binaryPath: "/usr/local/bin/grok" }, "/tmp/project", {
XAI_API_KEY: "secret",
GROK_OAUTH2_REFERRER: "other-client",
});
expect(spawn).toEqual({
command: "/usr/local/bin/grok",
args: ["agent", "stdio"],
cwd: "/tmp/project",
env: {
XAI_API_KEY: "secret",
GROK_OAUTH2_REFERRER: "t3code",
},
});
});
});
describe("applyGrokAcpModelSelection", () => {
const makeRecordingRuntime = (failure?: EffectAcpErrors.AcpError) => {
const modelCalls: Array<string> = [];
const runtime = {
setSessionModel: (modelId: string) =>
Effect.gen(function* () {
modelCalls.push(modelId);
if (failure) return yield* failure;
return {};
}),
};
return { runtime, modelCalls };
};
it.effect("calls session/set_model when the requested model differs from current", () =>
Effect.gen(function* () {
const { runtime, modelCalls } = makeRecordingRuntime();
const result = yield* applyGrokAcpModelSelection({
runtime,
currentModelId: "grok-build",
requestedModelId: "grok-mock-alt",
mapError: (cause) => cause.message,
});
expect(modelCalls).toEqual(["grok-mock-alt"]);
expect(result).toBe("grok-mock-alt");
}),
);
it.effect("skips set_model when requested matches current", () =>
Effect.gen(function* () {
const { runtime, modelCalls } = makeRecordingRuntime();
const result = yield* applyGrokAcpModelSelection({
runtime,
currentModelId: "grok-build",
requestedModelId: "grok-build",
mapError: (cause) => cause.message,
});
expect(modelCalls).toEqual([]);
expect(result).toBe("grok-build");
}),
);
it.effect("skips set_model when no model is requested", () =>
Effect.gen(function* () {
const { runtime, modelCalls } = makeRecordingRuntime();
const result = yield* applyGrokAcpModelSelection({
runtime,
currentModelId: "grok-build",
requestedModelId: undefined,
mapError: (cause) => cause.message,
});
expect(modelCalls).toEqual([]);
expect(result).toBe("grok-build");
}),
);
it.effect("propagates session/set_model failures via mapError", () =>
Effect.gen(function* () {
const failure = EffectAcpErrors.AcpRequestError.invalidParams("session id not known");
const { runtime } = makeRecordingRuntime(failure);
const error = yield* Effect.flip(
applyGrokAcpModelSelection({
runtime,
currentModelId: "grok-build",
requestedModelId: "grok-mock-alt",
mapError: (cause) => cause.message,
}),
);
expect(error).toBe(failure.message);
}),
);
});