chore(types): boot tool driver types

Agoric · Dec 8, 2023 · 881bf9d · 881bf9d
1 parent 8f806bf
commit 881bf9d
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 5 deletions.
diff --git a/packages/boot/tools/drivers.ts b/packages/boot/tools/drivers.ts
@@ -197,6 +197,8 @@ export const makePriceFeedDriver = async (
     },
   };
 };
+harden(makePriceFeedDriver);
+export type PriceFeedDriver = Awaited<ReturnType<typeof makePriceFeedDriver>>;
 
 export const makeGovernanceDriver = async (
   testKit: SwingsetTestKit,
@@ -318,6 +320,8 @@ export const makeGovernanceDriver = async (
     ecMembers,
   };
 };
+harden(makeGovernanceDriver);
+export type GovernanceDriver = Awaited<ReturnType<typeof makeGovernanceDriver>>;
 
 export const makeZoeDriver = async (testKit: SwingsetTestKit) => {
   const { EV } = testKit.runUtils;
@@ -402,3 +406,5 @@ export const makeZoeDriver = async (testKit: SwingsetTestKit) => {
     },
   };
 };
+harden(makeZoeDriver);
+export type ZoeDriver = Awaited<ReturnType<typeof makeZoeDriver>>;
diff --git a/packages/boot/tools/liquidation.ts b/packages/boot/tools/liquidation.ts
@@ -9,8 +9,11 @@ import {
 } from '@agoric/vats/tools/board-utils.js';
 import { Offers } from '@agoric/inter-protocol/src/clientSupport.js';
 import type { ExecutionContext } from 'ava';
-import { makeSwingsetTestKit } from './supports.ts';
+import { type SwingsetTestKit, makeSwingsetTestKit } from './supports.ts';
 import {
+  type GovernanceDriver,
+  type PriceFeedDriver,
+  type WalletFactoryDriver,
   makeGovernanceDriver,
   makePriceFeedDriver,
   makeWalletFactoryDriver,
@@ -76,10 +79,7 @@ export const makeLiquidationTestKit = async ({
   governanceDriver: GovernanceDriver;
   t: Pick<ExecutionContext, 'like'>;
 }) => {
-  const priceFeedDrivers = {} as Record<
-    string,
-    Awaited<ReturnType<typeof makePriceFeedDriver>>
-  >;
+  const priceFeedDrivers = {} as Record<string, PriceFeedDriver>;
 
   console.timeLog('DefaultTestContext', 'priceFeedDriver');