test(GeoFirestore): add tests for runTransaction

package.json

@@ -84,9 +84,6 @@
     "include": [
       "src/**/*.ts"
     ],
-    "exclude": [
-      "src/interfaces/*"
-    ],
     "reporter": [
       "html"
     ],

src/index.ts

@@ -5,4 +5,5 @@ export * from './GeoFirestore';
 export * from './GeoFirestoreTypes';
 export * from './GeoQuery';
 export * from './GeoQuerySnapshot';
+export * from './GeoTransaction';
 export * from './GeoWriteBatch';

test/GeoFirestore.test.ts

@@ -1,7 +1,10 @@
 import * as chai from 'chai';
 
-import { GeoFirestore } from '../src/GeoFirestore';
-import { afterEachHelper, beforeEachHelper, firestore, invalidFirestores, testCollectionName } from './common';
+import { GeoFirestore, GeoTransaction } from '../src';
+import {
+  afterEachHelper, beforeEachHelper, firestore, invalidFirestores,
+  testCollectionName, geofirestore, geocollection
+} from './common';
 
 const expect = chai.expect;
 
@@ -41,4 +44,21 @@ describe('GeoFirestore Tests:', () => {
       ).to.deep.equal(firestore.collection(testCollectionName));
     });
   });
+
+  describe('runTransaction():', () => {
+    it('runTransaction() doesn\'t throw an error when a valid `updateFunction` is passed in', () => {
+      expect(() => {
+        return geofirestore.runTransaction((transaction) => {
+          const geotransaction = new GeoTransaction(transaction);
+          const docRef = geocollection.doc('testdoc');
+          return geotransaction.get(docRef);
+        });
+      }).to.not.throw();
+    });
+
+    it('runTransaction() does throw an error when an invalid `updateFunction` is passed in', (done) => {
+      // @ts-ignore
+      geofirestore.runTransaction(() => Math).catch((e) => done());
+    });
+  });
 });