Merge pull request #196 from fleet-sdk/ergobox-cloning

Support instantiating ErgoBox from another ErgoBox

Author: arobsn

.changeset/sharp-boxes-take.md

@@ -0,0 +1,5 @@
+---
+"@fleet-sdk/core": patch
+---
+
+Support instantiating `ErgoBox` from another `ErgoBox`

packages/core/src/models/ergoBox.spec.ts

@@ -31,6 +31,14 @@ describe("Construction", () => {
     expect(ergoBox.index).to.be.equal(tv.index);
   });
 
+  test.each(regularBoxes)("Should construct from an ErgoBox candidate", (tv) => {
+    const ergoBox = new ErgoBox(new ErgoBox(tv), tv.transactionId, tv.index);
+
+    expect(ergoBox.boxId).to.be.equal(tv.boxId);
+    expect(ergoBox.transactionId).to.be.equal(tv.transactionId);
+    expect(ergoBox.index).to.be.equal(tv.index);
+  });
+
   it("Should throw if transactionId or index is not provided for box candidate", () => {
     const box = regularBoxes[0];
     const candidate = boxToCandidate(box);

packages/core/src/models/ergoBox.ts

@@ -62,12 +62,18 @@ export class ErgoBox<R extends NonMandatoryRegisters = NonMandatoryRegisters> {
   constructor(candidate: ErgoBoxCandidate<R>, transactionId: string, index: number);
   constructor(candidate: BoxCandidate<Amount, R>, transactionId: string, index: number);
   constructor(box: Box<Amount, R>);
+  constructor(box: ErgoBox<R>);
   constructor(
-    box: Box<Amount, R> | ErgoBoxCandidate<R> | BoxCandidate<Amount, R>,
+    box: Box<Amount, R> | ErgoBox<R> | ErgoBoxCandidate<R> | BoxCandidate<Amount, R>,
     transactionId?: string,
     index?: number
   ) {
-    if (isBox(box)) {
+    if (box instanceof ErgoBox) {
+      this.#candidate = box.#candidate;
+      this.#transactionId = box.transactionId;
+      this.#index = box.index;
+      this.#boxId = box.boxId;
+    } else if (isBox(box)) {
       this.#candidate = new ErgoBoxCandidate(box);
       this.#transactionId = box.transactionId;
       this.#index = box.index;

packages/core/src/models/ergoUnsignedInput.spec.ts

@@ -1,23 +1,36 @@
 import { SBool } from "@fleet-sdk/serializer";
 import { regularBoxes } from "_test-vectors";
-import { describe, expect, it } from "vitest";
+import { ErgoBox } from "packages/core/dist";
+import { describe, expect, it, test } from "vitest";
 import { ErgoUnsignedInput } from "./ergoUnsignedInput";
 
 describe("Construction", () => {
-  it("Should construct from a vanilla object", () => {
-    for (const box of regularBoxes) {
-      const input = new ErgoUnsignedInput(box);
+  test.each(regularBoxes)("Should construct from a vanilla object", (box) => {
+    const input = new ErgoUnsignedInput(box);
 
-      expect(input.boxId).toBe(box.boxId);
-      expect(input.value).toBe(box.value);
-      expect(input.ergoTree).toBe(box.ergoTree);
-      expect(input.assets).toEqual(box.assets);
-      expect(input.creationHeight).toBe(box.creationHeight);
-      expect(input.additionalRegisters).toBe(box.additionalRegisters);
-      expect(input.transactionId).toBe(box.transactionId);
-      expect(input.index).toBe(box.index);
-      expect(input.extension).to.be.deep.equal({});
-    }
+    expect(input.boxId).toBe(box.boxId);
+    expect(input.value).toBe(box.value);
+    expect(input.ergoTree).toBe(box.ergoTree);
+    expect(input.assets).toEqual(box.assets);
+    expect(input.creationHeight).toBe(box.creationHeight);
+    expect(input.additionalRegisters).toBe(box.additionalRegisters);
+    expect(input.transactionId).toBe(box.transactionId);
+    expect(input.index).toBe(box.index);
+    expect(input.extension).to.be.deep.equal({});
+  });
+
+  test.each(regularBoxes)("Should construct from a ErgoBox object", (box) => {
+    const input = new ErgoUnsignedInput(new ErgoBox(box));
+
+    expect(input.boxId).toBe(box.boxId);
+    expect(input.value).toBe(box.value);
+    expect(input.ergoTree).toBe(box.ergoTree);
+    expect(input.assets).toEqual(box.assets);
+    expect(input.creationHeight).toBe(box.creationHeight);
+    expect(input.additionalRegisters).toBe(box.additionalRegisters);
+    expect(input.transactionId).toBe(box.transactionId);
+    expect(input.index).toBe(box.index);
+    expect(input.extension).to.be.deep.equal({});
   });
 });