From d12e4fdc81df9ba0c85026bcc9e2bca703ea90cb Mon Sep 17 00:00:00 2001
From: Juan Tejada <juans.tejada@gmail.com>
Date: Mon, 4 Jan 2016 13:23:24 -0500
Subject: [PATCH] fix(extensions): Add missing return values

---
 spec/extensions/composer-extension-adapter-spec.es6 | 8 ++++++--
 src/extensions/composer-extension-adapter.es6       | 4 ++--
 2 files changed, 8 insertions(+), 4 deletions(-)

diff --git a/spec/extensions/composer-extension-adapter-spec.es6 b/spec/extensions/composer-extension-adapter-spec.es6
index 84ac1dd97..7a2adee14 100644
--- a/spec/extensions/composer-extension-adapter-spec.es6
+++ b/spec/extensions/composer-extension-adapter-spec.es6
@@ -129,10 +129,12 @@ describe('ComposerExtensionAdapter', ()=> {
       const extension = {
         finalizeSessionBeforeSending(sess) {
           methodSpy(sess);
+          return 'result';
         },
       };
       adapter.adaptComposerMethod(extension, 'finalizeSessionBeforeSending');
-      extension.finalizeSessionBeforeSending({session});
+      const res = extension.finalizeSessionBeforeSending({session});
+      expect(res).toEqual('result');
       expect(methodSpy).toHaveBeenCalledWith(session);
     });
 
@@ -142,10 +144,12 @@ describe('ComposerExtensionAdapter', ()=> {
       const extension = {
         warningsForSending(dr) {
           methodSpy(dr);
+          return 'result';
         },
       };
       adapter.adaptComposerMethod(extension, 'warningsForSending');
-      extension.warningsForSending({draft});
+      const res = extension.warningsForSending({draft});
+      expect(res).toEqual('result');
       expect(methodSpy).toHaveBeenCalledWith(draft);
     });
   });
diff --git a/src/extensions/composer-extension-adapter.es6 b/src/extensions/composer-extension-adapter.es6
index 00bdcf83d..90e6123f4 100644
--- a/src/extensions/composer-extension-adapter.es6
+++ b/src/extensions/composer-extension-adapter.es6
@@ -29,11 +29,11 @@ export function adaptComposerMethod(extension, method) {
 
   if (method === 'finalizeSessionBeforeSending') {
     extension[method] = (argsObj)=> {
-      original(argsObj.session);
+      return original(argsObj.session);
     };
   } else {
     extension[method] = (argsObj)=> {
-      original(argsObj.draft);
+      return original(argsObj.draft);
     };
   }
 }