Adding Firestore.setLogFunction() to the typings; Adding tests (#123)

Author: hiranya911

src/index.d.ts

@@ -410,6 +410,7 @@ declare namespace admin.firestore {
   export import FieldValue = _firestore.FieldValue;
   export import Firestore = _firestore.Firestore;
   export import GeoPoint = _firestore.GeoPoint;
+  export import setLogFunction = _firestore.setLogFunction;
 }
 
 declare module 'firebase-admin' {

test/integration/firestore.js

@@ -126,12 +126,33 @@ function test(utils) {
         });
   }
 
-  return Promise.resolve()
+  function testSetLogFunction() {
+    const logs = [];
+    const source = admin.firestore().collection('cities').doc();
+    return Promise.resolve().then(() => {
+      admin.firestore.setLogFunction((log) => {
+          logs.push(log);
+      })
+      return source.set({name: 'San Francisco'});
+    }).then(result => {
+      return source.delete();
+    }).then(result => {
+      utils.assert(
+          logs.length > 0,
+          'firestore.setLogFunction()',
+          'Log function did not update');
+    }).catch(err => {
+      utils.logFailure('firestore.setLogFunction()', 'Error while setting log function: ' + err);
+    });
+}
+
+return Promise.resolve()
     .then(testFirestore)
     .then(testFieldValue)
     .then(testFieldPath)
     .then(testGeoPoint)
-    .then(testSetDocumentReference);
+    .then(testSetDocumentReference)
+    .then(testSetLogFunction);
 }
 
 module.exports = {

test/unit/firebase-namespace.spec.ts

@@ -36,7 +36,8 @@ import {
   Firestore,
   FieldPath,
   FieldValue,
-  GeoPoint
+  GeoPoint,
+  setLogFunction,
 } from '@google-cloud/firestore';
 
 chai.should();
@@ -478,5 +479,9 @@ describe('FirebaseNamespace', () => {
     it('should return a reference to GeoPoint type', () => {
       expect(firebaseNamespace.firestore.GeoPoint).to.be.deep.equal(GeoPoint);
     });
+
+    it('should return a reference to setLogFunction', () => {
+      expect(firebaseNamespace.firestore.setLogFunction).to.be.deep.equal(setLogFunction);
+    });
   });
 });