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

package/src/ai/mockTool.ts

@@ -0,0 +1,135 @@
+import { z } from "zod";
+import { FlinkContext } from "@flink-app/flink";
+import type { FlinkTool, FlinkToolProps, ToolResult } from "@flink-app/flink/ai";
+
+export interface MockToolConfig<Input, Output> {
+  name: string;
+  description?: string;
+  inputSchema: z.ZodType<Input>;
+  outputSchema?: z.ZodType<Output>;
+
+  // Response strategies
+  response?: Output; // Canned success response
+  error?: { error: string; code?: string }; // Canned error response
+  fn?: FlinkTool<any, Input, Output>; // Custom function
+
+  // Permissions
+  permissions?: FlinkToolProps["permissions"];
+}
+
+export interface MockToolInvocation<Input> {
+  input: Input;
+  user?: any;
+}
+
+export interface MockToolResult<Input, Output> {
+  props: FlinkToolProps;
+  fn: FlinkTool<any, Input, Output>;
+
+  // Invocation tracking
+  invocations: MockToolInvocation<Input>[];
+  getLastInvocation(): MockToolInvocation<Input> | undefined;
+  reset(): void;
+}
+
+/**
+ * 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" }
+ * });
+ */
+export function mockTool<Input = any, Output = any>(
+  config: MockToolConfig<Input, Output>
+): MockToolResult<Input, Output> {
+  const invocations: MockToolInvocation<Input>[] = [];
+
+  // Create the tool props
+  const props: FlinkToolProps = {
+    id: config.name,
+    description: config.description || `Mock tool: ${config.name}`,
+    inputSchema: config.inputSchema,
+    outputSchema: config.outputSchema,
+    permissions: config.permissions,
+  };
+
+  // Create the tool function with invocation tracking
+  const fn: FlinkTool<any, Input, Output> = async ({ input, ctx, user }) => {
+    // Track invocation
+    invocations.push({ input, user });
+
+    // If custom function provided, use it
+    if (config.fn) {
+      return config.fn({ input, ctx, user });
+    }
+
+    // If error configured, return error
+    if (config.error) {
+      return {
+        success: false,
+        error: config.error.error,
+        code: config.error.code,
+      };
+    }
+
+    // If response configured, return success with data
+    if (config.response !== undefined) {
+      return {
+        success: true,
+        data: config.response,
+      };
+    }
+
+    // Default: return empty success
+    return {
+      success: true,
+      data: {} as Output,
+    };
+  };
+
+  const getLastInvocation = (): MockToolInvocation<Input> | undefined => {
+    return invocations[invocations.length - 1];
+  };
+
+  const reset = (): void => {
+    invocations.length = 0;
+  };
+
+  return {
+    props,
+    fn,
+    invocations,
+    getLastInvocation,
+    reset,
+  };
+}

package/src/index.ts

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

package/tsconfig.dist.json

@@ -0,0 +1,4 @@
+{
+  "extends": "./tsconfig.json",
+  "exclude": ["spec/**/*.ts", "node_modules/*", "dist/*"]
+}

package/tsconfig.json

@@ -8,16 +8,17 @@
     "allowSyntheticDefaultImports": true,
     "strict": true,
     "forceConsistentCasingInFileNames": true,
-    "module": "commonjs",
-    "moduleResolution": "node",
+    "module": "node16",
+    "moduleResolution": "node16",
     "resolveJsonModule": true,
     "isolatedModules": true,
     "noEmit": false,
     "declaration": true,
     "experimentalDecorators": true,
     "checkJs": false,
-    "outDir": "dist"
+    "outDir": "dist",
+    "types": ["jasmine", "node"]
   },
-  "include": ["./src/**/*.ts", "./.flink/**/*.ts"],
-  "exclude": ["./node_modules/*"]
+  "include": ["./src/**/*.ts", "./.flink/**/*.ts", "./spec/**/*.ts"],
+  "exclude": ["./node_modules/*", "./dist/*"]
 }