From a18d54b7064d6bb0fb2eec23da5d6453329cb92a Mon Sep 17 00:00:00 2001
From: Sojan Jose <sojan@pepalo.com>
Date: Fri, 17 Jul 2020 00:32:32 +0530
Subject: [PATCH] Chore: Cleanup attachment handling for Facebook & Whatsapp
 (#1051)

* Chore: Enable file upload for facebook messenger
* Chore: Fix attachments
* Chore: Fix Specs
* Fix ReplyBox file attachment logic
* Set default value for message

Co-authored-by: Pranav Raj Sreepuram <pranavrajs@gmail.com>
---
 .../widgets/conversation/ReplyBox.vue         | 178 ++++++++++--------
 .../store/modules/conversations/index.js      |   4 +
 .../shared/helpers/KeyboardHelpers.js         |  11 ++
 .../shared/helpers/MessageFormatter.js        |   2 +-
 .../shared/helpers/MessageTypeHelper.js       |   7 +
 .../helpers/specs/KeyboardHelpers.spec.js     |  21 +++
 app/jobs/send_reply_job.rb                    |  15 ++
 app/models/message.rb                         |   9 +-
 .../facebook/send_on_facebook_service.rb      |  11 +-
 docs/channels/supported-features.md           |   4 +-
 spec/factories/messages.rb                    |   2 +-
 .../facebook/send_on_facebook_service_spec.rb |  19 +-
 .../twilio/send_on_twilio_service_spec.rb     |  17 +-
 .../twitter/send_on_twitter_service_spec.rb   |  20 +-
 14 files changed, 212 insertions(+), 108 deletions(-)
 create mode 100644 app/javascript/shared/helpers/KeyboardHelpers.js
 create mode 100644 app/javascript/shared/helpers/specs/KeyboardHelpers.spec.js
 create mode 100644 app/jobs/send_reply_job.rb

diff --git a/app/javascript/dashboard/components/widgets/conversation/ReplyBox.vue b/app/javascript/dashboard/components/widgets/conversation/ReplyBox.vue
index 499394184..11fa825cf 100644
--- a/app/javascript/dashboard/components/widgets/conversation/ReplyBox.vue
+++ b/app/javascript/dashboard/components/widgets/conversation/ReplyBox.vue
@@ -15,9 +15,9 @@
       />
       <resizable-text-area
         ref="messageInput"
-        v-model="message"
+        v-model.trim="message"
         class="input"
-        :placeholder="$t(messagePlaceHolder())"
+        :placeholder="messagePlaceHolder"
         :min-height="4"
         @focus="onFocus"
         @blur="onBlur"
@@ -25,46 +25,43 @@
       <file-upload
         v-if="showFileUpload"
         :size="4096 * 4096"
-        accept="jpg,jpeg,png,mp3,ogg,amr,pdf,mp4"
+        accept="image/*, application/pdf, audio/mpeg, video/mp4, audio/ogg"
         @input-file="onFileUpload"
       >
-        <i
-          v-if="!isUploading.image"
-          class="icon ion-android-attach attachment"
-        />
-        <woot-spinner v-if="isUploading.image" />
+        <i v-if="!isUploading" class="icon ion-android-attach attachment" />
+        <woot-spinner v-if="isUploading" />
       </file-upload>
       <i
         class="icon ion-happy-outline"
         :class="{ active: showEmojiPicker }"
-        @click="toggleEmojiPicker()"
+        @click="toggleEmojiPicker"
       />
     </div>
 
     <div class="reply-box__bottom">
       <ul class="tabs">
         <li class="tabs-title" :class="{ 'is-active': !isPrivate }">
-          <a href="#" @click="makeReply">{{
+          <a href="#" @click="setReplyMode">{{
             $t('CONVERSATION.REPLYBOX.REPLY')
           }}</a>
         </li>
         <li class="tabs-title is-private" :class="{ 'is-active': isPrivate }">
-          <a href="#" @click="makePrivate">{{
+          <a href="#" @click="setPrivateReplyMode">{{
             $t('CONVERSATION.REPLYBOX.PRIVATE_NOTE')
           }}</a>
         </li>
         <li v-if="message.length" class="tabs-title message-length">
-          <a :class="{ 'message-error': message.length > maxLength - 40 }">
-            {{ message.length }} / {{ maxLength }}
-          </a>
+          <a :class="{ 'message-error': isMessageLengthReachingThreshold }">{{
+            characterCountIndicator
+          }}</a>
         </li>
       </ul>
       <button
         type="button"
         class="button send-button"
-        :disabled="disableButton()"
+        :disabled="isReplyButtonDisabled"
         :class="{
-          disabled: message.length === 0 || message.length > maxLength,
+          disabled: isReplyButtonDisabled,
           warning: isPrivate,
         }"
         @click="sendMessage"
@@ -93,6 +90,12 @@ import FileUpload from 'vue-upload-component';
 import EmojiInput from '../emoji/EmojiInput';
 import CannedResponse from './CannedResponse';
 import ResizableTextArea from 'shared/components/ResizableTextArea';
+import {
+  isEscape,
+  isEnter,
+  hasPressedShift,
+} from 'shared/helpers/KeyboardHelpers';
+import { MESSAGE_MAX_LENGTH } from 'shared/helpers/MessageTypeHelper';
 
 export default {
   components: {
@@ -109,19 +112,38 @@ export default {
       isFocused: false,
       showEmojiPicker: false,
       showCannedResponsesList: false,
-      isUploading: {
-        audio: false,
-        video: false,
-        image: false,
-      },
+      isUploading: false,
     };
   },
   computed: {
-    ...mapGetters({
-      currentChat: 'getSelectedChat',
-    }),
+    ...mapGetters({ currentChat: 'getSelectedChat' }),
+    inboxId() {
+      return this.currentChat.inbox_id;
+    },
+    inbox() {
+      return this.$store.getters['inboxes/getInbox'](this.inboxId);
+    },
+    messagePlaceHolder() {
+      return this.isPrivate
+        ? this.$t('CONVERSATION.FOOTER.PRIVATE_MSG_INPUT')
+        : this.$t('CONVERSATION.FOOTER.MSG_INPUT');
+    },
+    isMessageLengthReachingThreshold() {
+      return this.message.length > this.maxLength - 40;
+    },
+    characterCountIndicator() {
+      return `${this.message.length} / ${this.maxLength}`;
+    },
+    isReplyButtonDisabled() {
+      const isMessageEmpty = !this.message.replace(/\n/g, '').length;
+      return (
+        isMessageEmpty ||
+        this.message.length === 0 ||
+        this.message.length > this.maxLength
+      );
+    },
     channelType() {
-      return this.currentChat.meta.channel;
+      return this.inbox.channel_type;
     },
     conversationType() {
       const { additional_attributes: additionalAttributes } = this.currentChat;
@@ -129,18 +151,52 @@ export default {
       return type || '';
     },
     maxLength() {
-      if (this.channelType === 'Channel::FacebookPage') {
-        return 640;
+      if (this.isPrivate) {
+        return MESSAGE_MAX_LENGTH.GENERAL;
       }
-      if (this.channelType === 'Channel::TwitterProfile') {
+
+      if (this.isAFacebookInbox) {
+        return MESSAGE_MAX_LENGTH.FACEBOOK;
+      }
+      if (this.isATwilioSMSChannel) {
+        return MESSAGE_MAX_LENGTH.TWILIO_SMS;
+      }
+      if (this.isATwitterInbox) {
         if (this.conversationType === 'tweet') {
-          return 280;
+          return MESSAGE_MAX_LENGTH.TWEET;
         }
       }
-      return 10000;
+      return MESSAGE_MAX_LENGTH.GENERAL;
+    },
+    isATwitterInbox() {
+      return this.channelType === 'Channel::TwitterProfile';
+    },
+    isAFacebookInbox() {
+      return this.channelType === 'Channel::FacebookPage';
+    },
+    isAWebWidgetInbox() {
+      return this.channelType === 'Channel::WebWidget';
+    },
+    isATwilioSMSChannel() {
+      const { phone_number: phoneNumber = '' } = this.inbox;
+      return (
+        this.channelType === 'Channel::TwilioSms' &&
+        !phoneNumber.startsWith('whatsapp')
+      );
+    },
+    isATwilioWhatsappChannel() {
+      const { phone_number: phoneNumber = '' } = this.inbox;
+      return (
+        this.channelType === 'Channel::TwilioSms' &&
+        phoneNumber.startsWith('whatsapp')
+      );
     },
     showFileUpload() {
-      return this.channelType === 'Channel::WebWidget';
+      return (
+        this.isAWebWidgetInbox ||
+        this.isAFacebookInbox ||
+        this.isATwilioWhatsappChannel
+      );
     },
     replyButtonLabel() {
       if (this.isPrivate) {
@@ -158,20 +214,18 @@ export default {
     },
   },
   watch: {
-    message(val) {
+    message(updatedMessage) {
       if (this.isPrivate) {
         return;
       }
-      const isSlashCommand = val[0] === '/';
-      const hasNextWord = val.includes(' ');
+      const isSlashCommand = updatedMessage[0] === '/';
+      const hasNextWord = updatedMessage.includes(' ');
       const isShortCodeActive = isSlashCommand && !hasNextWord;
       if (isShortCodeActive) {
         this.showCannedResponsesList = true;
-        if (val.length > 1) {
-          const searchKey = val.substr(1, val.length);
-          this.$store.dispatch('getCannedResponse', {
-            searchKey,
-          });
+        if (updatedMessage.length > 1) {
+          const searchKey = updatedMessage.substr(1, updatedMessage.length);
+          this.$store.dispatch('getCannedResponse', { searchKey });
         } else {
           this.$store.dispatch('getCannedResponse');
         }
@@ -188,26 +242,18 @@ export default {
   },
   methods: {
     handleKeyEvents(e) {
-      if (this.isEscape(e)) {
+      if (isEscape(e)) {
         this.hideEmojiPicker();
         this.hideCannedResponse();
-      } else if (this.isEnter(e)) {
-        if (!e.shiftKey) {
+      } else if (isEnter(e)) {
+        if (!hasPressedShift(e)) {
           e.preventDefault();
           this.sendMessage();
         }
       }
     },
-    isEnter(e) {
-      return e.keyCode === 13;
-    },
-    isEscape(e) {
-      return e.keyCode === 27; // ESCAPE
-    },
     async sendMessage() {
-      const isMessageEmpty = !this.message.replace(/\n/g, '').length;
-      if (isMessageEmpty) return;
-      if (this.message.length > this.maxLength) {
+      if (this.isReplyButtonDisabled) {
         return;
       }
       const newMessage = this.message;
@@ -231,11 +277,11 @@ export default {
         this.message = message;
       }, 100);
     },
-    makePrivate() {
+    setPrivateReplyMode() {
       this.isPrivate = true;
       this.$refs.messageInput.focus();
     },
-    makeReply() {
+    setReplyMode() {
       this.isPrivate = false;
       this.$refs.messageInput.focus();
     },
@@ -258,7 +304,6 @@ export default {
     hideCannedResponse() {
       this.showCannedResponsesList = false;
     },
-
     onBlur() {
       this.isFocused = false;
       this.toggleTyping('off');
@@ -267,9 +312,8 @@ export default {
       this.isFocused = true;
       this.toggleTyping('on');
     },
-
     toggleTyping(status) {
-      if (this.channelType === 'Channel::WebWidget' && !this.isPrivate) {
+      if (this.isAWebWidgetInbox && !this.isPrivate) {
         const conversationId = this.currentChat.id;
         this.$store.dispatch('conversationTypingStatus/toggleTyping', {
           status,
@@ -277,35 +321,19 @@ export default {
         });
       }
     },
-    disableButton() {
-      const messageHasOnlyNewLines = !this.message.replace(/\n/g, '').length;
-      return (
-        this.message.length === 0 ||
-        this.message.length > 640 ||
-        messageHasOnlyNewLines
-      );
-    },
-
-    messagePlaceHolder() {
-      const placeHolder = this.isPrivate
-        ? 'CONVERSATION.FOOTER.PRIVATE_MSG_INPUT'
-        : 'CONVERSATION.FOOTER.MSG_INPUT';
-      return placeHolder;
-    },
-
     onFileUpload(file) {
       if (!file) {
         return;
       }
-      this.isUploading.image = true;
+      this.isUploading = true;
       this.$store
         .dispatch('sendAttachment', [this.currentChat.id, { file: file.file }])
         .then(() => {
-          this.isUploading.image = false;
+          this.isUploading = false;
           this.$emit('scrollToMessage');
         })
         .catch(() => {
-          this.isUploading.image = false;
+          this.isUploading = false;
           this.$emit('scrollToMessage');
         });
     },
diff --git a/app/javascript/dashboard/store/modules/conversations/index.js b/app/javascript/dashboard/store/modules/conversations/index.js
index af6a77307..ef978383b 100644
--- a/app/javascript/dashboard/store/modules/conversations/index.js
+++ b/app/javascript/dashboard/store/modules/conversations/index.js
@@ -12,6 +12,10 @@ const initialSelectedChat = {
   status: null,
   muted: false,
   seen: false,
+  inbox_id: null,
+  additional_attributes: {
+    type: '',
+  },
   dataFetched: false,
 };
 const state = {
diff --git a/app/javascript/shared/helpers/KeyboardHelpers.js b/app/javascript/shared/helpers/KeyboardHelpers.js
new file mode 100644
index 000000000..38bf62321
--- /dev/null
+++ b/app/javascript/shared/helpers/KeyboardHelpers.js
@@ -0,0 +1,11 @@
+export const isEnter = e => {
+  return e.keyCode === 13;
+};
+
+export const isEscape = e => {
+  return e.keyCode === 27;
+};
+
+export const hasPressedShift = e => {
+  return e.shiftKey;
+};
diff --git a/app/javascript/shared/helpers/MessageFormatter.js b/app/javascript/shared/helpers/MessageFormatter.js
index a5dc591e8..a31d9c8e8 100644
--- a/app/javascript/shared/helpers/MessageFormatter.js
+++ b/app/javascript/shared/helpers/MessageFormatter.js
@@ -2,7 +2,7 @@ import { escapeHtml } from './HTMLSanitizer';
 
 class MessageFormatter {
   constructor(message) {
-    this.message = escapeHtml(message) || '';
+    this.message = escapeHtml(message || '') || '';
   }
 
   formatMessage() {
diff --git a/app/javascript/shared/helpers/MessageTypeHelper.js b/app/javascript/shared/helpers/MessageTypeHelper.js
index 67a5d366e..7dc4457b1 100644
--- a/app/javascript/shared/helpers/MessageTypeHelper.js
+++ b/app/javascript/shared/helpers/MessageTypeHelper.js
@@ -1,3 +1,10 @@
 export const isAFormMessage = message => message.content_type === 'form';
 export const isASubmittedFormMessage = (message = {}) =>
   isAFormMessage(message) && !!message.content_attributes?.submitted_values;
+
+export const MESSAGE_MAX_LENGTH = {
+  GENERAL: 10000,
+  FACEBOOK: 640,
+  TWILIO_SMS: 160,
+  TWEET: 280,
+};
diff --git a/app/javascript/shared/helpers/specs/KeyboardHelpers.spec.js b/app/javascript/shared/helpers/specs/KeyboardHelpers.spec.js
new file mode 100644
index 000000000..851c7dfef
--- /dev/null
+++ b/app/javascript/shared/helpers/specs/KeyboardHelpers.spec.js
@@ -0,0 +1,21 @@
+import { isEnter, isEscape, hasPressedShift } from '../KeyboardHelpers';
+
+describe('#KeyboardHelpers', () => {
+  describe('#isEnter', () => {
+    it('return correct values', () => {
+      expect(isEnter({ keyCode: 13 })).toEqual(true);
+    });
+  });
+
+  describe('#isEscape', () => {
+    it('return correct values', () => {
+      expect(isEscape({ keyCode: 27 })).toEqual(true);
+    });
+  });
+
+  describe('#hasPressedShift', () => {
+    it('return correct values', () => {
+      expect(hasPressedShift({ shiftKey: true })).toEqual(true);
+    });
+  });
+});
diff --git a/app/jobs/send_reply_job.rb b/app/jobs/send_reply_job.rb
new file mode 100644
index 000000000..4f1b6f839
--- /dev/null
+++ b/app/jobs/send_reply_job.rb
@@ -0,0 +1,15 @@
+class SendReplyJob < ApplicationJob
+  queue_as :high
+
+  def perform(message_id)
+    message = Message.find(message_id)
+    channel_name = message.conversation.inbox.channel.class.to_s
+    if channel_name == 'Channel::FacebookPage'
+      ::Facebook::SendOnFacebookService.new(message: message).perform
+    elsif channel_name == 'Channel::TwitterProfile'
+      ::Twitter::SendOnTwitterService.new(message: message).perform
+    elsif channel_name == 'Channel::TwilioSms'
+      ::Twilio::SendOnTwilioService.new(message: message).perform
+    end
+  end
+end
diff --git a/app/models/message.rb b/app/models/message.rb
index 7b368fe93..e5f148f4b 100644
--- a/app/models/message.rb
+++ b/app/models/message.rb
@@ -134,14 +134,7 @@ class Message < ApplicationRecord
   end
 
   def send_reply
-    channel_name = conversation.inbox.channel.class.to_s
-    if channel_name == 'Channel::FacebookPage'
-      ::Facebook::SendOnFacebookService.new(message: self).perform
-    elsif channel_name == 'Channel::TwitterProfile'
-      ::Twitter::SendOnTwitterService.new(message: self).perform
-    elsif channel_name == 'Channel::TwilioSms'
-      ::Twilio::SendOnTwilioService.new(message: self).perform
-    end
+    ::SendReplyJob.perform_later(id)
   end
 
   def reopen_conversation
diff --git a/app/services/facebook/send_on_facebook_service.rb b/app/services/facebook/send_on_facebook_service.rb
index d27dfcf4d..629e680a6 100644
--- a/app/services/facebook/send_on_facebook_service.rb
+++ b/app/services/facebook/send_on_facebook_service.rb
@@ -17,19 +17,26 @@ class Facebook::SendOnFacebookService < Base::SendOnChannelService
   end
 
   def fb_attachment_message_params
+    attachment = message.attachments.first
     {
       recipient: { id: contact.get_source_id(inbox.id) },
       message: {
         attachment: {
-          type: 'image',
+          type: attachment_type(attachment),
           payload: {
-            url: message.attachments.first.file_url
+            url: attachment.file_url
           }
         }
       }
     }
   end
 
+  def attachment_type(attachment)
+    return attachment.file_type if %w[image audio video file].include? attachment.file_type
+
+    'file'
+  end
+
   def fb_message_params
     if message.attachments.blank?
       fb_text_message_params
diff --git a/docs/channels/supported-features.md b/docs/channels/supported-features.md
index 9175994ab..d8f03d7a1 100644
--- a/docs/channels/supported-features.md
+++ b/docs/channels/supported-features.md
@@ -56,10 +56,10 @@ title: "Supported features on channels"
 | Channel | Channel greeting | Attachments | Agent Auto assignment | Slack |
 | -- | -- | -- | -- | -- |
 | Website | Yes | Yes | Yes | Yes |
-| Facebook | Yes | No | Yes | No |
+| Facebook | Yes | Yes | Yes | No |
 | Twitter DM | Yes | No | Yes | No |
 | Tweets | Yes | No | Yes | No |
 | SMS | Yes | No | Yes | No |
-| Whatsapp | Yes | No | Yes | No |
+| Whatsapp | Yes | Yes | Yes | No |
 
 </div>
diff --git a/spec/factories/messages.rb b/spec/factories/messages.rb
index 264e6cae9..dee821a4f 100644
--- a/spec/factories/messages.rb
+++ b/spec/factories/messages.rb
@@ -10,8 +10,8 @@ FactoryBot.define do
 
     after(:build) do |message|
       message.sender ||= create(:user, account: message.account)
-      message.conversation ||= create(:conversation, account: message.account)
       message.inbox ||= create(:inbox, account: message.account)
+      message.conversation ||= create(:conversation, account: message.account, inbox: message.inbox)
     end
   end
 end
diff --git a/spec/services/facebook/send_on_facebook_service_spec.rb b/spec/services/facebook/send_on_facebook_service_spec.rb
index 742044abd..25a84f9e7 100644
--- a/spec/services/facebook/send_on_facebook_service_spec.rb
+++ b/spec/services/facebook/send_on_facebook_service_spec.rb
@@ -6,6 +6,7 @@ describe Facebook::SendOnFacebookService do
   before do
     allow(Facebook::Messenger::Subscriptions).to receive(:subscribe).and_return(true)
     allow(bot).to receive(:deliver)
+    create(:message, message_type: :incoming, inbox: facebook_inbox, account: account, conversation: conversation)
   end
 
   let!(:account) { create(:account) }
@@ -20,39 +21,43 @@ describe Facebook::SendOnFacebookService do
   describe '#perform' do
     context 'without reply' do
       it 'if message is private' do
-        create(:message, message_type: 'outgoing', private: true, inbox: facebook_inbox, account: account)
+        message = create(:message, message_type: 'outgoing', private: true, inbox: facebook_inbox, account: account)
+        ::Facebook::SendOnFacebookService.new(message: message).perform
         expect(bot).not_to have_received(:deliver)
       end
 
       it 'if inbox channel is not facebook page' do
-        create(:message, message_type: 'outgoing', inbox: widget_inbox, account: account)
+        message = create(:message, message_type: 'outgoing', inbox: widget_inbox, account: account)
+        expect { ::Facebook::SendOnFacebookService.new(message: message).perform }.to raise_error 'Invalid channel service was called'
         expect(bot).not_to have_received(:deliver)
       end
 
       it 'if message is not outgoing' do
-        create(:message, message_type: 'incoming', inbox: facebook_inbox, account: account)
+        message = create(:message, message_type: 'incoming', inbox: facebook_inbox, account: account)
+        ::Facebook::SendOnFacebookService.new(message: message).perform
         expect(bot).not_to have_received(:deliver)
       end
 
       it 'if message has an FB ID' do
-        create(:message, message_type: 'outgoing', inbox: facebook_inbox, account: account, source_id: SecureRandom.uuid)
+        message = create(:message, message_type: 'outgoing', inbox: facebook_inbox, account: account, source_id: SecureRandom.uuid)
+        ::Facebook::SendOnFacebookService.new(message: message).perform
         expect(bot).not_to have_received(:deliver)
       end
     end
 
     context 'with reply' do
       it 'if message is sent from chatwoot and is outgoing' do
-        create(:message, message_type: :incoming, inbox: facebook_inbox, account: account, conversation: conversation)
-        create(:message, message_type: 'outgoing', inbox: facebook_inbox, account: account, conversation: conversation)
+        message = create(:message, message_type: 'outgoing', inbox: facebook_inbox, account: account, conversation: conversation)
+        ::Facebook::SendOnFacebookService.new(message: message).perform
         expect(bot).to have_received(:deliver)
       end
 
       it 'if message with attachment is sent from chatwoot and is outgoing' do
-        create(:message, message_type: :incoming, inbox: facebook_inbox, account: account, conversation: conversation)
         message = build(:message, message_type: 'outgoing', inbox: facebook_inbox, account: account, conversation: conversation)
         attachment = message.attachments.new(account_id: message.account_id, file_type: :image)
         attachment.file.attach(io: File.open(Rails.root.join('spec/assets/avatar.png')), filename: 'avatar.png', content_type: 'image/png')
         message.save!
+        ::Facebook::SendOnFacebookService.new(message: message).perform
         expect(bot).to have_received(:deliver)
       end
     end
diff --git a/spec/services/twilio/send_on_twilio_service_spec.rb b/spec/services/twilio/send_on_twilio_service_spec.rb
index 56a8283c3..3f114a00b 100644
--- a/spec/services/twilio/send_on_twilio_service_spec.rb
+++ b/spec/services/twilio/send_on_twilio_service_spec.rb
@@ -25,22 +25,26 @@ describe Twilio::SendOnTwilioService do
   describe '#perform' do
     context 'without reply' do
       it 'if message is private' do
-        create(:message, message_type: 'outgoing', private: true, inbox: twilio_inbox, account: account)
+        message = create(:message, message_type: 'outgoing', private: true, inbox: twilio_inbox, account: account)
+        ::Twilio::SendOnTwilioService.new(message: message).perform
         expect(twilio_client).not_to have_received(:messages)
       end
 
-      it 'if inbox channel is not facebook page' do
-        create(:message, message_type: 'outgoing', inbox: widget_inbox, account: account)
+      it 'if inbox channel is not twilio' do
+        message = create(:message, message_type: 'outgoing', inbox: widget_inbox, account: account)
+        expect { ::Twilio::SendOnTwilioService.new(message: message).perform }.to raise_error 'Invalid channel service was called'
         expect(twilio_client).not_to have_received(:messages)
       end
 
       it 'if message is not outgoing' do
-        create(:message, message_type: 'incoming', inbox: twilio_inbox, account: account)
+        message = create(:message, message_type: 'incoming', inbox: twilio_inbox, account: account)
+        ::Twilio::SendOnTwilioService.new(message: message).perform
         expect(twilio_client).not_to have_received(:messages)
       end
 
       it 'if message has an source id' do
-        create(:message, message_type: 'outgoing', inbox: twilio_inbox, account: account, source_id: SecureRandom.uuid)
+        message = create(:message, message_type: 'outgoing', inbox: twilio_inbox, account: account, source_id: SecureRandom.uuid)
+        ::Twilio::SendOnTwilioService.new(message: message).perform
         expect(twilio_client).not_to have_received(:messages)
       end
     end
@@ -53,6 +57,7 @@ describe Twilio::SendOnTwilioService do
         outgoing_message = create(
           :message, message_type: 'outgoing', inbox: twilio_inbox, account: account, conversation: conversation
         )
+        ::Twilio::SendOnTwilioService.new(message: outgoing_message).perform
 
         expect(outgoing_message.reload.source_id).to eq('1234')
       end
@@ -69,6 +74,8 @@ describe Twilio::SendOnTwilioService do
       attachment = message.attachments.new(account_id: message.account_id, file_type: :image)
       attachment.file.attach(io: File.open(Rails.root.join('spec/assets/avatar.png')), filename: 'avatar.png', content_type: 'image/png')
       message.save!
+
+      ::Twilio::SendOnTwilioService.new(message: message).perform
     end
   end
 end
diff --git a/spec/services/twitter/send_on_twitter_service_spec.rb b/spec/services/twitter/send_on_twitter_service_spec.rb
index 017fcf27c..4e0f4e91e 100644
--- a/spec/services/twitter/send_on_twitter_service_spec.rb
+++ b/spec/services/twitter/send_on_twitter_service_spec.rb
@@ -41,22 +41,26 @@ describe Twitter::SendOnTwitterService do
   describe '#perform' do
     context 'without reply' do
       it 'if inbox channel is not twitter profile' do
-        create(:message, message_type: 'outgoing', inbox: widget_inbox, account: account)
+        message = create(:message, message_type: 'outgoing', inbox: widget_inbox, account: account)
+        expect { ::Twitter::SendOnTwitterService.new(message: message).perform }.to raise_error 'Invalid channel service was called'
         expect(twitter_client).not_to have_received(:send_direct_message)
       end
 
       it 'if message is private' do
-        create(:message, message_type: 'outgoing', private: true, inbox: twitter_inbox, account: account)
+        message = create(:message, message_type: 'outgoing', private: true, inbox: twitter_inbox, account: account)
+        ::Twitter::SendOnTwitterService.new(message: message).perform
         expect(twitter_client).not_to have_received(:send_direct_message)
       end
 
       it 'if message has source_id' do
-        create(:message, message_type: 'outgoing', source_id: '123', inbox: widget_inbox, account: account)
+        message = create(:message, message_type: 'outgoing', source_id: '123', inbox: twitter_inbox, account: account)
+        ::Twitter::SendOnTwitterService.new(message: message).perform
         expect(twitter_client).not_to have_received(:send_direct_message)
       end
 
       it 'if message is not outgoing' do
-        create(:message, message_type: 'incoming', inbox: twitter_inbox, account: account)
+        message = create(:message, message_type: 'incoming', inbox: twitter_inbox, account: account)
+        ::Twitter::SendOnTwitterService.new(message: message).perform
         expect(twitter_client).not_to have_received(:send_direct_message)
       end
     end
@@ -64,15 +68,17 @@ describe Twitter::SendOnTwitterService do
     context 'with reply' do
       it 'if conversation is a direct message' do
         create(:message, message_type: :incoming, inbox: twitter_inbox, account: account, conversation: dm_conversation)
-        create(:message, message_type: :outgoing, inbox: twitter_inbox, account: account, conversation: dm_conversation)
+        message = create(:message, message_type: :outgoing, inbox: twitter_inbox, account: account, conversation: dm_conversation)
+        ::Twitter::SendOnTwitterService.new(message: message).perform
         expect(twitter_client).to have_received(:send_direct_message)
       end
 
       it 'if conversation is a tweet' do
         create(:message, message_type: :incoming, inbox: twitter_inbox, account: account, conversation: tweet_conversation)
-        tweet = create(:message, message_type: :outgoing, inbox: twitter_inbox, account: account, conversation: tweet_conversation)
+        message = create(:message, message_type: :outgoing, inbox: twitter_inbox, account: account, conversation: tweet_conversation)
+        ::Twitter::SendOnTwitterService.new(message: message).perform
         expect(twitter_client).to have_received(:send_tweet_reply)
-        expect(tweet.reload.source_id).to eq '12345'
+        expect(message.reload.source_id).to eq '12345'
       end
     end
   end