@flink-app/test-utils 1.0.0 → 2.0.0-alpha.49

package/dist/ai/mockTool.js

@@ -0,0 +1,133 @@
+"use strict";
+var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
+    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
+    return new (P || (P = Promise))(function (resolve, reject) {
+        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
+        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
+        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
+        step((generator = generator.apply(thisArg, _arguments || [])).next());
+    });
+};
+var __generator = (this && this.__generator) || function (thisArg, body) {
+    var _ = { label: 0, sent: function() { if (t[0] & 1) throw t[1]; return t[1]; }, trys: [], ops: [] }, f, y, t, g = Object.create((typeof Iterator === "function" ? Iterator : Object).prototype);
+    return g.next = verb(0), g["throw"] = verb(1), g["return"] = verb(2), typeof Symbol === "function" && (g[Symbol.iterator] = function() { return this; }), g;
+    function verb(n) { return function (v) { return step([n, v]); }; }
+    function step(op) {
+        if (f) throw new TypeError("Generator is already executing.");
+        while (g && (g = 0, op[0] && (_ = 0)), _) try {
+            if (f = 1, y && (t = op[0] & 2 ? y["return"] : op[0] ? y["throw"] || ((t = y["return"]) && t.call(y), 0) : y.next) && !(t = t.call(y, op[1])).done) return t;
+            if (y = 0, t) op = [op[0] & 2, t.value];
+            switch (op[0]) {
+                case 0: case 1: t = op; break;
+                case 4: _.label++; return { value: op[1], done: false };
+                case 5: _.label++; y = op[1]; op = [0]; continue;
+                case 7: op = _.ops.pop(); _.trys.pop(); continue;
+                default:
+                    if (!(t = _.trys, t = t.length > 0 && t[t.length - 1]) && (op[0] === 6 || op[0] === 2)) { _ = 0; continue; }
+                    if (op[0] === 3 && (!t || (op[1] > t[0] && op[1] < t[3]))) { _.label = op[1]; break; }
+                    if (op[0] === 6 && _.label < t[1]) { _.label = t[1]; t = op; break; }
+                    if (t && _.label < t[2]) { _.label = t[2]; _.ops.push(op); break; }
+                    if (t[2]) _.ops.pop();
+                    _.trys.pop(); continue;
+            }
+            op = body.call(thisArg, _);
+        } catch (e) { op = [6, e]; y = 0; } finally { f = t = 0; }
+        if (op[0] & 5) throw op[1]; return { value: op[0] ? op[1] : void 0, done: true };
+    }
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.mockTool = mockTool;
+/**
+ * Creates a mock tool with tracking and canned responses
+ *
+ * Features:
+ * - Simple canned responses
+ * - Error simulation
+ * - Custom function support
+ * - Automatic invocation tracking
+ * - Validation helpers
+ *
+ * @example
+ * // Simple canned response
+ * const weatherTool = mockTool({
+ *   name: "get_weather",
+ *   inputSchema: z.object({ city: z.string() }),
+ *   response: { temperature: 22, conditions: "sunny" }
+ * });
+ *
+ * @example
+ * // Custom function with tracking
+ * const calculatorTool = mockTool({
+ *   name: "calculate",
+ *   inputSchema: z.object({ a: z.number(), b: z.number() }),
+ *   fn: async ({ input }) => ({
+ *     success: true,
+ *     data: { result: input.a + input.b }
+ *   })
+ * });
+ *
+ * @example
+ * // Error simulation
+ * const failingTool = mockTool({
+ *   name: "fail",
+ *   inputSchema: z.object({}),
+ *   error: { error: "Tool failed", code: "MOCK_ERROR" }
+ * });
+ */
+function mockTool(config) {
+    var _this = this;
+    var invocations = [];
+    // Create the tool props
+    var props = {
+        id: config.name,
+        description: config.description || "Mock tool: ".concat(config.name),
+        inputSchema: config.inputSchema,
+        outputSchema: config.outputSchema,
+        permissions: config.permissions,
+    };
+    // Create the tool function with invocation tracking
+    var fn = function (_a) { return __awaiter(_this, [_a], void 0, function (_b) {
+        var input = _b.input, ctx = _b.ctx, user = _b.user;
+        return __generator(this, function (_c) {
+            // Track invocation
+            invocations.push({ input: input, user: user });
+            // If custom function provided, use it
+            if (config.fn) {
+                return [2 /*return*/, config.fn({ input: input, ctx: ctx, user: user })];
+            }
+            // If error configured, return error
+            if (config.error) {
+                return [2 /*return*/, {
+                        success: false,
+                        error: config.error.error,
+                        code: config.error.code,
+                    }];
+            }
+            // If response configured, return success with data
+            if (config.response !== undefined) {
+                return [2 /*return*/, {
+                        success: true,
+                        data: config.response,
+                    }];
+            }
+            // Default: return empty success
+            return [2 /*return*/, {
+                    success: true,
+                    data: {},
+                }];
+        });
+    }); };
+    var getLastInvocation = function () {
+        return invocations[invocations.length - 1];
+    };
+    var reset = function () {
+        invocations.length = 0;
+    };
+    return {
+        props: props,
+        fn: fn,
+        invocations: invocations,
+        getLastInvocation: getLastInvocation,
+        reset: reset,
+    };
+}

package/dist/index.d.ts

@@ -1,2 +1,3 @@
 export * from "./http";
 export * from "./mocks";
+export * from "./ai";

package/dist/index.js

@@ -16,3 +16,4 @@ var __exportStar = (this && this.__exportStar) || function(m, exports) {
 Object.defineProperty(exports, "__esModule", { value: true });
 __exportStar(require("./http"), exports);
 __exportStar(require("./mocks"), exports);
+__exportStar(require("./ai"), exports);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@flink-app/test-utils",
-  "version": "1.0.0",
+  "version": "2.0.0-alpha.49",
   "description": "Test utils for Flink",
   "author": "joel@frost.se",
   "license": "MIT",
@@ -11,20 +11,24 @@
   },
   "dependencies": {
     "got": "^9.6.0",
-    "qs": "^6.7.0"
+    "qs": "^6.7.0",
+    "zod": "^4.3.6"
   },
   "devDependencies": {
     "@types/got": "^9.6.12",
+    "@types/jasmine": "^3.7.1",
     "@types/node": "22.13.10",
     "@types/qs": "^6.9.7",
+    "jasmine": "^3.10.0",
+    "jasmine-ts": "^0.3.3",
     "ts-node": "^10.9.2",
     "tsc-watch": "^4.2.9",
-    "@flink-app/flink": "1.0.0"
+    "@flink-app/flink": "2.0.0-alpha.49"
   },
   "gitHead": "4243e3b3cd6d4e1ca001a61baa8436bf2bbe4113",
   "scripts": {
-    "test": "echo \"Error: no test specified\"",
-    "build": "tsc",
+    "test": "jasmine-ts --config=./spec/support/jasmine.json",
+    "build": "tsc --project tsconfig.dist.json",
     "watch": "tsc-watch",
     "clean": "rimraf dist .flink"
   }

package/spec/ai/conversation.spec.ts

@@ -0,0 +1,280 @@
+import { createConversation } from "../../src/ai/conversation";
+
+describe("createConversation", () => {
+  describe("user messages", () => {
+    it("should add user message", () => {
+      const conversation = createConversation().user("Hello").build();
+
+      expect(conversation.length).toBe(1);
+      expect(conversation[0]).toEqual({
+        role: "user",
+        content: "Hello",
+      });
+    });
+
+    it("should add multiple user messages", () => {
+      const conversation = createConversation()
+        .user("First message")
+        .user("Second message")
+        .build();
+
+      expect(conversation.length).toBe(2);
+      expect(conversation[0]).toEqual({
+        role: "user",
+        content: "First message",
+      });
+      expect(conversation[1]).toEqual({
+        role: "user",
+        content: "Second message",
+      });
+    });
+  });
+
+  describe("assistant messages", () => {
+    it("should add assistant message without tool calls", () => {
+      const conversation = createConversation()
+        .assistant("I can help with that")
+        .build();
+
+      expect(conversation.length).toBe(1);
+      expect(conversation[0]).toEqual({
+        role: "assistant",
+        content: "I can help with that",
+        toolCalls: undefined,
+      });
+    });
+
+    it("should add assistant message with tool calls", () => {
+      const conversation = createConversation()
+        .assistant("Let me check", [
+          { id: "1", name: "get_weather", input: { city: "Stockholm" } },
+        ])
+        .build();
+
+      expect(conversation.length).toBe(1);
+      expect(conversation[0]).toEqual({
+        role: "assistant",
+        content: "Let me check",
+        toolCalls: [
+          { id: "1", name: "get_weather", input: { city: "Stockholm" } },
+        ],
+      });
+    });
+
+    it("should add multiple assistant messages", () => {
+      const conversation = createConversation()
+        .assistant("First response")
+        .assistant("Second response")
+        .build();
+
+      expect(conversation.length).toBe(2);
+      expect(conversation[0].role).toBe("assistant");
+      expect(conversation[1].role).toBe("assistant");
+    });
+  });
+
+  describe("tool messages", () => {
+    it("should add tool message", () => {
+      const conversation = createConversation()
+        .tool("1", "get_weather", JSON.stringify({ temp: 22 }))
+        .build();
+
+      expect(conversation.length).toBe(1);
+      expect(conversation[0]).toEqual({
+        role: "tool",
+        toolCallId: "1",
+        toolName: "get_weather",
+        result: JSON.stringify({ temp: 22 }),
+      });
+    });
+
+    it("should add multiple tool messages", () => {
+      const conversation = createConversation()
+        .tool("1", "tool_one", "result 1")
+        .tool("2", "tool_two", "result 2")
+        .build();
+
+      expect(conversation.length).toBe(2);
+      expect(conversation[0].role).toBe("tool");
+      expect(conversation[1].role).toBe("tool");
+    });
+  });
+
+  describe("fluent API", () => {
+    it("should chain different message types", () => {
+      const conversation = createConversation()
+        .user("What's the weather?")
+        .assistant("Let me check", [
+          { id: "1", name: "get_weather", input: { city: "Stockholm" } },
+        ])
+        .tool("1", "get_weather", JSON.stringify({ temp: 22 }))
+        .assistant("It's 22°C in Stockholm")
+        .build();
+
+      expect(conversation.length).toBe(4);
+      expect(conversation[0].role).toBe("user");
+      expect(conversation[1].role).toBe("assistant");
+      expect(conversation[2].role).toBe("tool");
+      expect(conversation[3].role).toBe("assistant");
+    });
+
+    it("should return builder instance for chaining", () => {
+      const builder = createConversation();
+      const result1 = builder.user("test");
+      const result2 = result1.assistant("response");
+      const result3 = result2.tool("1", "tool", "result");
+
+      expect(result1).toBe(builder);
+      expect(result2).toBe(builder);
+      expect(result3).toBe(builder);
+    });
+  });
+
+  describe("multi-turn conversations", () => {
+    it("should build complex multi-turn conversation", () => {
+      const conversation = createConversation()
+        .user("Book a flight to Paris")
+        .assistant("When would you like to travel?")
+        .user("Next Monday")
+        .assistant("Let me search for flights", [
+          {
+            id: "search_1",
+            name: "search_flights",
+            input: { destination: "Paris", date: "next Monday" },
+          },
+        ])
+        .tool(
+          "search_1",
+          "search_flights",
+          JSON.stringify({ flights: ["AF123", "BA456"] })
+        )
+        .assistant("I found these flights: AF123 and BA456")
+        .user("Book AF123")
+        .assistant("Booking flight AF123", [
+          { id: "book_1", name: "book_flight", input: { flightId: "AF123" } },
+        ])
+        .tool("book_1", "book_flight", JSON.stringify({ success: true }))
+        .assistant("Your flight has been booked!")
+        .build();
+
+      expect(conversation.length).toBe(10);
+
+      // Verify structure
+      expect(conversation[0]).toEqual({
+        role: "user",
+        content: "Book a flight to Paris",
+      });
+      expect(conversation[1]).toEqual({
+        role: "assistant",
+        content: "When would you like to travel?",
+        toolCalls: undefined,
+      });
+      expect(conversation[9]).toEqual({
+        role: "assistant",
+        content: "Your flight has been booked!",
+        toolCalls: undefined,
+      });
+    });
+  });
+
+  describe("edge cases", () => {
+    it("should build empty conversation", () => {
+      const conversation = createConversation().build();
+
+      expect(conversation).toEqual([]);
+    });
+
+    it("should handle empty strings", () => {
+      const conversation = createConversation()
+        .user("")
+        .assistant("")
+        .tool("1", "test", "")
+        .build();
+
+      expect(conversation.length).toBe(3);
+      expect((conversation[0] as any).content).toBe("");
+      expect((conversation[1] as any).content).toBe("");
+      expect((conversation[2] as any).result).toBe("");
+    });
+
+    it("should handle special characters in content", () => {
+      const specialContent = 'Hello "world" with \n newlines and \t tabs';
+      const conversation = createConversation()
+        .user(specialContent)
+        .assistant(specialContent)
+        .build();
+
+      expect((conversation[0] as any).content).toBe(specialContent);
+      expect((conversation[1] as any).content).toBe(specialContent);
+    });
+
+    it("should handle complex tool inputs", () => {
+      const conversation = createConversation()
+        .assistant("Processing", [
+          {
+            id: "complex_1",
+            name: "complex_tool",
+            input: {
+              nested: {
+                array: [1, 2, 3],
+                object: { key: "value" },
+              },
+              special: 'chars "quotes" \n newlines',
+            },
+          },
+        ])
+        .build();
+
+      const toolCall = (conversation[0] as any).toolCalls[0];
+      expect(toolCall.input.nested.array).toEqual([1, 2, 3]);
+      expect(toolCall.input.special).toContain("quotes");
+    });
+  });
+
+  describe("real-world usage patterns", () => {
+    it("should support weather agent conversation", () => {
+      const conversation = createConversation()
+        .user("What's the weather in Stockholm?")
+        .assistant("Let me check the weather for you", [
+          { id: "w1", name: "get_weather", input: { city: "Stockholm" } },
+        ])
+        .tool(
+          "w1",
+          "get_weather",
+          JSON.stringify({ temperature: 22, conditions: "sunny" })
+        )
+        .assistant("It's currently 22°C and sunny in Stockholm")
+        .build();
+
+      expect(conversation.length).toBe(4);
+    });
+
+    it("should support calculator agent conversation", () => {
+      const conversation = createConversation()
+        .user("What is 15 + 27?")
+        .assistant("Let me calculate that", [
+          { id: "c1", name: "calculate", input: { a: 15, b: 27 } },
+        ])
+        .tool("c1", "calculate", JSON.stringify({ result: 42 }))
+        .assistant("15 + 27 = 42")
+        .build();
+
+      expect(conversation.length).toBe(4);
+    });
+
+    it("should support multi-tool agent conversation", () => {
+      const conversation = createConversation()
+        .user("Get me the weather and news for Stockholm")
+        .assistant("I'll fetch both for you", [
+          { id: "w1", name: "get_weather", input: { city: "Stockholm" } },
+          { id: "n1", name: "get_news", input: { city: "Stockholm" } },
+        ])
+        .tool("w1", "get_weather", JSON.stringify({ temp: 22 }))
+        .tool("n1", "get_news", JSON.stringify({ headlines: ["News 1"] }))
+        .assistant("Weather: 22°C. Top news: News 1")
+        .build();
+
+      expect(conversation.length).toBe(5);
+    });
+  });
+});