From ce2b45c4f72bb5679814255234811af776d1bb27 Mon Sep 17 00:00:00 2001
From: Bill Richardson <william.richardson@safelite.com>
Date: Thu, 31 Oct 2024 11:28:14 -0400
Subject: [PATCH] fix unit test

---
 src/layouts/payment-method/payment-method.spec.js | 13 ++++---------
 1 file changed, 4 insertions(+), 9 deletions(-)

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();