Exposed the CollectionReference type from admin.firestore (#347)

* Exposed the CollectionReference type from admin.firestore

* Added WriteResult, WriteBatch and QueryDocumentSnapshot

* Updated changelog

* More integration tests

Author: hiranya911

CHANGELOG.md

@@ -1,6 +1,7 @@
 # Unreleased
 
--
+- [added] Exposed the `CollectionReference`, `WriteBatch`, `WriteResult` and
+  `QueryDocumentSnapshot` types from the `admin.firestore` namespace.
 
 # v6.0.0
 

src/index.d.ts

@@ -590,17 +590,22 @@ declare namespace admin.storage {
 }
 
 declare namespace admin.firestore {
+  export import CollectionReference = _firestore.CollectionReference;
+  export import DocumentData = _firestore.DocumentData;
   export import DocumentReference = _firestore.DocumentReference;
   export import DocumentSnapshot = _firestore.DocumentSnapshot;
   export import FieldPath = _firestore.FieldPath;
   export import FieldValue = _firestore.FieldValue;
   export import Firestore = _firestore.Firestore;
   export import GeoPoint = _firestore.GeoPoint;
-  export import setLogFunction = _firestore.setLogFunction;
   export import Query = _firestore.Query;
-  export import DocumentData = _firestore.DocumentData;
+  export import QueryDocumentSnapshot = _firestore.QueryDocumentSnapshot;
   export import QuerySnapshot = _firestore.QuerySnapshot;
   export import Timestamp = _firestore.Timestamp;
+  export import WriteBatch = _firestore.WriteBatch;
+  export import WriteResult = _firestore.WriteResult;
+
+  export import setLogFunction = _firestore.setLogFunction;
 }
 
 declare namespace admin.instanceId {

test/integration/firestore.spec.ts

@@ -18,7 +18,6 @@ import * as admin from '../../lib/index';
 import * as chai from 'chai';
 import * as chaiAsPromised from 'chai-as-promised';
 import {clone} from 'lodash';
-import {DocumentReference} from '@google-cloud/firestore';
 
 chai.should();
 chai.use(chaiAsPromised);
@@ -32,7 +31,7 @@ const mountainView = {
 
 describe('admin.firestore', () => {
 
-  let reference: DocumentReference;
+  let reference: admin.firestore.DocumentReference;
 
   before(() => {
     const db = admin.firestore();
@@ -84,6 +83,10 @@ describe('admin.firestore', () => {
       .should.eventually.be.fulfilled;
   }).timeout(5000);
 
+  it('admin.firestore.CollectionReference type is defined', () => {
+    expect(typeof admin.firestore.CollectionReference).to.be.not.undefined;
+  });
+
   it('admin.firestore.FieldPath type is defined', () => {
     expect(typeof admin.firestore.FieldPath).to.be.not.undefined;
   });
@@ -102,6 +105,14 @@ describe('admin.firestore', () => {
     expect(typeof now.nanoseconds).to.equal('number');
   });
 
+  it('admin.firestore.WriteBatch type is defined', () => {
+    expect(typeof admin.firestore.WriteBatch).to.be.not.undefined;
+  });
+
+  it('admin.firestore.WriteResult type is defined', () => {
+    expect(typeof admin.firestore.WriteResult).to.be.not.undefined;
+  });
+
   it('supports saving references in documents', () => {
     const source = admin.firestore().collection('cities').doc();
     const target = admin.firestore().collection('cities').doc();