Add tests for attributesToRetrieve in getDocuments

Author: bidoubiwa

tests/documents_tests.ts

@@ -67,11 +67,41 @@ describe.each([
       })
     await client.getIndex(uidAndPrimaryKey.uid).waitForPendingUpdate(updateId)
   })
+  test(`${permission} key: Get documents with string attributesToRetrieve`, async () => {
+    await client
+      .getIndex(uidNoPrimaryKey.uid)
+      .getDocuments({
+        attributesToRetrieve: 'id',
+      })
+      .then((response) => {
+        expect(response.find((x) => Object.keys(x).length !== 1)).toEqual(
+          undefined
+        )
+      })
+  })
+
+  test(`${permission} key: Get documents with array attributesToRetrieve`, async () => {
+    await client
+      .getIndex(uidNoPrimaryKey.uid)
+      .getDocuments({
+        attributesToRetrieve: ['id'],
+      })
+      .then((response) => {
+        expect(response.find((x) => Object.keys(x).length !== 1)).toEqual(
+          undefined
+        )
+      })
+  })
+
   test(`${permission} key: Get documents from index that has no primary key`, async () => {
     await client
       .getIndex(uidNoPrimaryKey.uid)
-      .getDocuments()
+      .getDocuments({
+        attributesToRetrieve: 'id',
+      })
       .then((response) => {
+        console.log({ response })
+
         expect(response.length).toEqual(dataset.length)
       })
   })
@@ -83,6 +113,7 @@ describe.each([
         expect(response.length).toEqual(dataset.length)
       })
   })
+
   test(`${permission} key: Replace documents from index that has NO primary key`, async () => {
     const id = 2
     const title = 'The Red And The Black'