From b119f4057649ee47858c17d6ead8af771ccc4c75 Mon Sep 17 00:00:00 2001
From: arobsn <87387688+arobsn@users.noreply.github.com>
Date: Mon, 16 Sep 2024 09:24:25 -0300
Subject: [PATCH] tweak plugin context to use `ensureInclusion` flag

---
 packages/core/src/builder/pluginContext.spec.ts | 15 ++++++---------
 packages/core/src/builder/pluginContext.ts      |  2 +-
 2 files changed, 7 insertions(+), 10 deletions(-)

diff --git a/packages/core/src/builder/pluginContext.spec.ts b/packages/core/src/builder/pluginContext.spec.ts
index 18ed862..c298a03 100644
--- a/packages/core/src/builder/pluginContext.spec.ts
+++ b/packages/core/src/builder/pluginContext.spec.ts
@@ -17,29 +17,26 @@ describe("Plugin context", () => {
 
   it("Should add inputs", () => {
     const fromSpy = vi.spyOn(builder, "from");
-    const configureSelectorMethod = vi.spyOn(builder, "configureSelector");
-
     const newLen = context.addInputs(regularBoxes);
 
     expect(fromSpy).toBeCalledTimes(1);
-    expect(configureSelectorMethod).toBeCalledTimes(1);
+    expect(fromSpy).toBeCalledWith(regularBoxes, { ensureInclusion: true });
     expect(builder.inputs).toHaveLength(newLen);
     expect(builder.inputs).toHaveLength(regularBoxes.length);
   });
 
   it("Should add a single input", () => {
-    const fromMethod = vi.spyOn(builder, "from");
-    const configureSelectorMethod = vi.spyOn(builder, "configureSelector");
+    const fromSpy = vi.spyOn(builder, "from");
 
+    const inputs = regularBoxes[0];
     let newLen = context.addInputs(regularBoxes[0]);
-    expect(fromMethod).toBeCalledTimes(1);
-    expect(configureSelectorMethod).toBeCalledTimes(1);
+    expect(fromSpy).toBeCalledTimes(1);
+    expect(fromSpy).toBeCalledWith(inputs, { ensureInclusion: true });
     expect(builder.inputs).toHaveLength(newLen);
     expect(builder.inputs).toHaveLength(1);
 
     newLen = context.addInputs(regularBoxes[1]);
-    expect(fromMethod).toBeCalledTimes(2);
-    expect(configureSelectorMethod).toBeCalledTimes(2);
+    expect(fromSpy).toBeCalledTimes(2);
     expect(builder.inputs).toHaveLength(newLen);
     expect(builder.inputs).toHaveLength(2);
   });
diff --git a/packages/core/src/builder/pluginContext.ts b/packages/core/src/builder/pluginContext.ts
index 2be5012..994242e 100644
--- a/packages/core/src/builder/pluginContext.ts
+++ b/packages/core/src/builder/pluginContext.ts
@@ -5,7 +5,7 @@ import type {
   OneOrMore,
   TokenAmount
 } from "@fleet-sdk/common";
-import { NotAllowedTokenBurning, type OutputBuilder, type TransactionBuilder } from "..";
+import { type OutputBuilder, type TransactionBuilder, NotAllowedTokenBurning } from "..";
 
 export type FleetPluginContext = {
   /**