Basic unit tests for setAncestor

Added tests for the query proto, one to make sure the query structure is right when setting ancestor once and one to make sure the query structure is right when setting ancestor twice. Also added a unit test to make sure that ancestor is set the right way internally when using setAncestor.

Author: danieljbruce

test/entity.ts

@@ -1915,6 +1915,52 @@ describe('entity', () => {
       assert.deepStrictEqual(entity.queryToQueryProto(query), queryProto);
     });
 
+    it('should support the filter method with setAncestor', () => {
+      const ancestorKey = new entity.Key({
+        path: ['Kind2', 'somename'],
+      });
+
+      const ds = new Datastore({projectId: 'project-id'});
+
+      const query = ds
+        .createQuery('Kind1')
+        .filter(new PropertyFilter('name', '=', 'John'))
+        .start('start')
+        .end('end')
+        .groupBy(['name'])
+        .order('name')
+        .select('name')
+        .limit(1)
+        .offset(1)
+        .setAncestor(ancestorKey);
+      assert.deepStrictEqual(entity.queryToQueryProto(query), queryProto);
+    });
+
+    it('should support the filter method using setAncestor twice', () => {
+      const oldAncestorKey = new entity.Key({
+        path: ['Kind1', 'somename1'],
+      });
+      const ancestorKey = new entity.Key({
+        path: ['Kind2', 'somename'],
+      });
+
+      const ds = new Datastore({projectId: 'project-id'});
+
+      const query = ds
+        .createQuery('Kind1')
+        .filter(new PropertyFilter('name', '=', 'John'))
+        .start('start')
+        .end('end')
+        .groupBy(['name'])
+        .order('name')
+        .select('name')
+        .limit(1)
+        .offset(1)
+        .setAncestor(oldAncestorKey)
+        .setAncestor(ancestorKey);
+      assert.deepStrictEqual(entity.queryToQueryProto(query), queryProto);
+    });
+
     it('should support the filter method with AND', () => {
       const ancestorKey = new entity.Key({
         path: ['Kind2', 'somename'],

test/query.ts

@@ -178,7 +178,13 @@ describe('Query', () => {
       new Query(['kind1']).filter('name', 'Stephen');
     });
   });
-
+  describe('setAncestor', () => {
+    it('should set the ancestor for the query to search against', done => {
+      const key = ['kind2', 123];
+      const query = new Query(['kind1']).setAncestor(key);
+      assert.deepStrictEqual(query.ancestor, key);
+    });
+  });
   describe('filter with Filter class', () => {
     it('should support filter with Filter', () => {
       const now = new Date();