From 4dea1263ab06767bbca5255f8c0179b6241839fa Mon Sep 17 00:00:00 2001
From: ejMina226 <118474890+ejMina226@users.noreply.github.com>
Date: Thu, 18 Jul 2024 12:28:14 +0100
Subject: [PATCH] Initial Commit

---
 .nvmrc                                 |   1 +
 packages/sdk/test/fees-failure.test.ts | 143 +++++++++++++++++++++++++
 2 files changed, 144 insertions(+)
 create mode 100644 .nvmrc
 create mode 100644 packages/sdk/test/fees-failure.test.ts

diff --git a/.nvmrc b/.nvmrc
new file mode 100644
index 00000000..e8b25b54
--- /dev/null
+++ b/.nvmrc
@@ -0,0 +1 @@
+v18.14.2
\ No newline at end of file
diff --git a/packages/sdk/test/fees-failure.test.ts b/packages/sdk/test/fees-failure.test.ts
new file mode 100644
index 00000000..558152f3
--- /dev/null
+++ b/packages/sdk/test/fees-failure.test.ts
@@ -0,0 +1,143 @@
+import "reflect-metadata";
+import { Balance, Balances, BalancesKey, TokenId } from "@proto-kit/library";
+import {
+  runtimeMethod,
+  runtimeModule,
+  RuntimeModule,
+  RuntimeModulesRecord,
+} from "@proto-kit/module";
+import { PrivateKey } from "o1js";
+import { inject } from "tsyringe";
+import { expectDefined } from "@proto-kit/common";
+
+import { TestingAppChain } from "../src";
+
+@runtimeModule()
+class Faucet extends RuntimeModule<unknown> {
+  public constructor(@inject("Balances") public balances: Balances) {
+    super();
+  }
+
+  @runtimeMethod()
+  public async drip() {
+    await this.balances.mint(
+      TokenId.from(0),
+      this.transaction.sender.value,
+      Balance.from(1000)
+    );
+  }
+}
+
+@runtimeModule()
+class Pit extends RuntimeModule<unknown> {
+  public constructor(@inject("Balances") public balances: Balances) {
+    super();
+  }
+
+  @runtimeMethod()
+  public async burn(amount: Balance) {
+    await this.balances.burn(
+      TokenId.from(0),
+      this.transaction.sender.value,
+      amount
+    );
+  }
+}
+
+// eslint-disable-next-line @typescript-eslint/no-unused-vars
+interface RuntimeModules extends RuntimeModulesRecord {
+  Faucet: typeof Faucet;
+  Pit: typeof Pit;
+}
+
+describe("fees", () => {
+  const feeRecipientKey = PrivateKey.random();
+  const senderKey = PrivateKey.random();
+
+  const appChain = TestingAppChain.fromRuntime({
+    Faucet,
+    Pit,
+  });
+
+  beforeAll(async () => {
+    appChain.configurePartial({
+      Runtime: {
+        Faucet: {},
+        Pit: {},
+        Balances: {},
+      },
+      Protocol: {
+        ...appChain.config.Protocol!,
+        TransactionFee: {
+          tokenId: 0n,
+          feeRecipient: feeRecipientKey.toPublicKey().toBase58(),
+          baseFee: 10000n,
+          perWeightUnitFee: 1n,
+          methods: {
+            "Faucet.drip": {
+              baseFee: 0n,
+              weight: 0n,
+              perWeightUnitFee: 0n,
+            },
+          },
+        },
+      },
+    });
+
+    await appChain.start();
+    appChain.setSigner(senderKey);
+  });
+
+  it("should allow a free faucet transaction", async () => {
+    expect.assertions(2);
+
+    const faucet = appChain.runtime.resolve("Faucet");
+
+    const tx = await appChain.transaction(senderKey.toPublicKey(), async () => {
+      await faucet.drip();
+    });
+
+    await tx.sign();
+    await tx.send();
+
+    await appChain.produceBlock();
+
+    const balance = await appChain.query.runtime.Balances.balances.get(
+      new BalancesKey({
+        tokenId: new TokenId(0),
+        address: senderKey.toPublicKey(),
+      })
+    );
+
+    expectDefined(balance);
+    expect(balance.toString()).toBe("1000");
+  });
+
+  it("should crash when the transaction fee exceeds the balance", async () => {
+
+    const balances = appChain.runtime.resolve("Balances");
+
+    const tx = await appChain.transaction(senderKey.toPublicKey(), async () => {
+      await balances.transfer(
+        TokenId.from(0),
+        senderKey.toPublicKey(),
+        feeRecipientKey.toPublicKey(),
+        Balance.from(100));
+    });
+
+    await tx.sign();
+    await tx.send();
+    await appChain.produceBlock();
+
+    console.log('Printed Block')
+    const balance = await appChain.query.runtime.Balances.balances.get(
+      new BalancesKey({
+        tokenId: new TokenId(0),
+        address: senderKey.toPublicKey(),
+      })
+    );
+
+    expectDefined(balance);
+    expect(balance.toString()).toBe("512");
+  });
+});