diff --git a/internal_packages/composer/lib/send-action-button.jsx b/internal_packages/composer/lib/send-action-button.jsx index 306b14742..dc6825eb1 100644 --- a/internal_packages/composer/lib/send-action-button.jsx +++ b/internal_packages/composer/lib/send-action-button.jsx @@ -123,5 +123,6 @@ Object.assign(EnhancedSendActionButton.prototype, { }, }) -export const UndecoratedSendActionButton = SendActionButton +EnhancedSendActionButton.UndecoratedSendActionButton = SendActionButton + export default EnhancedSendActionButton diff --git a/internal_packages/composer/spec/send-action-button-spec.jsx b/internal_packages/composer/spec/send-action-button-spec.jsx index 6fdff92d0..369a514af 100644 --- a/internal_packages/composer/spec/send-action-button-spec.jsx +++ b/internal_packages/composer/spec/send-action-button-spec.jsx @@ -2,7 +2,9 @@ import React from 'react'; import {mount} from 'enzyme'; import {ButtonDropdown, RetinaImg} from 'nylas-component-kit'; import {Actions, Message, SendActionsStore} from 'nylas-exports'; -import {UndecoratedSendActionButton as SendActionButton} from '../lib/send-action-button'; +import SendActionButton from '../lib/send-action-button'; + +const {UndecoratedSendActionButton} = SendActionButton; const {DefaultSendAction} = SendActionsStore @@ -40,7 +42,7 @@ describe('SendActionButton', function describeBlock() { const render = (draft, {isValid = true, sendActions = [], ordered = {}} = {}) => { this.isValidDraft.andReturn(isValid) return mount( - { const sendActionButton = render(this.draft); - expect(sendActionButton.is(SendActionButton)).toBe(true); + expect(sendActionButton.is(UndecoratedSendActionButton)).toBe(true); }); it("initializes with the default and shows the standard Send option", () => {