Fix unit tests

ChainSafe · Sep 14, 2021 · 62279f7 · 62279f7
1 parent 06f3da5
commit 62279f7
Show file tree

Hide file tree

Showing 4 changed files with 9 additions and 4 deletions.
diff --git a/packages/lodestar/test/unit/api/impl/beacon/pool/pool.test.ts b/packages/lodestar/test/unit/api/impl/beacon/pool/pool.test.ts
@@ -20,12 +20,13 @@ import {generateEmptySignedBlockHeader} from "../../../../../utils/block";
 import {setupApiImplTestServer} from "../../index.test";
 import {SinonStubFn} from "../../../../../utils/types";
 import {testLogger} from "../../../../../utils/logger";
-import {AggregatedAttestationPool} from "../../../../../../src/chain/opPools";
+import {AggregatedAttestationPool, OpPool} from "../../../../../../src/chain/opPools";
 
 describe("beacon pool api impl", function () {
   const logger = testLogger();
   let poolApi: ReturnType<typeof getBeaconPoolApi>;
   let chainStub: SinonStubbedInstance<IBeaconChain>;
+  let opPool: SinonStubbedInstance<OpPool>;
   let networkStub: SinonStubbedInstance<Network>;
   let gossipStub: SinonStubbedInstance<Eth2Gossipsub>;
   let validateGossipAttesterSlashing: SinonStubFn<typeof attesterSlashingValidation["validateGossipAttesterSlashing"]>;
@@ -40,6 +41,10 @@ describe("beacon pool api impl", function () {
     ((chainStub as unknown) as {
       aggregatedAttestationPool: SinonStubbedInstance<AggregatedAttestationPool>;
     }).aggregatedAttestationPool = aggregatedAttestationPool;
+    opPool = sinon.createStubInstance(OpPool);
+    ((chainStub as unknown) as {
+      opPool: SinonStubbedInstance<OpPool>;
+    }).opPool = opPool;
     gossipStub = sinon.createStubInstance(Eth2Gossipsub);
     gossipStub.publishAttesterSlashing = sinon.stub();
     gossipStub.publishProposerSlashing = sinon.stub();

diff --git a/packages/lodestar/test/unit/chain/validation/attesterSlashing.test.ts b/packages/lodestar/test/unit/chain/validation/attesterSlashing.test.ts
@@ -18,7 +18,7 @@ describe("GossipMessageValidator", () => {
   let chainStub: StubbedChain;
   let opPool: OpPool & SinonStubbedInstance<OpPool>;
 
-  before(() => {
+  beforeEach(() => {
     chainStub = sandbox.createStubInstance(BeaconChain) as StubbedChain;
     chainStub.forkChoice = sandbox.createStubInstance(ForkChoice);
     chainStub.bls = {verifySignatureSets: async () => true};

diff --git a/packages/lodestar/test/unit/chain/validation/proposerSlashing.test.ts b/packages/lodestar/test/unit/chain/validation/proposerSlashing.test.ts
@@ -17,7 +17,7 @@ describe("validate proposer slashing", () => {
   let chainStub: StubbedChain;
   let opPool: OpPool & SinonStubbedInstance<OpPool>;
 
-  before(() => {
+  beforeEach(() => {
     chainStub = sandbox.createStubInstance(BeaconChain) as StubbedChain;
     chainStub.forkChoice = sandbox.createStubInstance(ForkChoice);
     chainStub.bls = {verifySignatureSets: async () => true};

diff --git a/packages/lodestar/test/unit/chain/validation/voluntaryExit.test.ts b/packages/lodestar/test/unit/chain/validation/voluntaryExit.test.ts
@@ -87,7 +87,7 @@ describe("validate voluntary exit", () => {
     };
 
     // Return SignedVoluntaryExit known
-    opPool.hasSeenProposerSlashing.returns(true);
+    opPool.hasSeenVoluntaryExit.returns(true);
 
     await expectRejectedWithLodestarError(
       validateGossipVoluntaryExit(chainStub, signedVoluntaryExitInvalidSig),