fix tests

ChainSafe · Jan 9, 2024 · a289561 · a289561
1 parent 9782325
commit a289561
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/packages/beacon-node/test/unit/api/impl/validator/produceBlockV2.test.ts b/packages/beacon-node/test/unit/api/impl/validator/produceBlockV2.test.ts
@@ -85,7 +85,12 @@ describe("api/validator - produceBlockV2", function () {
     const feeRecipient = "0xcccccccccccccccccccccccccccccccccccccccc";
 
     const api = getValidatorApi(modules);
-    server.chainStub.produceBlock.mockResolvedValue({block: fullBlock, executionPayloadValue, consensusBlockValue});
+    server.chainStub.produceBlock.mockResolvedValue({
+      block: fullBlock,
+      executionPayloadValue,
+      consensusBlockValue,
+      shouldOverrideBuilder: false,
+    });
 
     // check if expectedFeeRecipient is passed to produceBlock
     await api.produceBlockV2(slot, randaoReveal, graffiti, {feeRecipient});
@@ -129,6 +134,7 @@ describe("api/validator - produceBlockV2", function () {
     executionEngineStub.getPayload.mockResolvedValue({
       executionPayload: ssz.bellatrix.ExecutionPayload.defaultValue(),
       executionPayloadValue,
+      shouldOverrideBuilder: false,
     });
 
     // use fee recipient passed in produceBlockBody call for payload gen in engine notifyForkchoiceUpdate

diff --git a/packages/beacon-node/test/unit/api/impl/validator/produceBlockV3.test.ts b/packages/beacon-node/test/unit/api/impl/validator/produceBlockV3.test.ts
@@ -89,6 +89,7 @@ describe("api/validator - produceBlockV3", function () {
             block: fullBlock,
             executionPayloadValue: BigInt(enginePayloadValue),
             consensusBlockValue: BigInt(consensusBlockValue),
+            shouldOverrideBuilder: false,
           });
         } else {
           chainStub.produceBlock.mockRejectedValue(Error("not produced"));
@@ -99,6 +100,7 @@ describe("api/validator - produceBlockV3", function () {
             block: blindedBlock,
             executionPayloadValue: BigInt(builderPayloadValue),
             consensusBlockValue: BigInt(consensusBlockValue),
+            shouldOverrideBuilder: false,
           });
         } else {
           chainStub.produceBlindedBlock.mockRejectedValue(Error("not produced"));