From aeb6518c72741adef2861a265b44890cf743dc24 Mon Sep 17 00:00:00 2001
From: Karl Prieb <karl@ar.io>
Date: Thu, 5 Sep 2024 15:28:03 -0300
Subject: [PATCH] test(signature-fetcher): add tests getting signatures from
 store

---
 src/data/signature-fetcher.test.ts | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

diff --git a/src/data/signature-fetcher.test.ts b/src/data/signature-fetcher.test.ts
index 654f8f46..78940a2a 100644
--- 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 () => ({