From f03e0737f872d190c2339ca85b3c082a0befa29a Mon Sep 17 00:00:00 2001
From: Katie <katie.oh@safelite.com>
Date: Fri, 27 May 2022 15:16:45 -0400
Subject: [PATCH] CSR-622 Fix tests

---
 src/helpers/heritage-integration/order-helper.spec.js | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/src/helpers/heritage-integration/order-helper.spec.js b/src/helpers/heritage-integration/order-helper.spec.js
index b66286d50..7d769a8d8 100644
--- a/src/helpers/heritage-integration/order-helper.spec.js
+++ b/src/helpers/heritage-integration/order-helper.spec.js
@@ -18,7 +18,8 @@ describe("loadOrderIfPresent", () => {
 
         const testCookieValue = {
             ShouldResetState: testShouldResetState,
-            ReferralCorrelationId: "xxx"
+            ReferralCorrelationId: "xxx",
+            ReferralNumber: "12345"
         }
 
         document.cookie = `${cookieNames.FUNNEL_SESSION_INFO}=${JSON.stringify(testCookieValue)}; path=/; ${cookieHelper.getCookieDomainValue()}`;
@@ -33,7 +34,11 @@ describe("loadOrderIfPresent", () => {
 
     test("ShouldResetState == true => reset store", () => {
         // Arrange
-        cookieHelper.getFunnelCookie = jest.spyOn(cookieHelper, "getFunnelCookie").mockReturnValueOnce({ ShouldResetState: true, ReferralCorrelationId: "xxx-xxx-xxx" });
+        cookieHelper.getFunnelCookie = jest.spyOn(cookieHelper, "getFunnelCookie").mockReturnValueOnce({ 
+            ShouldResetState: true, 
+            ReferralCorrelationId: "xxx-xxx-xxx",
+            ReferralNumber: "12345"
+        });
 
         const mockData = {
             actionList: [{