add idb tests

prax-wallet · Jun 13, 2024 · b60e273 · b60e273
1 parent bcd6c0b
commit b60e273
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 5 deletions.
diff --git a/packages/storage/src/indexed-db/index.ts b/packages/storage/src/indexed-db/index.ts
@@ -669,12 +669,18 @@ export class IndexedDb implements IndexedDbInterface {
   }
 
   async clearSwapBasedPrices(): Promise<void> {
-    for await (const cursor of this.db.transaction('PRICES').store) {
+    const tx = this.db.transaction('PRICES', 'readwrite');
+    const store = tx.objectStore('PRICES');
+
+    let cursor = await store.openCursor();
+    while (cursor) {
       const price = EstimatedPrice.fromJson(cursor.value);
-      // Do not delete prices for delegation assets
-      if (price.numeraire?.equals(this.stakingTokenAssetId)) continue;
-      await this.db.delete('PRICES', cursor.key);
+      if (!price.numeraire?.equals(this.stakingTokenAssetId)) {
+        cursor.delete();
+      }
+      cursor = await cursor.continue();
     }
+    await tx.done;
   }
 
   private determinePriceRelevanceThresholdForAsset(assetMetadata: Metadata): number {

diff --git a/packages/storage/src/indexed-db/indexed-db.test.ts b/packages/storage/src/indexed-db/indexed-db.test.ts
@@ -594,9 +594,12 @@ describe('IndexedDb', () => {
 
     const numeraireAssetId = new AssetId({ inner: new Uint8Array([5, 6, 7, 8]) });
 
+    const stakingAssetId = AssetId.fromJson({
+      inner: 'KeqcLzNx9qSH5+lcJHBB9KNW+YPrBk5dKzvPMiypahA=',
+    });
     beforeEach(async () => {
       db = await IndexedDb.initialize({ ...generateInitialProps() });
-      await db.updatePrice(delegationMetadataA.penumbraAssetId!, numeraireAssetId, 1.23, 50n);
+      await db.updatePrice(delegationMetadataA.penumbraAssetId!, stakingAssetId, 1.23, 50n);
       await db.updatePrice(metadataA.penumbraAssetId!, numeraireAssetId, 22.15, 40n);
     });
 
@@ -628,6 +631,11 @@ describe('IndexedDb', () => {
         }),
       ]);
     });
+
+    it('should delete only prices with a numeraires different from the staking token', async () => {
+      await db.clearSwapBasedPrices();
+      await expect(db.getPricesForAsset(metadataA, 50n)).resolves.toEqual([]);
+    });
   });
 
   describe('upsertAuction()', () => {