test(signature-fetcher): add tests getting signatures from store

ar-io · Sep 5, 2024 · afc0409 · afc0409
1 parent 8a5cffd
commit afc0409
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/src/data/signature-fetcher.test.ts b/src/data/signature-fetcher.test.ts
@@ -68,6 +68,14 @@ describe('SignatureFetcher', () => {
       assert.strictEqual(result, undefined);
     });
 
+    it('should return signature from signature store if it exists', async () => {
+      mock.method(signatureStore, 'get', async () => 'signature-from-store');
+
+      const result = await signatureFetcher.getDataItemSignature('testId');
+
+      assert.strictEqual(result, 'signature-from-store');
+    });
+
     it('should return signature if it exists in attributes', async () => {
       const testSignature = 'testSignature';
       mock.method(dataIndex, 'getDataItemAttributes', async () => ({
@@ -124,6 +132,14 @@ describe('SignatureFetcher', () => {
       assert.strictEqual(result, undefined);
     });
 
+    it('should return signature from signature store if it exists', async () => {
+      mock.method(signatureStore, 'get', async () => 'signature-from-store');
+
+      const result = await signatureFetcher.getDataItemSignature('testId');
+
+      assert.strictEqual(result, 'signature-from-store');
+    });
+
     it('should return signature if it exists in attributes', async () => {
       const testSignature = 'testSignature';
       mock.method(dataIndex, 'getTransactionAttributes', async () => ({