add setFee method to TransactionBuilderContext

.changeset/hungry-houses-melt.md

@@ -0,0 +1,5 @@
+---
+"@fleet-sdk/core": patch
+---
+
+**Transaction Builder**: add `setFee` method to plugin context.

packages/core/src/builder/pluginContext.spec.ts

@@ -3,7 +3,7 @@ import { NotAllowedTokenBurning } from "../errors";
 import { regularBoxesMock } from "../tests/mocks/mockBoxes";
 import { OutputBuilder } from "./outputBuilder";
 import { createPluginContext, FleetPluginContext } from "./pluginContext";
-import { TransactionBuilder } from "./transactionBuilder";
+import { RECOMMENDED_MIN_FEE_VALUE, TransactionBuilder } from "./transactionBuilder";
 
 describe("Plugin context", () => {
   const creationHeight = 894169;
@@ -121,4 +121,16 @@ describe("Plugin context", () => {
     expect(burnTokensMethod).not.toBeCalled();
     expect(builder.burning).toBeUndefined();
   });
+
+  it("Should set fee amount", () => {
+    builder.from(regularBoxesMock);
+
+    const payFeeMethod = vi.spyOn(builder, "payFee");
+    const fee = RECOMMENDED_MIN_FEE_VALUE * 3n;
+
+    context.setFee(RECOMMENDED_MIN_FEE_VALUE * 3n);
+
+    expect(builder.fee).to.be.equal(fee);
+    expect(payFeeMethod).toBeCalledWith(fee);
+  });
 });

packages/core/src/builder/pluginContext.ts

@@ -33,6 +33,12 @@ export type FleetPluginContext = {
    * it will thrown a {@link NotAllowedTokenBurning} exception.
    */
   burnTokens: (tokens: OneOrMore<TokenAmount<Amount>>) => void;
+
+  /**
+   * Set transaction fee amount
+   * @param amount amount in nanoergs
+   */
+  setFee: (amount: Amount) => void;
 };
 
 export function createPluginContext(transactionBuilder: TransactionBuilder): FleetPluginContext {
@@ -53,6 +59,7 @@ export function createPluginContext(transactionBuilder: TransactionBuilder): Fle
         throw new NotAllowedTokenBurning();
       }
       transactionBuilder.burnTokens(tokens);
-    }
+    },
+    setFee: (amount) => transactionBuilder.payFee(amount)
   };
 }