Add .run method to HTTPS callable functions (#206)

Author: laurenzlong

spec/providers/https.spec.ts

@@ -519,6 +519,22 @@ describe('callable.FunctionBuilder', () => {
       expect(fn.__trigger.availableMemoryMb).to.deep.equal(256);
       expect(fn.__trigger.timeout).to.deep.equal('90s');
     });
+
+    it('has a .run method', () => {
+      const cf = https.onCall((data, context) => {
+        return { data, context };
+      });
+
+      const data = 'data';
+      const context = {
+        instanceIdToken: 'token',
+        auth: {
+          uid: 'abc',
+          token: 'token',
+        },
+      };
+      expect(cf.run(data, context)).to.deep.equal({ data, context });
+    });
   });
 });
 

src/cloud-functions.ts

@@ -180,7 +180,7 @@ export interface TriggerAnnotated {
 
 /** A Runnable has a `run` method which directly invokes the user-defined function - useful for unit testing. */
 export interface Runnable<T> {
-  run: (data: T, context: EventContext) => PromiseLike<any> | any;
+  run: (data: T, context: any) => PromiseLike<any> | any;
 }
 
 /**

src/providers/https.ts

@@ -25,7 +25,7 @@ import * as firebase from 'firebase-admin';
 import * as _ from 'lodash';
 import * as cors from 'cors';
 import { apps } from '../apps';
-import { HttpsFunction, optsToTrigger } from '../cloud-functions';
+import { HttpsFunction, optsToTrigger, Runnable } from '../cloud-functions';
 import { DeploymentOptions } from '../function-builder';
 
 /**
@@ -415,7 +415,7 @@ const corsHandler = cors({ origin: true, methods: 'POST' });
 export function _onCallWithOpts(
   handler: (data: any, context: CallableContext) => any | Promise<any>,
   opts: DeploymentOptions
-): HttpsFunction {
+): HttpsFunction & Runnable<any> {
   const func = async (req: express.Request, res: express.Response) => {
     try {
       if (!isValidRequest(req)) {
@@ -486,5 +486,7 @@ export function _onCallWithOpts(
     labels: { 'deployment-callable': 'true' },
   });
 
+  corsFunc.run = handler;
+
   return corsFunc;
 }