chore: use new functions

Signed-off-by: james-a-morris <jaamorris@cs.stonybrook.edu>
across-protocol · Sep 4, 2024 · aa2d926 · aa2d926
1 parent 0ff3828
commit aa2d926
Showing 1 changed file with 12 additions and 8 deletions.
diff --git a/test/SpokePoolClient.ValidateFill.ts b/test/SpokePoolClient.ValidateFill.ts
@@ -4,9 +4,9 @@ import {
   bnOne,
   InvalidFill,
   fillStatusArray,
-  relayFillStatus,
   validateFillForDeposit,
   queryHistoricalDepositForFill,
+  DepositSearchResult,
 } from "../src/utils";
 import { CHAIN_ID_TEST_LIST, originChainId, destinationChainId, repaymentChainId } from "./constants";
 import {
@@ -126,11 +126,11 @@ describe("SpokePoolClient: Fill Validation", function () {
       outputAmount
     );
 
-    let filled = await relayFillStatus(spokePool_2, deposit);
+    let filled = await spokePoolClient2.relayFillStatus(deposit);
     expect(filled).to.equal(FillStatus.Unfilled);
 
     await fillV3Relay(spokePool_2, deposit, relayer);
-    filled = await relayFillStatus(spokePool_2, deposit);
+    filled = await spokePoolClient2.relayFillStatus(deposit);
     expect(filled).to.equal(FillStatus.Filled);
   });
 
@@ -439,7 +439,9 @@ describe("SpokePoolClient: Fill Validation", function () {
 
     const historicalDeposit = await queryHistoricalDepositForFill(spokePoolClient1, fill);
     assert.equal(historicalDeposit.found, true, "Test is broken"); // Help tsc to narrow the discriminated union.
-    expect(historicalDeposit.deposit.depositId).to.deep.equal(deposit.depositId);
+    expect((historicalDeposit as Extract<DepositSearchResult, { found: true }>).deposit.depositId).to.deep.equal(
+      deposit.depositId
+    );
   });
 
   it("Can fetch younger deposit matching fill", async function () {
@@ -473,7 +475,9 @@ describe("SpokePoolClient: Fill Validation", function () {
 
     const historicalDeposit = await queryHistoricalDepositForFill(spokePoolClient1, fill);
     assert.equal(historicalDeposit.found, true, "Test is broken"); // Help tsc to narrow the discriminated union.
-    expect(historicalDeposit.deposit.depositId).to.deep.equal(deposit.depositId);
+    expect((historicalDeposit as Extract<DepositSearchResult, { found: true }>).deposit.depositId).to.deep.equal(
+      deposit.depositId
+    );
   });
 
   it("Loads fills from memory with deposit ID > spoke pool client's earliest deposit ID queried", async function () {
@@ -547,7 +551,7 @@ describe("SpokePoolClient: Fill Validation", function () {
     const search = await queryHistoricalDepositForFill(spokePoolClient1, fill);
 
     assert.equal(search.found, false, "Test is broken"); // Help tsc to narrow the discriminated union.
-    expect(search.code).to.equal(InvalidFill.DepositIdInvalid);
+    expect((search as Extract<DepositSearchResult, { found: false }>).code).to.equal(InvalidFill.DepositIdInvalid);
     expect(lastSpyLogIncludes(spy, "Queried RPC for deposit")).is.not.true;
   });
 
@@ -572,7 +576,7 @@ describe("SpokePoolClient: Fill Validation", function () {
     const search = await queryHistoricalDepositForFill(spokePoolClient1, fill);
 
     assert.equal(search.found, false, "Test is broken"); // Help tsc to narrow the discriminated union.
-    expect(search.code).to.equal(InvalidFill.DepositIdInvalid);
+    expect((search as Extract<DepositSearchResult, { found: false }>).code).to.equal(InvalidFill.DepositIdInvalid);
     expect(lastSpyLogIncludes(spy, "Queried RPC for deposit")).is.not.true;
   });
 
@@ -594,7 +598,7 @@ describe("SpokePoolClient: Fill Validation", function () {
 
     const search = await queryHistoricalDepositForFill(spokePoolClient1, fill);
     assert.equal(search.found, false, "Test is broken"); // Help tsc to narrow the discriminated union.
-    expect(search.code).to.equal(InvalidFill.FillMismatch);
+    expect((search as Extract<DepositSearchResult, { found: false }>).code).to.equal(InvalidFill.FillMismatch);
   });
 
   it("Returns sped up deposit matched with fill", async function () {