fix(warp contract): added test for getting state after connecting wit…

…h warp
ar-io · Mar 22, 2024 · 060ee2c · 060ee2c
1 parent 9869415
commit 060ee2c
Show file tree

Hide file tree

Showing 18 changed files with 24 additions and 5 deletions.
diff --git a/cache/warp/contracts/000003.log b/cache/warp/contracts/000003.log
diff --git a/cache/warp/contracts/CURRENT b/cache/warp/contracts/CURRENT
@@ -0,0 +1 @@
+MANIFEST-000002
diff --git a/cache/warp/contracts/LOCK b/cache/warp/contracts/LOCK
diff --git a/cache/warp/contracts/LOG b/cache/warp/contracts/LOG
@@ -0,0 +1 @@
+2024/03/22-08:17:56.172701 17335b000 Delete type=3 #1
diff --git a/cache/warp/contracts/MANIFEST-000002 b/cache/warp/contracts/MANIFEST-000002
diff --git a/cache/warp/source/000003.log b/cache/warp/source/000003.log
diff --git a/cache/warp/source/CURRENT b/cache/warp/source/CURRENT
@@ -0,0 +1 @@
+MANIFEST-000002
diff --git a/cache/warp/source/LOCK b/cache/warp/source/LOCK
diff --git a/cache/warp/source/LOG b/cache/warp/source/LOG
@@ -0,0 +1 @@
+2024/03/22-08:17:57.525010 173b63000 Delete type=3 #1
diff --git a/cache/warp/source/MANIFEST-000002 b/cache/warp/source/MANIFEST-000002
diff --git a/cache/warp/state/000003.log b/cache/warp/state/000003.log
diff --git a/cache/warp/state/CURRENT b/cache/warp/state/CURRENT
@@ -0,0 +1 @@
+MANIFEST-000002
diff --git a/cache/warp/state/LOCK b/cache/warp/state/LOCK
diff --git a/cache/warp/state/LOG b/cache/warp/state/LOG
@@ -0,0 +1 @@
+2024/03/22-08:17:56.169630 17436b000 Delete type=3 #1
diff --git a/cache/warp/state/MANIFEST-000002 b/cache/warp/state/MANIFEST-000002
diff --git a/src/common/ant.ts b/src/common/ant.ts
@@ -60,7 +60,7 @@ export class ANT implements ANTContract, BaseContract<ANTState> {
   /**
    * Returns the current state of the contract.
    */
-  async getState(params: EvaluationParameters): Promise<ANTState> {
+  async getState(params: EvaluationParameters = {}): Promise<ANTState> {
     const state = await this.contract.getState(params);
     return state;
   }

diff --git a/src/common/contracts/warp-contract.ts b/src/common/contracts/warp-contract.ts
@@ -43,16 +43,20 @@ export class WarpContract<T> implements BaseContract<T>, ReadContract {
   constructor({
     contractTxId,
     cacheUrl,
-    warp = WarpFactory.forMainnet({
-      ...defaultCacheOptions,
-      inMemory: true, // default to in memory for now, a custom warp implementation can be provided
-    }),
+    warp = WarpFactory.forMainnet(
+      {
+        ...defaultCacheOptions,
+        // inMemory: true, // default to in memory for now, a custom warp implementation can be provided
+      },
+      true,
+    ),
   }: {
     contractTxId: string;
     cacheUrl?: string;
     warp?: Warp;
     signer?: ContractSigner;
   }) {
+    this.contractTxId = contractTxId;
     this.contract = warp.contract<T>(contractTxId);
     this.cacheUrl = cacheUrl;
   }

diff --git a/tests/ant.test.ts b/tests/ant.test.ts
@@ -143,4 +143,13 @@ describe('ANT contract apis', () => {
       expect(state).toBeDefined();
     },
   );
+
+  it('Should get state with warp contract', async () => {
+    const jwk = await arweave.wallets.generate();
+    const signer = new ArweaveSigner(jwk);
+    // connecting updates contract to use warp
+    ant.connect(signer);
+    const state = await ant.getState();
+    expect(state).toBeDefined();
+  });
 });