fix: fix tests

Author: rictorlome

src/fixtures/pvm.ts

@@ -389,7 +389,7 @@ export const disableL1ValidatorTxBytes = () =>
   concatBytes(baseTxbytes(), idBytes(), bytesForInt(10), inputBytes());
 
 export const feeState = (): FeeState => ({
-  capacity: 1n,
+  capacity: 1_000_000_000n,
   excess: 1n,
   price: 1n,
   timestamp: new Date().toISOString(),

src/vms/pvm/etna-builder/spend-reducers/verifyGasUsage.test.ts

@@ -7,9 +7,9 @@ describe('verifyGasUsage', () => {
   test('returns original state if gas is under the threshold', () => {
     const initialState = getInitialReducerState();
     const spendHelper = getSpendHelper();
-    const spy = vi.spyOn(spendHelper, 'verifyAssetsConsumed');
+    const spy = vi.spyOn(spendHelper, 'verifyGasUsage');
 
-    const state = verifyAssetsConsumed(initialState, spendHelper, testContext);
+    const state = verifyGasUsage(initialState, spendHelper, testContext);
 
     expect(state).toBe(initialState);
     expect(spy).toHaveBeenCalledTimes(1);
@@ -19,14 +19,14 @@ describe('verifyGasUsage', () => {
     const initialState = getInitialReducerState();
     const spendHelper = getSpendHelper();
 
-    // Mock the verifyAssetsConsumed method to throw an error
+    // Mock the verifyGasUsage method to throw an error
     // Testing for this function can be found in the spendHelper.test.ts file
-    spendHelper.verifyAssetsConsumed = vi.fn(() => {
+    spendHelper.verifyGasUsage = vi.fn(() => {
       throw new Error('Test error');
     });
 
     expect(() =>
-      verifyAssetsConsumed(initialState, spendHelper, testContext),
+      verifyGasUsage(initialState, spendHelper, testContext),
     ).toThrow('Test error');
   });
 });

src/vms/pvm/etna-builder/spend-reducers/verifyGasUsage.ts

@@ -5,15 +5,12 @@ import type { SpendReducerFunction } from './types';
  *
  * Calls the spendHelper's verifyGasUsage method.
  */
-export const verifyGasUsage: SpendReducerFunction = (
-  state,
-  spendHelper,
-) => {
+export const verifyGasUsage: SpendReducerFunction = (state, spendHelper) => {
   const verifyError = spendHelper.verifyGasUsage();
 
   if (verifyError) {
     throw verifyError;
   }
 
   return state;
-};
+};

src/vms/pvm/etna-builder/spend.ts

@@ -9,7 +9,11 @@ import type { Dimensions } from '../../common/fees/dimensions';
 import type { Context } from '../../context';
 import type { FeeState } from '../models';
 import type { SpendReducerFunction, SpendReducerState } from './spend-reducers';
-import { handleFeeAndChange, verifyAssetsConsumed, verifyGasUsage } from './spend-reducers';
+import {
+  handleFeeAndChange,
+  verifyAssetsConsumed,
+  verifyGasUsage,
+} from './spend-reducers';
 import { SpendHelper } from './spendHelper';
 
 type SpendResult = Readonly<{
@@ -145,7 +149,7 @@ export const spend = (
       ...spendReducers,
       verifyAssetsConsumed,
       handleFeeAndChange,
-      verifyGasUsage // This should happen after change is added
+      verifyGasUsage, // This should happen after change is added
       // Consolidation and sorting happens in the SpendHelper.
     ];
 

src/vms/pvm/etna-builder/spendHelper.test.ts

@@ -374,6 +374,37 @@ describe('src/vms/pvm/etna-builder/spendHelper', () => {
       );
     });
   });
+  describe('SpendHelper.verifyGasUsage', () => {
+    test('returns null when gas is under capacity', () => {
+      const spendHelper = new SpendHelper({
+        ...DEFAULT_PROPS,
+      });
+
+      const changeOutput = transferableOutput();
+
+      spendHelper.addChangeOutput(changeOutput);
+
+      expect(spendHelper.verifyGasUsage()).toBe(null);
+    });
+
+    test('returns an error when gas is over capacity', () => {
+      const spendHelper = new SpendHelper({
+        ...DEFAULT_PROPS,
+        feeState: {
+          ...DEFAULT_FEE_STATE,
+          capacity: 0n,
+        },
+      });
+
+      const changeOutput = transferableOutput();
+
+      spendHelper.addChangeOutput(changeOutput);
+
+      expect(spendHelper.verifyGasUsage()).toEqual(
+        new Error('Gas usage of transaction (113) exceeds capacity (0)'),
+      );
+    });
+  });
 
   test('no consolidated outputs when `shouldConsolidateOutputs` is `false`', () => {
     const spendHelper = new SpendHelper(DEFAULT_PROPS);

src/vms/pvm/etna-builder/spendHelper.ts

@@ -296,7 +296,7 @@ export class SpendHelper {
 
     return null;
   }
-  
+
   /**
    * Verifies that gas usage does not exceed the fee state maximum.
    *
@@ -305,7 +305,9 @@ export class SpendHelper {
   verifyGasUsage(): Error | null {
     const gas = this.calculateGas();
     if (this.feeState.capacity <= gas) {
-      return new Error(`Gas usage of transaction (${gas.toString()}) exceeds capacity (${this.feeState.capacity.toString()})`)
+      return new Error(
+        `Gas usage of transaction (${gas.toString()}) exceeds capacity (${this.feeState.capacity.toString()})`,
+      );
     }
 
     return null;