refactor: move TestBalances to a helpers folder in src

Author: 4ytuncs

packages/stack/test/graphql/utils.ts packages/stack/src/helpers/TestBalance.tspackages/stack/test/graphql/utils.ts renamed to packages/stack/src/helpers/TestBalance.ts

@@ -5,13 +5,10 @@ import {
   InMemoryTransactionSender,
   StateServiceQueryModule,
 } from "@proto-kit/sdk";
-import { PrivateKey, PublicKey } from "o1js";
-import { Runtime, runtimeMethod, runtimeModule } from "@proto-kit/module";
-import { Protocol, State, state } from "@proto-kit/protocol";
+import { PrivateKey } from "o1js";
+import { Runtime } from "@proto-kit/module";
+import { Protocol } from "@proto-kit/protocol";
 import {
-  Balance,
-  Balances,
-  BalancesKey,
   TokenId,
   VanillaProtocolModules,
   VanillaRuntimeModules,
@@ -44,42 +41,7 @@ import {
 } from "@proto-kit/api";
 import { container } from "tsyringe";
 
-@runtimeModule()
-export class TestBalances extends Balances {
-  /**
-   * We use `satisfies` here in order to be able to access
-   * presets by key in a type safe way.
-   */
-  // public static presets = {} satisfies Presets<object>;
-
-  @state() public totalSupply = State.from<UInt64>(UInt64);
-
-  @runtimeMethod()
-  public async getBalanceForUser(
-    tokenId: TokenId,
-    address: PublicKey
-  ): Promise<Balance> {
-    return await super.getBalance(tokenId, address);
-  }
-
-  @runtimeMethod()
-  public async addBalance(
-    tokenId: TokenId,
-    address: PublicKey,
-    balance: UInt64
-  ) {
-    const totalSupply = await this.totalSupply.get();
-    await this.totalSupply.set(totalSupply.orElse(UInt64.zero).add(balance));
-
-    const previous = await this.balances.get(
-      new BalancesKey({ tokenId, address })
-    );
-    await this.balances.set(
-      new BalancesKey({ tokenId, address }),
-      previous.orElse(UInt64.zero).add(balance)
-    );
-  }
-}
+import { TestBalances } from "../../helpers/TestBalance";
 
 export async function startServer() {
   log.setLevel("DEBUG");

packages/stack/src/scripts/graphql/server.ts

@@ -6,7 +6,7 @@ import {
 import { Protocol } from "@proto-kit/protocol";
 import { ModulesConfig, ModuleContainer, TypedClass } from "@proto-kit/common";
 
-import { TestBalances } from "../graphql/server";
+import { TestBalances } from "../../helpers/TestBalance";
 
 const runtime = Runtime.from(
   VanillaRuntimeModules.with({

packages/stack/src/scripts/worker/app.ts

@@ -6,7 +6,7 @@ import {
 import { Runtime } from "@proto-kit/module";
 import { Protocol } from "@proto-kit/protocol";
 
-import { TestBalances } from "../graphql/server";
+import { TestBalances } from "../../helpers/TestBalance";
 
 const client = ClientAppChain.fromRemoteEndpoint(
   Runtime.from(

packages/stack/src/scripts/worker/client.ts

@@ -31,7 +31,7 @@ import {
 import { container } from "tsyringe";
 import { S3RemoteCache } from "@proto-kit/deployment";
 
-import { TestBalances } from "../../src";
+import { TestBalances } from "../../src/helpers/TestBalance";
 import * as fs from "node:fs";
 import cachedir from "cachedir";
 

packages/stack/test-integration/caching/caching.test.ts

@@ -41,7 +41,7 @@ import {
 } from "@proto-kit/api";
 import { container } from "tsyringe";
 
-import { TestBalances } from "./utils";
+import { TestBalances } from "../../src/helpers/TestBalance";
 
 export async function startGraphqlServer() {
   log.setLevel("DEBUG");

packages/stack/test/graphql/graphql-server.ts

@@ -27,8 +27,9 @@ import {
 import { beforeAll } from "@jest/globals";
 import { container } from "tsyringe";
 
+import { TestBalances } from "../../src/helpers/TestBalance";
+
 import { startGraphqlServer } from "./graphql-server";
-import { TestBalances } from "./utils";
 
 const pk = PrivateKey.random();