Function testService

sdk-node/src/polling.test.ts:8–60

()

describe("Polling functions", () => {
  // Create a test service with functions we can call
  const testService = () => {
    const client = testInstance();
    const prefix = `test${Math.random().toString(36).substring(2, 15)}`;

    client.register({
      name: `${prefix}_echo`,
      handler: async (input: { text: string }) => {
        return { echo: input.text };
      },
      schema: z.object({
        text: z.string(),
      }),
    });

    client.register({
      name: `${prefix}_error`,
      handler: async (_input) => {
        throw new Error("This is an error");
      },
      schema: z.object({
        text: z.string(),
      }),
    });

    // Register a function that takes some time to complete
    client.register({
      name: `${prefix}_slow`,
      handler: async (input: { text: string; delay: number }) => {
        await new Promise((resolve) => setTimeout(resolve, input.delay));
        return { echo: input.text, delayed: true };
      },
      schema: z.object({
        text: z.string(),
        delay: z.number(),
      }),
    });

    // Register a function for testing invalid tools
    client.register({
      name: `${prefix}_valid`,
      handler: async (input: { text: string }) => {
        return { success: true, text: input.text };
      },
      schema: z.object({
        text: z.string(),
      }),
    });

    return {
      client,
      prefix,
    };
  };

  // Set up our test service
  const service = testService();