Add recursiveDelete & batch.create mocks (#186)

* Add mockBatchCreate

* Add mockBatchCreate type def

* Add mockBatchCreate test to full-setup.test.js

* Add mockBatchCreate test to full-setup-library-firestore.test.js

* Add mockRecursiveDelete, missing export of mockBatchCreate

* Add mockRecursiveDelete type def, missing create def from FirestoreBatch

* Add recursiveDelete test to full-setup.test.js

* Add recursiveDelete test to full-setup-library-firestore.test.js

* Fix tests full-setup-library-firestore.test.js

__tests__/full-setup-library-firestore.test.js

@@ -19,11 +19,13 @@ describe.each([
     mockBatchDelete,
     mockBatchUpdate,
     mockBatchSet,
+    mockBatchCreate,
     mockSettings,
     mockOnSnapShot,
     mockListCollections,
     mockTimestampNow,
     mockCreate,
+    mockRecursiveDelete,
   } = require('firestore-jest-mock/mocks/firestore');
 
   describe('we can start a firestore application', () => {
@@ -243,6 +245,10 @@ describe.each([
         const nycRef = firestore.collection('cities').doc('NYC');
         batch.set(nycRef, { name: 'New York City' });
 
+        // Create new city 'CHI'
+	      const chiRef = firestore.collection('cities').doc('CHI');
+	      batch.create(chiRef, { name: 'Chicago', state: 'IL', country: 'USA' });
+
         // Update the population of 'SF'
         const sfRef = firestore.collection('cities').doc('SF');
         batch.update(sfRef, { population: 1000000 });
@@ -257,6 +263,7 @@ describe.each([
           expect(mockBatchDelete).toHaveBeenCalledWith(laRef);
           expect(mockBatchUpdate).toHaveBeenCalledWith(sfRef, { population: 1000000 });
           expect(mockBatchSet).toHaveBeenCalledWith(nycRef, { name: 'New York City' });
+          expect(mockBatchCreate).toHaveBeenCalledWith(chiRef, { name: 'Chicago', state: 'IL', country: 'USA' });
           expect(mockBatchCommit).toHaveBeenCalled();
         });
       });
@@ -375,6 +382,15 @@ describe.each([
         expect(mockWhere).toHaveBeenCalled();
         expect(mockOnSnapShot).toHaveBeenCalled();
       });
+
+      test('recursiveDelete', async () => {
+        const firestore = new this.Firestore();
+	      const citiesRef = firestore.collection('cities');
+
+	      firestore.recursiveDelete(citiesRef)
+
+        expect(mockRecursiveDelete).toHaveBeenCalledWith(citiesRef);
+      });
     });
   });
 });

__tests__/full-setup.test.js

@@ -25,6 +25,7 @@ describe.each`
     mockBatchDelete,
     mockBatchUpdate,
     mockBatchSet,
+    mockBatchCreate,
     mockSettings,
     mockOnSnapShot,
     mockUseEmulator,
@@ -34,6 +35,7 @@ describe.each`
     FakeFirestore,
     mockQueryOnSnapshot,
     mockTimestampNow,
+    mockRecursiveDelete,
   } = require('firestore-jest-mock/mocks/firestore');
 
   mockFirebase(
@@ -292,6 +294,10 @@ describe.each`
       const nycRef = db.collection('cities').doc('NYC');
       batch.set(nycRef, { name: 'New York City' });
 
+      // Create new city 'CHI'
+	    const chiRef = db.collection('cities').doc('CHI');
+	    batch.create(chiRef, { name: 'Chicago', state: 'IL', country: 'USA' });
+
       // Update the population of 'SF'
       const sfRef = db.collection('cities').doc('SF');
       batch.update(sfRef, { population: 1000000 });
@@ -307,6 +313,7 @@ describe.each`
         expect(mockBatchDelete).toHaveBeenCalledWith(laRef);
         expect(mockBatchUpdate).toHaveBeenCalledWith(sfRef, { population: 1000000 });
         expect(mockBatchSet).toHaveBeenCalledWith(nycRef, { name: 'New York City' });
+        expect(mockBatchCreate).toHaveBeenCalledWith(chiRef, { name: 'Chicago', state: 'IL', country: 'USA' });
         expect(mockBatchCommit).toHaveBeenCalled();
       });
     });
@@ -410,6 +417,15 @@ describe.each`
       expect(mockQueryOnSnapshot).toHaveBeenCalled();
     });
 
+    test('recursiveDelete', async () => {
+      const db = firebase.firestore();
+	    const citiesRef = db.collection('cities');
+
+	    db.recursiveDelete(citiesRef)
+
+      expect(mockRecursiveDelete).toHaveBeenCalledWith(citiesRef);
+    });
+
     describe('withConverter', () => {
       const converter = {
         fromFirestore: () => {},

src/mocks/firestore.d.ts

@@ -24,6 +24,7 @@ interface FirestoreBatch {
   delete(): FirestoreBatch;
   set(doc: DocumentReference, data: DocumentData, options?: SetOptions): FirestoreBatch;
   update(doc: DocumentReference, data: DocumentData): FirestoreBatch;
+  create(doc: DocumentReference, data: DocumentData): FirestoreBatch;
   commit(): Promise<void>;
 }
 
@@ -54,6 +55,7 @@ export class FakeFirestore {
   collectionGroup(collectionName: string): Query;
   doc(path: string): DocumentReference;
   runTransaction<T>(updateFunction: (transaction: Transaction) => Promise<T>): Promise<T>;
+  recursiveDelete(ref: DocumentReference|CollectionReference): Promise<void>;
 }
 
 declare class DocumentReference {
@@ -118,6 +120,7 @@ declare class CollectionReference extends FakeFirestore.Query {
 // Mocks exported from this module
 export const mockBatch: jest.Mock;
 export const mockRunTransaction: jest.Mock;
+export const mockRecursiveDelete: jest.Mock;
 
 export const mockCollection: jest.Mock;
 export const mockCollectionGroup: jest.Mock;
@@ -137,6 +140,7 @@ export const mockBatchDelete: jest.Mock;
 export const mockBatchCommit: jest.Mock;
 export const mockBatchUpdate: jest.Mock;
 export const mockBatchSet: jest.Mock;
+export const mockBatchCreate: jest.Mock;
 
 export const mockOnSnapShot: jest.Mock;
 

src/mocks/firestore.js

@@ -1,6 +1,7 @@
 const mockCollectionGroup = jest.fn();
 const mockBatch = jest.fn();
 const mockRunTransaction = jest.fn();
+const mockRecursiveDelete = jest.fn();
 
 const mockSettings = jest.fn();
 const mockUseEmulator = jest.fn();
@@ -18,6 +19,7 @@ const mockBatchDelete = jest.fn();
 const mockBatchCommit = jest.fn();
 const mockBatchUpdate = jest.fn();
 const mockBatchSet = jest.fn();
+const mockBatchCreate = jest.fn();
 
 const mockOnSnapShot = jest.fn();
 
@@ -73,6 +75,11 @@ class FakeFirestore {
         this._ref._updateData(doc.path, data, true);
         return this;
       },
+      create(doc, data) {
+        mockBatchCreate(...arguments);
+        this._ref._updateData(doc.path, data, false);
+        return this;
+      },
       commit() {
         mockBatchCommit(...arguments);
         return Promise.resolve([]);
@@ -224,6 +231,11 @@ class FakeFirestore {
       id: docId,
     };
   }
+
+  recursiveDelete(ref, bulkWriter) {
+    mockRecursiveDelete(...arguments);
+  	return Promise.resolve();
+  }
 }
 
 FakeFirestore.Query = query.Query;
@@ -544,6 +556,7 @@ module.exports = {
   FakeFirestore,
   mockBatch,
   mockRunTransaction,
+  mockRecursiveDelete,
   mockCollection,
   mockCollectionGroup,
   mockDoc,
@@ -558,6 +571,7 @@ module.exports = {
   mockBatchCommit,
   mockBatchUpdate,
   mockBatchSet,
+  mockBatchCreate,
   mockOnSnapShot,
   mockListDocuments,
   mockListCollections,