From 844cbf088f7079ae453835cb8de584d6b31ecd86 Mon Sep 17 00:00:00 2001
From: Katie <katie.oh@safelite.com>
Date: Fri, 20 May 2022 13:18:46 -0400
Subject: [PATCH] CSR-628 Fix tests

---
 .../heritage-integration/navigation-helper.spec.js        | 4 ++--
 src/helpers/heritage-integration/order-helper.spec.js     | 8 +++++---
 src/helpers/unit-test-helper.js                           | 5 +++--
 3 files changed, 10 insertions(+), 7 deletions(-)

diff --git a/src/helpers/heritage-integration/navigation-helper.spec.js b/src/helpers/heritage-integration/navigation-helper.spec.js
index 01bf5dcdc..e212284b8 100644
--- a/src/helpers/heritage-integration/navigation-helper.spec.js
+++ b/src/helpers/heritage-integration/navigation-helper.spec.js
@@ -305,7 +305,7 @@ describe("getPageToRouteExistingOrderTo", () => {
 describe("navigateToHeritageFunnel", () => {
     test("should save order", async () => {
         // Arrange
-        const mockReferralNumber = 2;
+        const mockReferralNumber = "2";
         const mockCorrelationId = "55";
         const mockReferralDate = "2022";
 
@@ -337,7 +337,7 @@ describe("navigateToHeritageFunnel", () => {
 
     test("should go to heritage funnel", async () => {
         // Arrange
-        const mockReferralNumber = 2;
+        const mockReferralNumber = "2";
         const mockCorrelationId = "55";
         const mockReferralDate = "2022";
 
diff --git a/src/helpers/heritage-integration/order-helper.spec.js b/src/helpers/heritage-integration/order-helper.spec.js
index 803c94b6b..b66286d50 100644
--- a/src/helpers/heritage-integration/order-helper.spec.js
+++ b/src/helpers/heritage-integration/order-helper.spec.js
@@ -103,11 +103,12 @@ describe("saveOrder", () => {
 
     test("saveOrder => should set state order values", async () => {
         // Arrange
-        const mockReferralNumber = 2;
+        const mockReferralNumber = "2";
         const mockCorrelationId = "55";
         const mockReferralDate = "2022";
+        const mockAccountNumber = "5";
 
-        const mockOrderInfo = getMockOrderInfo(mockReferralNumber, mockCorrelationId, mockReferralDate);
+        const mockOrderInfo = getMockOrderInfo(mockReferralNumber, mockCorrelationId, mockReferralDate, mockAccountNumber);
 
         const mockData = {
             actionList: [
@@ -131,7 +132,8 @@ describe("saveOrder", () => {
         expect(mocks.baseMixin.methods.dispatchStoreAction).toHaveBeenCalledWith(storeActions.SET_REFERRAL_INFORMATION, {
             referralNumber: mockReferralNumber,
             referralDate: mockReferralDate,
-            referralCorrelationId: mockCorrelationId
+            referralCorrelationId: mockCorrelationId,
+            accountNumber: mockAccountNumber
         }, false);
     });
 
diff --git a/src/helpers/unit-test-helper.js b/src/helpers/unit-test-helper.js
index 6577ed6ff..4f4214120 100644
--- a/src/helpers/unit-test-helper.js
+++ b/src/helpers/unit-test-helper.js
@@ -89,11 +89,12 @@ export function removeAllTestCookies() {
   });
 }
 
-export function getMockOrderInfo(mockReferralNumber, mockCorrelationId, mockReferralDate) {
+export function getMockOrderInfo(mockReferralNumber, mockCorrelationId, mockReferralDate, accountNumber = "0") {
   return {
       referralNumber: mockReferralNumber,
       referralCorrelationId: mockCorrelationId,
-      referralDate: mockReferralDate
+      referralDate: mockReferralDate,
+      accountNumber: accountNumber
   }
 }