Lazily load modules that require firebase-functions (#3)

Author: laurenzlong

package.json

@@ -22,7 +22,7 @@
     "@types/chai": "^4.1.2",
     "@types/mocha": "^5.0.0",
     "chai": "^4.1.2",
-    "firebase-functions": "https://storage.googleapis.com/functions-dev-preview/firebase-functions-1.0.0-rc4.tgz",
+    "firebase-functions": "https://storage.googleapis.com/functions-dev-preview/firebase-functions-1.0.0-rc5.tgz",
     "mocha": "^5.0.5",
     "sinon": "^4.4.9",
     "tslint": "^5.8.0",

src/features.ts

@@ -0,0 +1,31 @@
+import { makeChange, wrap, mockConfig } from './main';
+import * as analytics from './providers/analytics';
+import * as crashlytics from './providers/crashlytics';
+import * as database from './providers/database';
+import * as firestore from './providers/firestore';
+import * as pubsub from './providers/pubsub';
+import * as storage from './providers/storage';
+
+export interface FeaturesList {
+  mockConfig: typeof mockConfig;
+  wrap: typeof wrap;
+  makeChange: typeof makeChange;
+  analytics: typeof analytics;
+  crashlytics: typeof crashlytics;
+  database: typeof database;
+  firestore: typeof firestore;
+  pubsub: typeof pubsub;
+  storage: typeof storage;
+}
+
+export const features: FeaturesList = {
+  mockConfig,
+  wrap,
+  makeChange,
+  analytics,
+  crashlytics,
+  database,
+  firestore,
+  pubsub,
+  storage,
+};

src/index.ts

@@ -21,29 +21,19 @@
 // SOFTWARE.
 
 import { AppOptions } from 'firebase-admin';
+import { merge } from 'lodash';
 
 import { FirebaseFunctionsTest } from './lifecycle';
-import { makeChange, wrap, mockConfig } from './main';
-import * as analytics from './providers/analytics';
-import * as crashlytics from './providers/crashlytics';
-import * as database from './providers/database';
-import * as firestore from './providers/firestore';
-import * as pubsub from './providers/pubsub';
-import * as storage from './providers/storage';
+import { features as lazyFeatures, FeaturesList } from './features';
 
 export = (firebaseConfig?: AppOptions) => {
   const test = new FirebaseFunctionsTest();
   test.init(firebaseConfig);
-  return {
-    cleanup: test.cleanup,
-    mockConfig,
-    wrap,
-    makeChange,
-    analytics,
-    crashlytics,
-    database,
-    firestore,
-    pubsub,
-    storage,
-  };
+  // Ensure other files get loaded after init function, since they load `firebase-functions`
+  // which will issue warning if process.env.FIREBASE_CONFIG is not yet set.
+  let features = require('./features') as typeof lazyFeatures;
+  features = merge({}, features, {
+    cleanup: () => test.cleanup,
+  });
+  return features;
 };