import { BaseCheckpointSaver } from "@langchain/langgraph-checkpoint";
import { z } from "zod/v3";

//#region src/types.d.ts
interface CheckpointerTestInitializer<CheckpointerT extends BaseCheckpointSaver> {
  /**
   * The name of the checkpointer being tested. This will be used to identify the checkpointer in test output.
   */
  checkpointerName: string;
  /**
   * Called once before any tests are run. Use this to perform any setup that your checkpoint checkpointer may require, like
   * starting docker containers, etc.
   */
  beforeAll?(): void | Promise<void>;
  /**
   * Optional timeout for beforeAll. Useful for test setups that might take a while to complete, e.g. due to needing to
   * pull a docker container.
   *
   * @default 10000
   */
  beforeAllTimeout?: number;
  /**
   * Called once after all tests are run. Use this to perform any infrastructure cleanup that your checkpointer may
   * require, like tearing down docker containers, etc.
   */
  afterAll?(): void | Promise<void>;
  /**
   * Called before each set of validations is run. The checkpointer returned will be used during test execution.
   *
   * @returns A new checkpointer, or promise that resolves to a new checkpointer.
   */
  createCheckpointer(): CheckpointerT | Promise<CheckpointerT>;
  /**
   * Called after each set of validations is run. Use this to clean up any resources that your checkpointer may
   * have been using. This should include cleaning up any state that the checkpointer wrote during the tests that just ran.
   *
   * @param checkpointer The @see BaseCheckpointSaver that was used during the test.
   */
  destroyCheckpointer?(checkpointer: CheckpointerT): void | Promise<void>;
}
declare const testTypeFilters: readonly ["getTuple", "list", "put", "putWrites", "deleteThread"];
type TestTypeFilter = (typeof testTypeFilters)[number];
//#endregion
export { CheckpointerTestInitializer, TestTypeFilter };
//# sourceMappingURL=types.d.ts.map