fix(document datasource): handle the correct return type for the AQL …

…query
danwkennedy · Jul 2, 2019 · 69204fb · 69204fb
1 parent 7f4b4b1
commit 69204fb
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 7 deletions.
diff --git a/arango-document-datasource.js b/arango-document-datasource.js
@@ -113,7 +113,7 @@ class ArangoDocumentDataSource extends DataSource {
       FOR key in ${keys}
         RETURN (DOCUMENT(key))._id
     `);
-    const [nodes] = await cursor.all();
+    const nodes = await cursor.all();
 
     const output = [];
     for (const [index, id] of keys.entries()) {

diff --git a/arango-document-datasource.test.js b/arango-document-datasource.test.js
@@ -44,7 +44,7 @@ describe('ArangDocumentDataSource', () => {
   describe(`exists`, () => {
     test(`it returns true if a document exists`, async () => {
       const results = ['123'];
-      const db = createDb(results);
+      const db = createExistsDb(results);
       const datasource = new ArangoDocumentDataSource(db);
       datasource.initialize();
 
@@ -55,7 +55,7 @@ describe('ArangDocumentDataSource', () => {
 
     test(`it caches the results`, async () => {
       const results = ['123'];
-      const db = createDb(results);
+      const db = createExistsDb(results);
       const datasource = new ArangoDocumentDataSource(db);
       datasource.initialize();
 
@@ -67,7 +67,7 @@ describe('ArangDocumentDataSource', () => {
 
     test(`it batches the calls`, async () => {
       const results = ['123'];
-      const db = createDb(results);
+      const db = createExistsDb(results);
       const datasource = new ArangoDocumentDataSource(db);
       datasource.initialize();
 
@@ -80,7 +80,7 @@ describe('ArangDocumentDataSource', () => {
   describe(`manyExist`, () => {
     test(`it returns a list of document existence`, async () => {
       const results = ['123'];
-      const db = createDb(results);
+      const db = createExistsDb(results);
       const datasource = new ArangoDocumentDataSource(db);
       datasource.initialize();
 
@@ -92,7 +92,7 @@ describe('ArangDocumentDataSource', () => {
 
     test(`it caches the results`, async () => {
       const results = ['123'];
-      const db = createDb(results);
+      const db = createExistsDb(results);
       const datasource = new ArangoDocumentDataSource(db);
       datasource.initialize();
 
@@ -104,7 +104,7 @@ describe('ArangDocumentDataSource', () => {
 
     test(`it batches the calls`, async () => {
       const results = ['123'];
-      const db = createDb(results);
+      const db = createExistsDb(results);
       const datasource = new ArangoDocumentDataSource(db);
       datasource.initialize();
 
@@ -125,3 +125,11 @@ function createDb(queryResults) {
     })
   };
 }
+
+function createExistsDb(queryResults) {
+  return {
+    query: jest.fn().mockResolvedValue({
+      all: () => queryResults
+    })
+  };
+}