tests: adding seek tests for getClaims

[ci skip]

Author: tegefaulkes

src/sigchain/Sigchain.ts

@@ -284,7 +284,7 @@ class Sigchain {
   ): AsyncGenerator<[ClaimId, Claim]> {
     if (tran == null) {
       return yield* this.db.withTransactionG((tran) =>
-        this.getClaims({ order, seek }, tran),
+        this.getClaims({ order, seek, limit }, tran),
       );
     }
     const orderOptions =

tests/sigchain/Sigchain.test.ts

@@ -16,6 +16,7 @@ import * as sigchainErrors from '@/sigchain/errors';
 import * as keysUtils from '@/keys/utils';
 import * as claimsUtils from '@/claims/utils';
 import * as utils from '@/utils';
+import * as testsNodesUtils from '../nodes/utils';
 
 describe(Sigchain.name, () => {
   const password = keysUtils.getRandomBytes(10).toString('utf-8');
@@ -317,4 +318,62 @@ describe(Sigchain.name, () => {
       await sigchain.stop();
     },
   );
+  test('getClaims with seek ascending', async () => {
+    const sigchain = await Sigchain.createSigchain({
+      keyRing,
+      db,
+      logger,
+      fresh: true,
+    });
+    const claims: Array<[ClaimId, SignedClaim]> = [];
+    for (let i = 0; i < 3; i++) {
+      claims.push(await sigchain.addClaim({}));
+    }
+    const claimsAsc = await AsyncIterable.as(
+      sigchain.getClaims({ seek: claims[1][0], order: 'asc' }),
+    ).toArray();
+    expect(claimsAsc).toHaveLength(2);
+    // The claim we seeked to is included
+    expect(claimsAsc[0][0].equals(claims[1][0])).toBeTrue();
+    // And the claim after
+    expect(claimsAsc[1][0].equals(claims[2][0])).toBeTrue();
+  });
+  test('getClaims with seek descending', async () => {
+    const sigchain = await Sigchain.createSigchain({
+      keyRing,
+      db,
+      logger,
+      fresh: true,
+    });
+    const claims: Array<[ClaimId, SignedClaim]> = [];
+    for (let i = 0; i < 3; i++) {
+      claims.push(await sigchain.addClaim({}));
+    }
+    const claimsAsc = await AsyncIterable.as(
+      sigchain.getClaims({ seek: claims[1][0], order: 'desc' }),
+    ).toArray();
+    expect(claimsAsc).toHaveLength(2);
+    // The claim we seeked to is included
+    expect(claimsAsc[0][0].equals(claims[1][0])).toBeTrue();
+    // And the claim after
+    expect(claimsAsc[1][0].equals(claims[0][0])).toBeTrue();
+  });
+  test('getClaims with seek with limit', async () => {
+    const sigchain = await Sigchain.createSigchain({
+      keyRing,
+      db,
+      logger,
+      fresh: true,
+    });
+    const claims: Array<[ClaimId, SignedClaim]> = [];
+    for (let i = 0; i < 3; i++) {
+      claims.push(await sigchain.addClaim({}));
+    }
+    const claimsAsc = await AsyncIterable.as(
+      sigchain.getClaims({ seek: claims[1][0], limit: 1 }),
+    ).toArray();
+    expect(claimsAsc).toHaveLength(1);
+    // The claim we seeked to is included
+    expect(claimsAsc[0][0].equals(claims[1][0])).toBeTrue();
+  });
 });