@fragno-dev/test 1.0.2 → 2.0.2

package/dist/db-test.d.ts

@@ -1,24 +1,102 @@
 import { AdapterContext, SupportedAdapter } from "./adapters.js";
 import { BaseTestContext } from "./index.js";
-import { DatabaseAdapter } from "@fragno-dev/db";
+import { DatabaseAdapter, FragnoPublicConfigWithDatabase } from "@fragno-dev/db";
 import { AnySchema } from "@fragno-dev/db/schema";
 import { SimpleQueryInterface } from "@fragno-dev/db/query";
-import { FragmentInstantiationBuilder, FragnoInstantiatedFragment, FragnoPublicConfig, RequestThisContext } from "@fragno-dev/core";
 import { AnyRouteOrFactory, FlattenRouteFactories } from "@fragno-dev/core/route";
+import { FragmentDefinition, FragmentInstantiationBuilder, FragnoInstantiatedFragment, FragnoPublicConfig, RequestThisContext } from "@fragno-dev/core";
 
 //#region src/db-test.d.ts
 type BoundServices<T> = { [K in keyof T]: T[K] extends ((this: any, ...args: infer A) => infer R) ? (...args: A) => R : T[K] extends Record<string, unknown> ? BoundServices<T[K]> : T[K] };
-type ExtractSchemaFromDeps<TDeps> = TDeps extends {
+type FragmentFactoryContext = {
+  adapter: DatabaseAdapter<any>;
+  test: BaseTestContext & AdapterContext<SupportedAdapter>;
+};
+type FragmentFactoryResult = FragmentInstantiationBuilder<any,
+// eslint-disable-line @typescript-eslint/no-explicit-any
+any,
+// eslint-disable-line @typescript-eslint/no-explicit-any
+any,
+// eslint-disable-line @typescript-eslint/no-explicit-any
+any,
+// eslint-disable-line @typescript-eslint/no-explicit-any
+any,
+// eslint-disable-line @typescript-eslint/no-explicit-any
+any,
+// eslint-disable-line @typescript-eslint/no-explicit-any
+any,
+// eslint-disable-line @typescript-eslint/no-explicit-any
+any,
+// eslint-disable-line @typescript-eslint/no-explicit-any
+any,
+// eslint-disable-line @typescript-eslint/no-explicit-any
+any,
+// eslint-disable-line @typescript-eslint/no-explicit-any
+any,
+// eslint-disable-line @typescript-eslint/no-explicit-any
+any> | FragnoInstantiatedFragment<any,
+// eslint-disable-line @typescript-eslint/no-explicit-any
+any,
+// eslint-disable-line @typescript-eslint/no-explicit-any
+any,
+// eslint-disable-line @typescript-eslint/no-explicit-any
+any,
+// eslint-disable-line @typescript-eslint/no-explicit-any
+any,
+// eslint-disable-line @typescript-eslint/no-explicit-any
+any,
+// eslint-disable-line @typescript-eslint/no-explicit-any
+any>;
+type HandlerThisContextFromFactoryResult<T> = T extends FragmentInstantiationBuilder<any,
+// eslint-disable-line @typescript-eslint/no-explicit-any
+any,
+// eslint-disable-line @typescript-eslint/no-explicit-any
+any,
+// eslint-disable-line @typescript-eslint/no-explicit-any
+any,
+// eslint-disable-line @typescript-eslint/no-explicit-any
+any,
+// eslint-disable-line @typescript-eslint/no-explicit-any
+any,
+// eslint-disable-line @typescript-eslint/no-explicit-any
+any,
+// eslint-disable-line @typescript-eslint/no-explicit-any
+any,
+// eslint-disable-line @typescript-eslint/no-explicit-any
+infer THandlerThisContext, any,
+// eslint-disable-line @typescript-eslint/no-explicit-any
+any,
+// eslint-disable-line @typescript-eslint/no-explicit-any
+any> ? THandlerThisContext : T extends FragnoInstantiatedFragment<any,
+// eslint-disable-line @typescript-eslint/no-explicit-any
+any,
+// eslint-disable-line @typescript-eslint/no-explicit-any
+any,
+// eslint-disable-line @typescript-eslint/no-explicit-any
+any,
+// eslint-disable-line @typescript-eslint/no-explicit-any
+infer THandlerThisContext, any,
+// eslint-disable-line @typescript-eslint/no-explicit-any
+any> ? THandlerThisContext : RequestThisContext;
+type FragmentResultFromFactoryResult<T> = T extends FragmentInstantiationBuilder<any,
+// eslint-disable-line @typescript-eslint/no-explicit-any
+any,
+// eslint-disable-line @typescript-eslint/no-explicit-any
+infer TDeps, infer TBaseServices, infer TServices, any,
+// eslint-disable-line @typescript-eslint/no-explicit-any
+any,
+// eslint-disable-line @typescript-eslint/no-explicit-any
+infer TServiceThisContext, infer THandlerThisContext, infer TRequestStorage, infer TRoutesOrFactories, any> ? FragmentResult<TDeps, BoundServices<TBaseServices & TServices>, TServiceThisContext, THandlerThisContext, TRequestStorage, FlattenRouteFactories<TRoutesOrFactories>, ExtractSchemaFromDeps<TDeps>> : T extends FragnoInstantiatedFragment<infer TRoutes, infer TDeps, infer TServices, infer TServiceThisContext, infer THandlerThisContext, infer TRequestStorage, any> ? FragmentResult<TDeps, BoundServices<TServices>, TServiceThisContext, THandlerThisContext, TRequestStorage, TRoutes, ExtractSchemaFromDeps<TDeps>> : never;
+type ExtractSchemaFromDeps<TDeps$1> = TDeps$1 extends {
   schema: infer TSchema extends AnySchema;
 } ? TSchema : AnySchema;
-type AnyLinkedFragments = Record<string, any>;
-interface FragmentResult<TDeps, TServices extends Record<string, unknown>, TServiceThisContext extends RequestThisContext, THandlerThisContext extends RequestThisContext, TRequestStorage, TRoutes extends readonly any[],
-// eslint-disable-line @typescript-eslint/no-explicit-any
-TSchema$1 extends AnySchema, TLinkedFragments extends AnyLinkedFragments = {}> {
-  fragment: FragnoInstantiatedFragment<TRoutes, TDeps, TServices, TServiceThisContext, THandlerThisContext, TRequestStorage, FragnoPublicConfig, TLinkedFragments>;
-  services: TServices;
-  deps: TDeps;
-  callRoute: FragnoInstantiatedFragment<TRoutes, TDeps, TServices, TServiceThisContext, THandlerThisContext, TRequestStorage, FragnoPublicConfig, TLinkedFragments>["callRoute"];
+interface FragmentResult<TDeps$1, TServices$1 extends Record<string, unknown>, TServiceThisContext$1 extends RequestThisContext, THandlerThisContext$1 extends RequestThisContext, TRequestStorage$1, TRoutes$1 extends readonly any[],
+// eslint-disable-line @typescript-eslint/no-explicit-any
+TSchema$1 extends AnySchema> {
+  fragment: FragnoInstantiatedFragment<TRoutes$1, TDeps$1, TServices$1, TServiceThisContext$1, THandlerThisContext$1, TRequestStorage$1, FragnoPublicConfig>;
+  services: TServices$1;
+  deps: TDeps$1;
+  callRoute: FragnoInstantiatedFragment<TRoutes$1, TDeps$1, TServices$1, TServiceThisContext$1, THandlerThisContext$1, TRequestStorage$1, FragnoPublicConfig>["callRoute"];
   db: SimpleQueryInterface<TSchema$1>;
 }
 type AnyFragmentResult = FragmentResult<any,
@@ -33,8 +111,6 @@ any,
 // eslint-disable-line @typescript-eslint/no-explicit-any
 any,
 // eslint-disable-line @typescript-eslint/no-explicit-any
-any,
-// eslint-disable-line @typescript-eslint/no-explicit-any
 any>;
 /**
  * Test context combining base and adapter-specific functionality
@@ -64,6 +140,11 @@ declare class DatabaseFragmentsTestBuilder<TFragments extends Record<string, Any
    * Set the test adapter configuration
    */
   withTestAdapter<TNewAdapter extends SupportedAdapter>(adapter: TNewAdapter): DatabaseFragmentsTestBuilder<TFragments, TNewAdapter, TFirstFragmentThisContext>;
+  /**
+   * Opt out of the default roundtrip guard (enabled by default), or override its configuration.
+   * Useful for allowing multi-roundtrip routes in tests.
+   */
+  withDbRoundtripGuard(guard?: FragnoPublicConfigWithDatabase["dbRoundtripGuard"]): DatabaseFragmentsTestBuilder<TFragments, TAdapter, TFirstFragmentThisContext>;
   /**
    * Add a fragment to the test setup
    *
@@ -71,11 +152,32 @@ declare class DatabaseFragmentsTestBuilder<TFragments extends Record<string, Any
    * @param builder - Pre-configured instantiation builder
    * @param options - Additional options (optional)
    */
-  withFragment<TName extends string, TConfig, TOptions extends FragnoPublicConfig, TDeps, TBaseServices extends Record<string, unknown>, TServices extends Record<string, unknown>, TServiceDependencies, TPrivateServices extends Record<string, unknown>, TServiceThisContext extends RequestThisContext, THandlerThisContext extends RequestThisContext, TRequestStorage, TRoutesOrFactories extends readonly AnyRouteOrFactory[], TLinkedFragments extends AnyLinkedFragments>(name: TName, builder: FragmentInstantiationBuilder<TConfig, TOptions, TDeps, TBaseServices, TServices, TServiceDependencies, TPrivateServices, TServiceThisContext, THandlerThisContext, TRequestStorage, TRoutesOrFactories, TLinkedFragments>, options?: {
+  withFragment<TName extends string, TConfig, TOptions extends FragnoPublicConfig, TDeps$1, TBaseServices$1 extends Record<string, unknown>, TServices$1 extends Record<string, unknown>, TServiceDependencies, TPrivateServices extends Record<string, unknown>, TServiceThisContext$1 extends RequestThisContext, THandlerThisContext$1 extends RequestThisContext, TRequestStorage$1, TRoutesOrFactories$1 extends readonly AnyRouteOrFactory[], TInternalRoutes extends readonly AnyRouteOrFactory[]>(name: TName, builder: FragmentInstantiationBuilder<TConfig, TOptions, TDeps$1, TBaseServices$1, TServices$1, TServiceDependencies, TPrivateServices, TServiceThisContext$1, THandlerThisContext$1, TRequestStorage$1, TRoutesOrFactories$1, TInternalRoutes>, options?: {
+    migrateToVersion?: number;
+  }): DatabaseFragmentsTestBuilder<TFragments & { [K in TName]: FragmentResult<TDeps$1, BoundServices<TBaseServices$1 & TServices$1>, TServiceThisContext$1, THandlerThisContext$1, TRequestStorage$1, FlattenRouteFactories<TRoutesOrFactories$1>, ExtractSchemaFromDeps<TDeps$1>> }, TAdapter, keyof TFragments extends never ? THandlerThisContext$1 : TFirstFragmentThisContext>;
+  /**
+   * Add a pre-built fragment instance to the test setup
+   *
+   * @param name - Unique name for the fragment
+   * @param fragment - Already-built fragment instance
+   */
+  withFragmentInstance<TName extends string, TRoutes$1 extends readonly any[],
+  // eslint-disable-line @typescript-eslint/no-explicit-any
+  TDeps$1, TServices$1 extends Record<string, unknown>, TServiceThisContext$1 extends RequestThisContext, THandlerThisContext$1 extends RequestThisContext, TRequestStorage$1>(name: TName, fragment: FragnoInstantiatedFragment<TRoutes$1, TDeps$1, TServices$1, TServiceThisContext$1, THandlerThisContext$1, TRequestStorage$1, FragnoPublicConfig>, options?: {
+    migrateToVersion?: number;
+  }): DatabaseFragmentsTestBuilder<TFragments & { [K in TName]: FragmentResult<TDeps$1, BoundServices<TServices$1>, TServiceThisContext$1, THandlerThisContext$1, TRequestStorage$1, TRoutes$1, ExtractSchemaFromDeps<TDeps$1>> }, TAdapter, keyof TFragments extends never ? THandlerThisContext$1 : TFirstFragmentThisContext>;
+  /**
+   * Add a fragment factory to the test setup.
+   * The factory runs after the adapter is created.
+   *
+   * @param name - Unique name for the fragment
+   * @param definition - Fragment definition (used to extract schema/namespace)
+   * @param factory - Factory that returns a builder or a pre-built fragment
+   */
+  withFragmentFactory<TName extends string, TFactoryResult extends FragmentFactoryResult>(name: TName, definition: FragmentDefinition<any, any, any, any, any, any, any, any, any, any, any>, factory: (context: FragmentFactoryContext) => TFactoryResult, options?: {
     migrateToVersion?: number;
-  }): DatabaseFragmentsTestBuilder<TFragments & { [K in TName]: FragmentResult<TDeps, BoundServices<TBaseServices & TServices>, TServiceThisContext, THandlerThisContext, TRequestStorage, FlattenRouteFactories<TRoutesOrFactories>, ExtractSchemaFromDeps<TDeps>,
-  // Extract actual schema type from deps
-  TLinkedFragments> }, TAdapter, keyof TFragments extends never ? THandlerThisContext : TFirstFragmentThisContext>;
+    config?: any;
+  }): DatabaseFragmentsTestBuilder<TFragments & { [K in TName]: FragmentResultFromFactoryResult<TFactoryResult> }, TAdapter, keyof TFragments extends never ? HandlerThisContextFromFactoryResult<TFactoryResult> : TFirstFragmentThisContext>;
   /**
    * Build the test setup and return fragments and test context
    */
@@ -125,5 +227,5 @@ declare class DatabaseFragmentsTestBuilder<TFragments extends Record<string, Any
  */
 declare function buildDatabaseFragmentsTest(): DatabaseFragmentsTestBuilder<{}, undefined, RequestThisContext>;
 //#endregion
-export { DatabaseFragmentsTestBuilder, buildDatabaseFragmentsTest };
+export { AnyFragmentResult, DatabaseFragmentsTestBuilder, buildDatabaseFragmentsTest };
 //# sourceMappingURL=db-test.d.ts.map

package/dist/db-test.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"db-test.d.ts","names":[],"sources":["../src/db-test.ts"],"sourcesContent":[],"mappings":";;;;;;;;;KAqBK,iCACS,IAAI,EAAE,kEACJ,MAAM,IAChB,EAAE,WAAW,0BACX,cAAc,EAAE,MAChB,EAAE,IAT+B;KAcpC,qBATS,CAAA,KAAA,CAAA,GASsB,KATtB,SAAA;EAAI,MAAA,EAAA,KAAA,iBASgE,SAThE;CAAE,GAUhB,OAVgB,GAWhB,SAXgB;KAcf,kBAAA,GAAqB,MAbV,CAAA,MAAA,EAAA,GAAA,CAAA;UAgBN,cAhBY,CAAA,KAAA,EAAA,kBAkBF,MAlBE,CAAA,MAAA,EAAA,OAAA,CAAA,EAAA,4BAmBQ,kBAnBR,EAAA,4BAoBQ,kBApBR,EAAA,eAAA,EAAA,gBAAA,SAAA,GAAA,EAAA;AAAA;kBAuBJ,SAtBZ,EAAA,yBAuBqB,kBAvBrB,GAAA,CAAA,CAAA,CAAA,CAAA;EAAE,QAAA,EAyBI,0BAzBJ,CA0BJ,OA1BI,EA2BJ,KA3BI,EA4BJ,SA5BI,EA6BJ,mBA7BI,EA8BJ,mBA9BI,EA+BJ,eA/BI,EAgCJ,kBAhCI,EAiCJ,gBAjCI,CAAA;EAAW,QAAA,EAmCP,SAnCO;EACG,IAAA,EAmCd,KAnCc;EAAE,SAAA,EAoCX,0BApCW,CAqCpB,OArCoB,EAsCpB,KAtCoB,EAuCpB,SAvCoB,EAwCpB,mBAxCoB,EAyCpB,mBAzCoB,EA0CpB,eA1CoB,EA2CpB,kBA3CoB,EA4CpB,gBA5CoB,CAAA,CAAA,WAAA,CAAA;EAAhB,EAAA,EA8CF,oBA9CE,CA8CmB,SA9CnB,CAAA;;KAkDH,iBAAA,GAAoB,cAjDf,CAAA,GAAA;AAAA;GAAC;AAAA;AAAA,GAAA;AAKN;GAA+B;AAAA;GAA8C;AAAA;GAC9E;AAAA;GACA;AAAA;GAAS,CAAA;AAAA;AAGmB;;KAoH3B,WA9GyB,CAAA,UA+GlB,gBA/GkB,EAAA,kCAgHM,kBAhHN,GAgH2B,kBAhH3B,CAAA,GAiH1B,eAjH0B,GAkH5B,cAlH4B,CAkHb,CAlHa,CAAA,GAAA;EACA,OAAA,EAmHjB,eAnHiB,CAAA,GAAA,CAAA;EAGZ;;;;EAMd,SAAA,CAAA,OAAA,CAAA,CAAA,QAAA,EAAA,CAAA,IAAA,EA+GoC,yBA/GpC,EAAA,GA+GkE,OA/GlE,CAAA,EA+G4E,OA/G5E;EACA,SAAA,CAAA,OAAA,CAAA,CAAA,QAAA,EAAA,CAAA,IAAA,EAgHmB,yBAhHnB,EAAA,GAgHiD,OAhHjD,CAgHyD,OAhHzD,CAAA,CAAA,EAiHG,OAjHH,CAiHW,OAjHX,CAAA;CACA;;;;UAsHM,2BA3HE,CAAA,mBA4HS,MA5HT,CAAA,MAAA,EA4HwB,iBA5HxB,CAAA,EAAA,iBA6HO,gBA7HP,EAAA,kCA8HwB,kBA9HxB,GA8H6C,kBA9H7C,CAAA,CAAA;EAUA,SAAA,EAsHC,UAtHD;EACJ,IAAA,EAsHA,WAtHA,CAsHY,QAtHZ,EAsHsB,yBAtHtB,CAAA;;;;;AAMJ,cA2HS,4BA3HT,CAAA,mBA4HiB,MA5HjB,CAAA,MAAA,EA4HgC,iBA5HhC,CAAA,EAAA,iBA6He,gBA7Hf,GAAA,SAAA,GAAA,SAAA,EAAA,kCA8HgC,kBA9HhC,GA8HqD,kBA9HrD,CAAA,CAAA;EACA,CAAA,OAAA;EACA;;;EAGuB,eAAA,CAAA,oBAiIW,gBAjIX,CAAA,CAAA,OAAA,EAkId,WAlIc,CAAA,EAmItB,4BAnIsB,CAmIO,UAnIP,EAmImB,WAnInB,EAmIgC,yBAnIhC,CAAA;EAArB;;AAAoB;AAIa;;;;EAgFnC,YAAA,CAAA,cAAA,MAAA,EAAA,OAAA,EAAA,iBA8DiB,kBA9DjB,EAAA,KAAA,EAAA,sBAgEsB,MAhEtB,CAAA,MAAA,EAAA,OAAA,CAAA,EAAA,kBAiEkB,MAjElB,CAAA,MAAA,EAAA,OAAA,CAAA,EAAA,oBAAA,EAAA,yBAmEyB,MAnEzB,CAAA,MAAA,EAAA,OAAA,CAAA,EAAA,4BAoE4B,kBApE5B,EAAA,4BAqE4B,kBArE5B,EAAA,eAAA,EAAA,2BAAA,SAuEoC,iBAvEpC,EAAA,EAAA,yBAwEyB,kBAxEzB,CAAA,CAAA,IAAA,EA0EM,KA1EN,EAAA,OAAA,EA2ES,4BA3ET,CA4EE,OA5EF,EA6EE,QA7EF,EA8EE,KA9EF,EA+EE,aA/EF,EAgFE,SAhFF,EAiFE,oBAjFF,EAkFE,gBAlFF,EAmFE,mBAnFF,EAoFE,mBApFF,EAqFE,eArFF,EAsFE,kBAtFF,EAuFE,gBAvFF,CAAA,EAAA,OACF,CADE,EAAA;IACa,gBAAA,CAAA,EAAA,MAAA;EAAf,CAAA,CAAA,EA2FG,4BA3FH,CA4FE,UA5FF,GAAA,QA6FU,KA3FC,GA2FO,cA3FP,CA4FL,KA5FK,EA6FL,aA7FK,CA6FS,aA7FT,GA6FyB,SA7FzB,CAAA,EA8FL,mBA9FK,EA+FL,mBA/FK,EAgGL,eAhGK,EAiGL,qBAjGK,CAiGiB,kBAjGjB,CAAA,EAkGL,qBAlGK,CAkGiB,KAlGjB,CAAA;EAAA;EAmGL,gBA9FgC,CAAA,EAA8B,EAiGlE,QAjGkE,EAAA,MAmG5D,UAnG4D,SAAA,KAAA,GAmGjC,mBAnGiC,GAmGX,yBAnGW,CAAA;EAAU;;;EAE3B,KAAA,CAAA,CAAA,EA8GpC,OA9GoC,CA+GjD,QA/GiD,SA+GhC,gBA/GgC,GAgH7C,2BAhH6C,CAgHjB,UAhHiB,EAgHL,QAhHK,EAgHK,yBAhHL,CAAA,GAAA,KAAA,CAAA;;;;AACvC;;;;;;;;;;;AAuBd;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAkDM,iBAgVU,0BAAA,CAAA,CAhVV,EAgVwC,4BAhVxC,CAAA,CAAA,CAAA,EAAA,SAAA,EAmVJ,kBAnVI,CAAA"}
+{"version":3,"file":"db-test.d.ts","names":[],"sources":["../src/db-test.ts"],"sourcesContent":[],"mappings":";;;;;;;;;KAyBK,iCACS,IAAI,EAAE,kEACJ,MAAM,IAChB,EAAE,WAAW,0BACX,cAAc,EAAE,MAChB,EAAE,IAVU;KAaf,sBAAA,GAPS;EAAI,OAAA,EASP,eATO,CAAA,GAAA,CAAA;EAAE,IAAA,EAUZ,eAVY,GAUM,cAVN,CAUqB,gBAVrB,CAAA;CACJ;KAuBX,qBAAA,GACD,4BAxBkB,CAAA,GAAA;AAAA;GAChB;AAAA;GAAE;AAAA;GAAW;AAAA;GACG;AAAA;GAAE;AAAA;GAAhB;AAAA;GACA;AAAA;GAAE;AAAA;GAAC;AAAA;AAAA,GAAA;AAGN;GAEM,CAAA,GA8BP,0BA9BO,CAAA,GAAA;AAAA;GACH;AAAA;GAAiC;AAAA;GAAf;AAAA;GAAc;AAAA;AAAA,GAAA;AAcnC;AAeyB,GAAA,CAUzB;KAAA,mCACH,CAAA,CAAA,CAAA,GAAA,CAAA,SAAU,4BAAV,CAAA,GAAA;AAAA;GAAU;AAAA;GAcN;AAAA;GACA;AAAA;GAAU;AAAA;GASR;AAAA;GACA;AAAA;GAAkB;AAAA;AAAA,KAErB,oBAAA,EAAA,GAAA;AAAA;GACH;AAAA;GAAU,CAAA,GAdN,mBAcM,GAbN,CAaM,SAbI,0BAaJ,CAAA,GAAA;AAAA;GAeJ;AAAA;GACc;AAAA;GAAgB;AAAA;KAA9B,oBAAA,EAAA,GAAA;AAAA;GACA,CAAA,GArBA,mBAqBA,GApBA,kBAoBA;KAlBH,+BAmBG,CAAA,CAAA,CAAA,GAlBN,CAkBM,SAlBI,4BAkBJ,CAAA,GAAA;AAAA;GACA;AAAA;KACsB,MAAA,EAAA,KAAA,cAAA,EAAA,KAAA,UAAA,EAAA,GAAA;AAAA;GAAtB;AAAA;KACsB,oBAAA,EAAA,KAAA,oBAAA,EAAA,KAAA,gBAAA,EAAA,KAAA,mBAAA,EAAA,GAAA,CAAA,GAPxB,cAOwB,CANtB,KAMsB,EALtB,aAKsB,CALR,aAKQ,GALQ,SAKR,CAAA,EAJtB,mBAIsB,EAHtB,mBAGsB,EAFtB,eAEsB,EADtB,qBACsB,CADA,kBACA,CAAA,EAAtB,qBAAsB,CAAA,KAAA,CAAA,CAAA,GAExB,CAFwB,SAEd,0BAFc,CAAA,KAAA,QAAA,EAAA,KAAA,MAAA,EAAA,KAAA,UAAA,EAAA,KAAA,oBAAA,EAAA,KAAA,oBAAA,EAAA,KAAA,gBAAA,EAAA,GAAA,CAAA,GAWtB,cAXsB,CAYpB,KAZoB,EAapB,aAboB,CAaN,SAbM,CAAA,EAcpB,mBAdoB,EAepB,mBAfoB,EAgBpB,eAhBoB,EAiBpB,OAjBoB,EAkBpB,qBAlBoB,CAkBE,KAlBF,CAAA,CAAA,GAAA,KAAA;KAwBzB,qBAxBG,CAAA,OAAA,CAAA,GAwB4B,OAxB5B,SAAA;EAPF,MAAA,EAAA,KAAA,iBA+B4E,SA/B5E;CASA,GAuBF,OAvBE,GAwBF,SAxBE;UA2BI,cA3BM,CAAA,OAAA,EAAA,oBA6BI,MA7BJ,CAAA,MAAA,EAAA,OAAA,CAAA,EAAA,8BA8Bc,kBA9Bd,EAAA,8BA+Bc,kBA/Bd,EAAA,iBAAA,EAAA,kBAAA,SAAA,GAAA,EAAA;AAAA;kBAkCE,SAxBR,CAAA,CAAA;EACc,QAAA,EAyBZ,0BAzBY,CA0BpB,SA1BoB,EA2BpB,OA3BoB,EA4BpB,WA5BoB,EA6BpB,qBA7BoB,EA8BpB,qBA9BoB,EA+BpB,iBA/BoB,EAgCpB,kBAhCoB,CAAA;EAAd,QAAA,EAkCE,WAlCF;EACA,IAAA,EAkCF,OAlCE;EACA,SAAA,EAkCG,0BAlCH,CAmCN,SAnCM,EAoCN,OApCM,EAqCN,WArCM,EAsCN,qBAtCM,EAuCN,qBAvCM,EAwCN,iBAxCM,EAyCN,kBAzCM,CAAA,CAAA,WAAA,CAAA;EACA,EAAA,EA0CJ,oBA1CI,CA0CiB,SA1CjB,CAAA;;AAEsB,KA4CpB,iBAAA,GAAoB,cA5CA,CAAA,GAAA;AAAA;GAAtB;AAAA;GAPF;AAAA;GAAc;AAAA;AAAA,GAAA;AAajB;GAA+B;AAAA;GAA8C,CAAA;;;;AAErE,KAsJR,WAnJK,CAAA,UAoJE,gBApJY,EAAA,kCAqJY,kBArJZ,GAqJiC,kBArJjC,CAAA,GAsJpB,eAtJoB,GAuJtB,cAvJsB,CAuJP,CAvJO,CAAA,GAAA;EAEJ,OAAA,EAuJP,eAvJO,CAAA,GAAA,CAAA;EACU;;;;EAQ1B,SAAA,CAAA,OAAA,CAAA,CAAA,QAAA,EAAA,CAAA,IAAA,EAmJoC,yBAnJpC,EAAA,GAmJkE,OAnJlE,CAAA,EAmJ4E,OAnJ5E;EACA,SAAA,CAAA,OAAA,CAAA,CAAA,QAAA,EAAA,CAAA,IAAA,EAoJmB,yBApJnB,EAAA,GAoJiD,OApJjD,CAoJyD,OApJzD,CAAA,CAAA,EAqJG,OArJH,CAqJW,OArJX,CAAA;CACA;;;;UA0JM,2BA9JE,CAAA,mBA+JS,MA/JT,CAAA,MAAA,EA+JwB,iBA/JxB,CAAA,EAAA,iBAgKO,gBAhKP,EAAA,kCAiKwB,kBAjKxB,GAiK6C,kBAjK7C,CAAA,CAAA;EASA,SAAA,EA0JC,UA1JD;EACJ,IAAA,EA0JA,WA1JA,CA0JY,QA1JZ,EA0JsB,yBA1JtB,CAAA;;;;;AAMJ,cA+KS,4BA/KT,CAAA,mBAgLiB,MAhLjB,CAAA,MAAA,EAgLgC,iBAhLhC,CAAA,EAAA,iBAiLe,gBAjLf,GAAA,SAAA,GAAA,SAAA,EAAA,kCAkLgC,kBAlLhC,GAkLqD,kBAlLrD,CAAA,CAAA;EACA,CAAA,OAAA;EACA;;;EAEE,eAAA,CAAA,oBAuLgC,gBAvLhC,CAAA,CAAA,OAAA,EAwLO,WAxLP,CAAA,EAyLD,4BAzLC,CAyL4B,UAzL5B,EAyLwC,WAzLxC,EAyLqD,yBAzLrD,CAAA;EAAoB;AAI1B;AAQE;;EA4GkC,oBAAA,CAAA,KAAA,CAAA,EA2EzB,8BA3EyB,CAAA,kBAAA,CAAA,CAAA,EA4E/B,4BA5E+B,CA4EF,UA5EE,EA4EU,QA5EV,EA4EoB,yBA5EpB,CAAA;EAAqB;;;;;;;EASuB,YAAA,CAAA,cAAA,MAAA,EAAA,OAAA,EAAA,iBAkF3D,kBAlF2D,EAAA,OAAA,EAAA,wBAoFtD,MApFsD,CAAA,MAAA,EAAA,OAAA,CAAA,EAAA,oBAqF1D,MArF0D,CAAA,MAAA,EAAA,OAAA,CAAA,EAAA,oBAAA,EAAA,yBAuFnD,MAvFmD,CAAA,MAAA,EAAA,OAAA,CAAA,EAAA,8BAwFhD,kBAxFgD,EAAA,8BAyFhD,kBAzFgD,EAAA,iBAAA,EAAA,6BAAA,SA2FxC,iBA3FwC,EAAA,EAAA,wBAAA,SA4F3C,iBA5F2C,EAAA,CAAA,CAAA,IAAA,EA8FtE,KA9FsE,EAAA,OAAA,EA+FnE,4BA/FmE,CAgG1E,OAhG0E,EAiG1E,QAjG0E,EAkG1E,OAlG0E,EAmG1E,eAnG0E,EAoG1E,WApG0E,EAqG1E,oBArG0E,EAsG1E,gBAtG0E,EAuG1E,qBAvG0E,EAwG1E,qBAxG0E,EAyG1E,iBAzG0E,EA0G1E,oBA1G0E,EA2G1E,eA3G0E,CAAA,EAAA,OAEnB,CAFmB,EAAA;IAEzD,gBAAA,CAAA,EAAA,MAAA;EAAsC,CAAA,CAAA,EA8GxD,4BA9GwD,CA+GzD,UA/GyD,GAAA,QAgHjD,KAhHyC,GAgHjC,cAhHiC,CAiH7C,OAjH6C,EAkH7C,aAlH6C,CAkH/B,eAlH+B,GAkHf,WAlHe,CAAA,EAmH7C,qBAnH6C,EAoH7C,qBApH6C,EAqH7C,iBArH6C,EAsH7C,qBAtH6C,CAsHvB,oBAtHuB,CAAA,EAuH7C,qBAvH6C,CAuHvB,OAvHuB,CAAA,CAAA,EACtC,EAyHX,QAzHW,EAAA,MA2HL,UA3HK,SAAA,KAAA,GA2HsB,qBA3HtB,GA2H4C,yBA3H5C,CAAA;EAAR;;AAAO;;;;EASsB,oBAAA,CAAA,cAAA,MAAA,EAAA,kBAAA,SAAA,GAAA,EAAA;EAAA;EAAqB,OAAA,EAAA,oBAuInC,MAvImC,CAAA,MAAA,EAAA,OAAA,CAAA,EAAA,8BAwIzB,kBAxIyB,EAAA,8BAyIzB,kBAzIyB,EAAA,iBAAA,CAAA,CAAA,IAAA,EA4I/C,KA5I+C,EAAA,QAAA,EA6I3C,0BA7I2C,CA8InD,SA9ImD,EA+InD,OA/ImD,EAgJnD,WAhJmD,EAiJnD,qBAjJmD,EAkJnD,qBAlJmD,EAmJnD,iBAnJmD,EAoJnD,kBApJmD,CAAA,EAAA,OAGrC,CAHqC,EAAA;IAE5C,gBAAA,CAAA,EAAA,MAAA;EACO,CAAA,CAAA,EAoJf,4BApJe,CAqJhB,UArJgB,GAAA,QAsJR,KAtJkB,GAsJV,cAtJU,CAuJtB,OAvJsB,EAwJtB,aAxJsB,CAwJR,WAxJQ,CAAA,EAyJtB,qBAzJsB,EA0JtB,qBA1JsB,EA2JtB,iBA3JsB,EA4JtB,SA5JsB,EA6JtB,qBA7JsB,CA6JA,OA7JA,CAAA,CAAA,EAAtB,EAgKJ,QAhKI,EAAA,MAiKE,UAjKF,SAAA,KAAA,GAiK6B,qBAjK7B,GAiKmD,yBAjKnD,CAAA;EAAW;AA2BnB;;;;;;;EAaa,mBAAA,CAAA,cAAA,MAAA,EAAA,uBA4IsD,qBA5ItD,CAAA,CAAA,IAAA,EA6IH,KA7IG,EAAA,UAAA,EA+IG,kBA/IH,CAAA,GAAA,EAAA,GAAA,EAAA,GAAA,EAAA,GAAA,EAAA,GAAA,EAAA,GAAA,EAAA,GAAA,EAAA,GAAA,EAAA,GAAA,EAAA,GAAA,EAAA,GAAA,CAAA,EAAA,OAAA,EAAA,CAAA,OAAA,EAgJU,sBAhJV,EAAA,GAgJqC,cAhJrC,EAAA,OAC8C,CAD9C,EAAA;IACqB,gBAAA,CAAA,EAAA,MAAA;IAAY,MAAA,CAAA,EAAA,GAAA;EAAa,CAAA,CAAA,EAqJtD,4BArJsD,CAsJvD,UAtJuD,GAAA,QAuJ/C,KAvJP,GAuJe,+BAvJf,CAuJ+C,cAvJ/C,CAAA,EAUM,EA+IP,QA/IO,EAAA,MAgJD,UAhJC,SAAA,KAAA,GAiJH,mCAjJG,CAiJiC,cAjJjC,CAAA,GAkJH,yBAlJG,CAAA;EACuB;;;EAA7B,KAAA,CAAA,CAAA,EAiKY,OAjKZ,CAkKD,QAlKC,SAkKgB,gBAlKhB,GAmKG,2BAnKH,CAmK+B,UAnK/B,EAmK2C,QAnK3C,EAmKqD,yBAnKrD,CAAA,GAAA,KAAA,CAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAkF2B,iBAslBhB,0BAAA,CAAA,CAtlBgB,EAslBc,4BAtlBd,CAAA,CAAA,CAAA,EAAA,SAAA,EAylB9B,kBAzlB8B,CAAA"}

package/dist/db-test.js

@@ -1,12 +1,24 @@
 import { createAdapter } from "./adapters.js";
+import { drainDurableHooks } from "./durable-hooks.js";
 
 //#region src/db-test.ts
+const disableAutoSchedule = (options) => {
+	const durableHooks = options.durableHooks ?? {};
+	return {
+		...options,
+		durableHooks: {
+			...durableHooks,
+			autoSchedule: false
+		}
+	};
+};
 /**
 * Builder for creating multiple database fragments for testing
 */
 var DatabaseFragmentsTestBuilder = class {
 	#adapter;
 	#fragments = /* @__PURE__ */ new Map();
+	#dbRoundtripGuard = true;
 	/**
 	* Set the test adapter configuration
 	*/
@@ -15,6 +27,14 @@ var DatabaseFragmentsTestBuilder = class {
 		return this;
 	}
 	/**
+	* Opt out of the default roundtrip guard (enabled by default), or override its configuration.
+	* Useful for allowing multi-roundtrip routes in tests.
+	*/
+	withDbRoundtripGuard(guard = false) {
+		this.#dbRoundtripGuard = guard;
+		return this;
+	}
+	/**
 	* Add a fragment to the test setup
 	*
 	* @param name - Unique name for the fragment
@@ -23,6 +43,7 @@ var DatabaseFragmentsTestBuilder = class {
 	*/
 	withFragment(name, builder, options) {
 		this.#fragments.set(name, {
+			kind: "builder",
 			definition: builder.definition,
 			builder,
 			migrateToVersion: options?.migrateToVersion
@@ -30,30 +51,73 @@ var DatabaseFragmentsTestBuilder = class {
 		return this;
 	}
 	/**
+	* Add a pre-built fragment instance to the test setup
+	*
+	* @param name - Unique name for the fragment
+	* @param fragment - Already-built fragment instance
+	*/
+	withFragmentInstance(name, fragment, options) {
+		this.#fragments.set(name, {
+			kind: "instance",
+			fragment,
+			migrateToVersion: options?.migrateToVersion
+		});
+		return this;
+	}
+	/**
+	* Add a fragment factory to the test setup.
+	* The factory runs after the adapter is created.
+	*
+	* @param name - Unique name for the fragment
+	* @param definition - Fragment definition (used to extract schema/namespace)
+	* @param factory - Factory that returns a builder or a pre-built fragment
+	*/
+	withFragmentFactory(name, definition, factory, options) {
+		this.#fragments.set(name, {
+			kind: "factory",
+			definition,
+			factory,
+			config: options?.config,
+			migrateToVersion: options?.migrateToVersion
+		});
+		return this;
+	}
+	/**
 	* Build the test setup and return fragments and test context
 	*/
 	async build() {
 		if (!this.#adapter) throw new Error("Test adapter must be set using withTestAdapter()");
-		if (this.#fragments.size === 0) throw new Error("At least one fragment must be added using withFragment()");
+		if (this.#fragments.size === 0) throw new Error("At least one fragment must be added using withFragment(), withFragmentFactory(), or withFragmentInstance().");
 		const adapterConfig = this.#adapter;
-		const fragmentNames = Array.from(this.#fragments.keys());
-		const fragmentConfigs = Array.from(this.#fragments.values());
+		const fragmentEntries = Array.from(this.#fragments.entries());
+		const fragmentNames = fragmentEntries.map(([name]) => name);
 		const schemaConfigs = [];
-		const builderConfigs = [];
-		for (const fragmentConfig of fragmentConfigs) {
-			const builder = fragmentConfig.builder;
-			const definition = builder.definition;
+		const fragmentPlans = [];
+		const extractSchemaFromDefinition = (definition, actualConfig, actualOptions) => {
 			let schema;
 			let namespace;
 			if (definition.dependencies) try {
 				const mockAdapter = {
 					createQueryEngine: () => ({ schema: null }),
-					contextStorage: { run: (_data, fn) => fn() }
+					getSchemaVersion: async () => void 0,
+					namingStrategy: {
+						namespaceScope: "suffix",
+						namespaceToSchema: (value) => value,
+						tableName: (logicalTable, ns) => ns ? `${logicalTable}_${ns}` : logicalTable,
+						columnName: (logicalColumn) => logicalColumn,
+						indexName: (logicalIndex) => logicalIndex,
+						uniqueIndexName: (logicalIndex) => logicalIndex,
+						foreignKeyName: ({ referenceName }) => referenceName
+					},
+					contextStorage: { run: (_data, fn) => fn() },
+					close: async () => {}
 				};
-				const actualConfig = builder.config ?? {};
 				const deps = definition.dependencies({
-					config: actualConfig,
-					options: { databaseAdapter: mockAdapter }
+					config: actualConfig ?? {},
+					options: {
+						...actualOptions,
+						databaseAdapter: mockAdapter
+					}
 				});
 				if (deps && typeof deps === "object" && "schema" in deps) {
 					schema = deps.schema;
@@ -64,69 +128,165 @@ var DatabaseFragmentsTestBuilder = class {
 				throw new Error(`Failed to extract schema from fragment '${definition.name}'.\nOriginal error: ${errorMessage}\n\nMake sure the fragment is a database fragment using defineFragment().extend(withDatabase(schema)).`);
 			}
 			if (!schema) throw new Error(`Fragment '${definition.name}' does not have a database schema. Make sure you're using defineFragment().extend(withDatabase(schema)).`);
-			if (!namespace) throw new Error(`Fragment '${definition.name}' does not have a namespace in dependencies. This should be automatically provided by withDatabase().`);
-			schemaConfigs.push({
+			if (namespace === void 0) throw new Error(`Fragment '${definition.name}' does not have a namespace in dependencies. This should be automatically provided by withDatabase().`);
+			return {
 				schema,
-				namespace,
+				namespace
+			};
+		};
+		for (const [name, fragmentConfig] of fragmentEntries) {
+			if (fragmentConfig.kind === "builder") {
+				const builder = fragmentConfig.builder;
+				const definition = builder.definition;
+				const { schema, namespace } = extractSchemaFromDefinition(definition, builder.config ?? {}, builder.options ?? {});
+				schemaConfigs.push({
+					schema,
+					namespace,
+					migrateToVersion: fragmentConfig.migrateToVersion
+				});
+				fragmentPlans.push({
+					name,
+					kind: "builder",
+					schema,
+					namespace,
+					migrateToVersion: fragmentConfig.migrateToVersion,
+					builderConfig: {
+						builder: fragmentConfig.builder,
+						definition: fragmentConfig.definition,
+						config: builder.config ?? {},
+						routes: builder.routes ?? [],
+						options: builder.options ?? {}
+					}
+				});
+				continue;
+			}
+			if (fragmentConfig.kind === "factory") {
+				const definition = fragmentConfig.definition;
+				const { schema, namespace } = extractSchemaFromDefinition(definition, fragmentConfig.config ?? {});
+				schemaConfigs.push({
+					schema,
+					namespace,
+					migrateToVersion: fragmentConfig.migrateToVersion
+				});
+				fragmentPlans.push({
+					name,
+					kind: "factory",
+					schema,
+					namespace,
+					migrateToVersion: fragmentConfig.migrateToVersion,
+					factory: fragmentConfig.factory
+				});
+				continue;
+			}
+			const fragment = fragmentConfig.fragment;
+			const deps = fragment.$internal?.deps;
+			if (!deps?.schema) throw new Error(`Fragment '${name}' does not have a database schema in deps. Make sure you're using defineFragment().extend(withDatabase(schema)).`);
+			schemaConfigs.push({
+				schema: deps.schema,
+				namespace: deps.namespace ?? null,
 				migrateToVersion: fragmentConfig.migrateToVersion
 			});
-			builderConfigs.push({
-				config: builder.config ?? {},
-				routes: builder.routes ?? [],
-				options: builder.options ?? {}
+			fragmentPlans.push({
+				name,
+				kind: "instance",
+				schema: deps.schema,
+				namespace: deps.namespace ?? null,
+				migrateToVersion: fragmentConfig.migrateToVersion,
+				fragment
 			});
 		}
 		const { testContext, adapter } = await createAdapter(adapterConfig, schemaConfigs);
+		const resolveDbRoundtripGuardOption = (options) => {
+			if (options && typeof options === "object") {
+				if (Object.prototype.hasOwnProperty.call(options, "dbRoundtripGuard")) return options.dbRoundtripGuard;
+			}
+			return this.#dbRoundtripGuard;
+		};
+		const mergeBuilderOptions = (options) => {
+			const resolvedOptions = disableAutoSchedule(options ?? {});
+			const merged = {
+				...resolvedOptions,
+				databaseAdapter: adapter
+			};
+			const guardOption = resolveDbRoundtripGuardOption(resolvedOptions);
+			if (guardOption !== void 0) merged["dbRoundtripGuard"] = guardOption;
+			return merged;
+		};
 		const createFragments = () => {
+			const resolveBuilderConfig = (builder) => ({
+				builder,
+				definition: builder.definition,
+				config: builder.config ?? {},
+				routes: builder.routes ?? [],
+				options: builder.options ?? {}
+			});
+			const isBuilder = (value) => Boolean(value) && typeof value === "object" && "build" in value && "definition" in value;
 			const providedServicesByName = {};
-			const fragmentResults$1 = [];
-			for (let i = 0; i < fragmentConfigs.length; i++) {
-				const fragmentConfig = fragmentConfigs[i];
-				const builderConfig = builderConfigs[i];
-				const namespace = schemaConfigs[i].namespace;
-				const schema = schemaConfigs[i].schema;
-				const orm = testContext.getOrm(namespace);
-				const mergedOptions = {
-					...builderConfig.options,
-					databaseAdapter: adapter
-				};
-				const fragment = fragmentConfig.builder.withOptions(mergedOptions).build();
+			const instanceResults = /* @__PURE__ */ new Map();
+			const builderConfigs = /* @__PURE__ */ new Map();
+			for (const plan of fragmentPlans) {
+				const orm = testContext.getOrm(plan.namespace);
+				let fragment;
+				let builderConfig = plan.builderConfig;
+				if (plan.kind === "factory") {
+					const result = plan.cachedFactoryResult ?? plan.factory({
+						adapter,
+						test: testContext
+					});
+					if (!plan.cachedFactoryResult) plan.cachedFactoryResult = result;
+					if (isBuilder(result)) builderConfig = resolveBuilderConfig(result);
+					else fragment = result;
+				}
+				const usesBuilder = plan.kind === "builder" || !!builderConfig;
+				if (usesBuilder) {
+					const resolvedBuilderConfig = builderConfig ?? plan.builderConfig;
+					const mergedOptions = mergeBuilderOptions(resolvedBuilderConfig.options);
+					fragment = resolvedBuilderConfig.builder.withOptions(mergedOptions).build();
+					builderConfigs.set(plan.name, resolvedBuilderConfig);
+				} else {
+					fragment = fragment ?? plan.fragment;
+					const deps = fragment.$internal?.deps;
+					if (deps?.databaseAdapter && deps.databaseAdapter !== adapter) throw new Error(`Fragment '${plan.name}' was built with a different database adapter instance. Use withFragment() or ensure the fragment uses the same adapter instance as the test builder.`);
+				}
+				if (!fragment) throw new Error(`Fragment '${plan.name}' did not return a valid fragment instance from its factory.`);
 				for (const [serviceName, serviceImpl] of Object.entries(fragment.services)) providedServicesByName[serviceName] = {
 					service: serviceImpl,
 					orm
 				};
-				const deps = fragment.$internal?.deps;
-				fragmentResults$1.push({
-					fragment,
-					services: fragment.services,
-					deps: deps || {},
-					callRoute: fragment.callRoute.bind(fragment),
-					get db() {
-						return orm;
-					},
-					_orm: orm,
-					_schema: schema
-				});
+				if (!usesBuilder) {
+					const deps = fragment.$internal?.deps;
+					instanceResults.set(plan.name, {
+						fragment,
+						services: fragment.services,
+						deps: deps || {},
+						callRoute: fragment.callRoute.bind(fragment),
+						get db() {
+							return orm;
+						},
+						_orm: orm,
+						_schema: plan.schema
+					});
+				}
 			}
-			for (let i = 0; i < fragmentConfigs.length; i++) {
-				const fragmentConfig = fragmentConfigs[i];
-				const definition = fragmentConfig.builder.definition;
-				const builderConfig = builderConfigs[i];
-				const namespace = schemaConfigs[i].namespace;
-				const schema = schemaConfigs[i].schema;
-				const orm = testContext.getOrm(namespace);
+			const fragmentResults$1 = [];
+			for (const plan of fragmentPlans) {
+				const orm = testContext.getOrm(plan.namespace);
+				if (instanceResults.has(plan.name)) {
+					fragmentResults$1.push(instanceResults.get(plan.name));
+					continue;
+				}
+				const builderConfig = builderConfigs.get(plan.name);
+				if (!builderConfig) throw new Error(`Fragment '${plan.name}' was expected to produce a builder for service wiring.`);
+				const definition = builderConfig.definition;
 				const serviceImplementations = {};
 				const serviceDependencies = definition.serviceDependencies;
 				if (serviceDependencies) {
 					for (const serviceName of Object.keys(serviceDependencies)) if (providedServicesByName[serviceName]) serviceImplementations[serviceName] = providedServicesByName[serviceName].service;
 				}
-				const mergedOptions = {
-					...builderConfig.options,
-					databaseAdapter: adapter
-				};
-				const fragment = fragmentConfig.builder.withOptions(mergedOptions).withServices(serviceImplementations).build();
+				const mergedOptions = mergeBuilderOptions(builderConfig.options);
+				const fragment = builderConfig.builder.withOptions(mergedOptions).withServices(serviceImplementations).build();
 				const deps = fragment.$internal?.deps;
-				fragmentResults$1[i] = {
+				fragmentResults$1.push({
 					fragment,
 					services: fragment.services,
 					deps: deps || {},
@@ -135,8 +295,8 @@ var DatabaseFragmentsTestBuilder = class {
 						return orm;
 					},
 					_orm: orm,
-					_schema: schema
-				};
+					_schema: plan.schema
+				});
 			}
 			return fragmentResults$1;
 		};
@@ -155,9 +315,28 @@ var DatabaseFragmentsTestBuilder = class {
 		};
 		const firstFragment = fragmentResults[0]?.fragment;
 		if (!firstFragment) throw new Error("At least one fragment must be added");
+		const originalCleanup = testContext.cleanup;
+		const cleanup = async () => {
+			let drainError;
+			let cleanupError;
+			for (const result of fragmentResults) try {
+				await drainDurableHooks(result.fragment);
+			} catch (error) {
+				if (!drainError) drainError = error;
+			}
+			try {
+				await originalCleanup();
+			} catch (error) {
+				cleanupError = error;
+			}
+			if (drainError && cleanupError) throw new AggregateError([drainError, cleanupError], "Failed to drain durable hooks and clean up test context");
+			if (drainError) throw drainError;
+			if (cleanupError) throw cleanupError;
+		};
 		const finalTestContext = {
 			...testContext,
 			resetDatabase,
+			cleanup,
 			adapter,
 			inContext: firstFragment.inContext.bind(firstFragment)
 		};