test: add namespaced datastore query tests

ipfs · Feb 21, 2024 · f15cd79 · f15cd79
1 parent ca09950
commit f15cd79
Showing 1 changed file with 45 additions and 9 deletions.
diff --git a/packages/datastore-core/test/namespace.spec.ts b/packages/datastore-core/test/namespace.spec.ts
@@ -13,19 +13,20 @@ describe('NamespaceDatastore', () => {
     'abc',
     ''
   ]
+
+  const keys = [
+    'foo',
+    'foo/bar',
+    'foo/bar/baz',
+    'foo/barb',
+    'foo/bar/bazb',
+    'foo/bar/baz/barb'
+  ].map((s) => new Key(s))
+
   prefixes.forEach((prefix) => it(`basic '${prefix}'`, async () => {
     const mStore = new MemoryDatastore()
     const store = new NamespaceDatastore(mStore, new Key(prefix))
 
-    const keys = [
-      'foo',
-      'foo/bar',
-      'foo/bar/baz',
-      'foo/barb',
-      'foo/bar/bazb',
-      'foo/bar/baz/barb'
-    ].map((s) => new Key(s))
-
     await Promise.all(keys.map(async key => { await store.put(key, uint8ArrayFromString(key.toString())) }))
     const nResults = Promise.all(keys.map(async (key) => store.get(key)))
     const mResults = Promise.all(keys.map(async (key) => mStore.get(new Key(prefix).child(key))))
@@ -45,6 +46,41 @@ describe('NamespaceDatastore', () => {
     expect(results[0]).to.eql(results[1])
   }))
 
+  const setupStores = async (keys: Key[]): Promise<NamespaceDatastore[]> => {
+    const prefixes = ['abc', '123', 'sub/prefix']
+    const mStore = new MemoryDatastore()
+
+    return Promise.all(prefixes.map(async prefix => {
+      const store = new NamespaceDatastore(mStore, new Key(prefix))
+
+      await Promise.all(keys.map(async key => { await store.put(key, uint8ArrayFromString(key.toString())) }))
+
+      return store
+    }))
+  }
+
+  it('queries keys under each prefix', async () => {
+    const stores = await setupStores(keys)
+
+    for (const store of stores) {
+      const nRes = await all(store.queryKeys({}))
+
+      expect(nRes).deep.equal(keys)
+    }
+  })
+
+  it('queries values under each prefix', async () => {
+    const stores = await setupStores(keys)
+
+    for (const store of stores) {
+      const nRes = await all(store.query({}))
+      const values = keys.map(key => uint8ArrayFromString(key.toString()))
+
+      expect(nRes.map(p => p.key)).deep.equal(keys)
+      expect(nRes.map(p => p.value)).deep.equal(values)
+    }
+  })
+
   prefixes.forEach((prefix) => {
     describe(`interface-datastore: '${prefix}'`, () => {
       interfaceDatastoreTests({