@posthog/agent 2.0.0 → 2.0.2

package/src/server/agent-server.test.ts

@@ -0,0 +1,220 @@
+import jwt from "jsonwebtoken";
+import { type SetupServerApi, setupServer } from "msw/node";
+import { afterEach, beforeEach, describe, expect, it } from "vitest";
+import { createTestRepo, type TestRepo } from "../test/fixtures/api.js";
+import { createPostHogHandlers } from "../test/mocks/msw-handlers.js";
+import { AgentServer } from "./agent-server.js";
+import { type JwtPayload, SANDBOX_CONNECTION_AUDIENCE } from "./jwt.js";
+
+function createTestJwt(
+  payload: JwtPayload,
+  privateKey: string,
+  expiresInSeconds = 3600,
+): string {
+  return jwt.sign(
+    { ...payload, aud: SANDBOX_CONNECTION_AUDIENCE },
+    privateKey,
+    {
+      algorithm: "RS256",
+      expiresIn: expiresInSeconds,
+    },
+  );
+}
+
+// Test RSA key pair (2048-bit, for testing only)
+const TEST_PRIVATE_KEY = `-----BEGIN PRIVATE KEY-----
+MIIEvQIBADANBgkqhkiG9w0BAQEFAASCBKcwggSjAgEAAoIBAQDqh94SYMFsvG4C
+Co9BSGjtPr2/OxzuNGr41O4+AMkDQRd9pKO49DhTA4VzwnOvrH8y4eI9N8OQne7B
+wpdoouSn4DoDAS/b3SUfij/RoFUSyZiTQoWz0H6o2Vuufiz0Hf+BzlZEVnhSQ1ru
+vqSf+4l8cWgeMXaFXgdD5kQ8GjvR5uqKxvO2Env1hMJRKeOOEGgCep/0c6SkMUTX
+SeC+VjypVg9+8yPxtIpOQ7XKv+7e/PA0ilqehRQh4fo9BAWjUW1+HnbtsjJAjjfv
+ngzIjpajuQVyMi7G79v8OvijhLMJjJBh3TdbVIfi+RkVj/H94UUfKWRfJA0eLykA
+VvTiFf0nAgMBAAECggEABkLBQWFW2IXBNAm/IEGEF408uH2l/I/mqSTaBUq1EwKq
+U17RRg8y77hg2CHBP9fNf3i7NuIltNcaeA6vRwpOK1MXiVv/QJHLO2fP41Mx4jIC
+gi/c7NtsfiprQaG5pnykhP0SnXlndd65bzUkpOasmWdXnbK5VL8ZV40uliInJafE
+1Eo9qSYCJxHmivU/4AbiBgygOAo1QIiuuUHcx0YGknLrBaMQETuvWJGE3lxVQ30/
+EuRyA3r6BwN2T0z47PZBzvCpg/C1KeoYuKSMwMyEXfl+a8NclqdROkVaenmZpvVH
+0lAvFDuPrBSDmU4XJbKCEfwfHjRkiWAFaTrKntGQtQKBgQD/ILoK4U9DkJoKTYvY
+9lX7dg6wNO8jGLHNufU8tHhU+QnBMH3hBXrAtIKQ1sGs+D5rq/O7o0Balmct9vwb
+CQZ1EpPfa83Thsv6Skd7lWK0JF7g2vVk8kT4nY/eqkgZUWgkfdMp+OMg2drYiIE8
+u+sRPTCdq4Tv5miRg0OToX2H/QKBgQDrVR2GXm6ZUyFbCy8A0kttXP1YyXqDVq7p
+L4kqyUq43hmbjzIRM4YDN3EvgZvVf6eub6L/3HfKvWD/OvEhHovTvHb9jkwZ3FO+
+YQllB/ccAWJs/Dw5jLAsX9O+eIe4lfwROib3vYLnDTAmrXD5VL35R5F0MsdRoxk5
+lTCq1sYI8wKBgGA9ZjDIgXAJUjJkwkZb1l9/T1clALiKjjf+2AXIRkQ3lXhs5G9H
+8+BRt5cPjAvFsTZIrS6xDIufhNiP/NXt96OeGG4FaqVKihOmhYSW+57cwXWs4zjr
+Mx1dwnHKZlw2m0R4unlwy60OwUFBbQ8ODER6gqZXl1Qv5G5Px+Qe3Q25AoGAUl+s
+wgfz9r9egZvcjBEQTeuq0pVTyP1ipET7YnqrKSK1G/p3sAW09xNFDzfy8DyK2UhC
+agUl+VVoym47UTh8AVWK4R4aDUNOHOmifDbZjHf/l96CxjI0yJOSbq2J9FarsOwG
+D9nKJE49eIxlayD6jnM6us27bxwEDF/odSRQlXkCgYEAxn9l/5kewWkeEA0Afe1c
+Uf+mepHBLw1Pbg5GJYIZPC6e5+wRNvtFjM5J6h5LVhyb7AjKeLBTeohoBKEfUyUO
+rl/ql9qDIh5lJFn3uNh7+r7tmG21Zl2pyh+O8GljjZ25mYhdiwl0uqzVZaINe2Wa
+vbMnD1ZQKgL8LHgb02cbTsc=
+-----END PRIVATE KEY-----`;
+
+const TEST_PUBLIC_KEY = `-----BEGIN PUBLIC KEY-----
+MIIBIjANBgkqhkiG9w0BAQEFAAOCAQ8AMIIBCgKCAQEA6ofeEmDBbLxuAgqPQUho
+7T69vzsc7jRq+NTuPgDJA0EXfaSjuPQ4UwOFc8Jzr6x/MuHiPTfDkJ3uwcKXaKLk
+p+A6AwEv290lH4o/0aBVEsmYk0KFs9B+qNlbrn4s9B3/gc5WRFZ4UkNa7r6kn/uJ
+fHFoHjF2hV4HQ+ZEPBo70ebqisbzthJ79YTCUSnjjhBoAnqf9HOkpDFE10ngvlY8
+qVYPfvMj8bSKTkO1yr/u3vzwNIpanoUUIeH6PQQFo1Ftfh527bIyQI43754MyI6W
+o7kFcjIuxu/b/Dr4o4SzCYyQYd03W1SH4vkZFY/x/eFFHylkXyQNHi8pAFb04hX9
+JwIDAQAB
+-----END PUBLIC KEY-----`;
+
+describe("AgentServer HTTP Mode", () => {
+  let repo: TestRepo;
+  let server: AgentServer;
+  let mswServer: SetupServerApi;
+  let appendLogCalls: unknown[][];
+  const port = 3099;
+
+  beforeEach(async () => {
+    repo = await createTestRepo("agent-server-http");
+    appendLogCalls = [];
+    mswServer = setupServer(
+      ...createPostHogHandlers({
+        baseUrl: "http://localhost:8000",
+        onAppendLog: (entries) => appendLogCalls.push(entries),
+      }),
+    );
+    mswServer.listen({ onUnhandledRequest: "bypass" });
+  });
+
+  afterEach(async () => {
+    if (server) {
+      await server.stop();
+    }
+    mswServer.close();
+    await repo.cleanup();
+  });
+
+  const createServer = () => {
+    server = new AgentServer({
+      port,
+      jwtPublicKey: TEST_PUBLIC_KEY,
+      repositoryPath: repo.path,
+      apiUrl: "http://localhost:8000",
+      apiKey: "test-api-key",
+      projectId: 1,
+      mode: "interactive",
+      taskId: "test-task-id",
+      runId: "test-run-id",
+    });
+    return server;
+  };
+
+  const createToken = (overrides = {}) => {
+    return createTestJwt(
+      {
+        run_id: "test-run-id",
+        task_id: "test-task-id",
+        team_id: 1,
+        user_id: 1,
+        distinct_id: "test-distinct-id",
+        mode: "interactive",
+        ...overrides,
+      },
+      TEST_PRIVATE_KEY,
+    );
+  };
+
+  describe("GET /health", () => {
+    it("returns ok status with active session", async () => {
+      await createServer().start();
+
+      const response = await fetch(`http://localhost:${port}/health`);
+      const body = await response.json();
+
+      expect(response.status).toBe(200);
+      expect(body).toEqual({ status: "ok", hasSession: true });
+    });
+  });
+
+  describe("GET /events", () => {
+    it("returns 401 without authorization header", async () => {
+      await createServer().start();
+
+      const response = await fetch(`http://localhost:${port}/events`);
+      const body = await response.json();
+
+      expect(response.status).toBe(401);
+      expect(body.error).toBe("Missing authorization header");
+    });
+
+    it("returns 401 with invalid token", async () => {
+      await createServer().start();
+
+      const response = await fetch(`http://localhost:${port}/events`, {
+        headers: { Authorization: "Bearer invalid-token" },
+      });
+      const body = await response.json();
+
+      expect(response.status).toBe(401);
+      expect(body.code).toBe("invalid_signature");
+    });
+
+    it("accepts valid JWT and returns SSE stream", async () => {
+      await createServer().start();
+      const token = createToken();
+
+      const response = await fetch(`http://localhost:${port}/events`, {
+        headers: { Authorization: `Bearer ${token}` },
+      });
+
+      expect(response.status).toBe(200);
+      expect(response.headers.get("content-type")).toBe("text/event-stream");
+    });
+  });
+
+  describe("POST /command", () => {
+    it("returns 401 without authorization", async () => {
+      await createServer().start();
+
+      const response = await fetch(`http://localhost:${port}/command`, {
+        method: "POST",
+        headers: { "Content-Type": "application/json" },
+        body: JSON.stringify({
+          jsonrpc: "2.0",
+          method: "user_message",
+          params: { content: "test" },
+        }),
+      });
+
+      expect(response.status).toBe(401);
+    });
+
+    it("returns 400 when run_id does not match active session", async () => {
+      await createServer().start();
+      const token = createToken({ run_id: "different-run-id" });
+
+      const response = await fetch(`http://localhost:${port}/command`, {
+        method: "POST",
+        headers: {
+          Authorization: `Bearer ${token}`,
+          "Content-Type": "application/json",
+        },
+        body: JSON.stringify({
+          jsonrpc: "2.0",
+          method: "user_message",
+          params: { content: "test" },
+        }),
+      });
+
+      expect(response.status).toBe(400);
+      const body = await response.json();
+      expect(body.error).toBe("No active session for this run");
+    });
+  });
+
+  describe("404 handling", () => {
+    it("returns 404 for unknown routes", async () => {
+      await createServer().start();
+
+      const response = await fetch(`http://localhost:${port}/unknown`);
+      const body = await response.json();
+
+      expect(response.status).toBe(404);
+      expect(body.error).toBe("Not found");
+    });
+  });
+});