diff --git a/src/layouts/payment-method/payment-method.spec.js b/src/layouts/payment-method/payment-method.spec.js index f791e600..58811293 100644 --- a/src/layouts/payment-method/payment-method.spec.js +++ b/src/layouts/payment-method/payment-method.spec.js @@ -16,7 +16,8 @@ import coverageType from '@/constants/coverage-type'; // Mock so it can be used as an assertion jest.mock('@/helpers/order-helper.js', () => ({ - submitWorkOrder: jest.fn() + submitWorkOrder: jest.fn(), + saveSession: jest.fn() })); function setupMocks({ customMountOptions = {}, queryString }, mainInitialState = {}, customMixin = null) { @@ -319,14 +320,8 @@ describe('payment-method.vue', () => { describe('forwardButtonAction', () => { test('Pay at the time of service is selected, should call saveWorkOrder', async () => { // Arrange - const store = { - order: { - payment: { - paymentMethod: paymentMethods.PAY_AT_TIME_OF_SERVICE - } - } - }; - const wrapper = setupMocks({}, store); + const wrapper = setupMocks({}); + wrapper.vm.paymentMethod = paymentMethods.PAY_AT_TIME_OF_SERVICE; // Act await wrapper.vm.forwardButtonAction(); diff --git a/src/layouts/payment-method/payment-method.vue b/src/layouts/payment-method/payment-method.vue index b253f2e0..90d12c7b 100644 --- a/src/layouts/payment-method/payment-method.vue +++ b/src/layouts/payment-method/payment-method.vue @@ -127,12 +127,13 @@ export default { next((vm) => { vm.setCmsContent(resultMap.cmsContent); + vm.initializeComponent(); vm.updateFooterButtonText(vm.customCallToActionButtonCopy); }); }, data() { return { - paymentMethod: this.getPaymentMethodFromStore(), + paymentMethod: null, widget: { paymentMethod: 'PaymentMethodWidget', paymentMethodApplePay: 'PaymentMethodWidgetApplePay' @@ -251,8 +252,10 @@ export default { && serviceLocationReqs && scheduleReqs && customerReqs - && contactInfoReqs - ); + && contactInfoReqs); + }, + initializeComponent() { + this.paymentMethod = this.getPaymentMethodFromStore(); }, getPaymentMethodFromStore() { if (this.isPayInAdvanceDisabled) {