diff --git a/src/helpers/heritage-integration/cookie-helper.spec.js b/src/helpers/heritage-integration/cookie-helper.spec.js index 34ae00b86..a18906381 100644 --- a/src/helpers/heritage-integration/cookie-helper.spec.js +++ b/src/helpers/heritage-integration/cookie-helper.spec.js @@ -11,9 +11,15 @@ import { setSessionIdIfUnset, isCookieSet, refreshCookieExpiration, + getAffiliateCookies, } from "@/helpers/heritage-integration/cookie-helper.js"; import { cookieNames } from "@/constants/cookie-names"; -import { removeAllTestCookies, setupCookies, setupCrypto } from "@/helpers/unit-test-helper"; +import { + removeAllTestCookies, + setupCookies, + setupCrypto, + setupAffiliateCookies, +} from "@/helpers/unit-test-helper"; import { deleteFunnelCookie } from "./cookie-helper"; const randomUUID = "68d89736-c277-46f1-8fee-c3dacdb23c08"; @@ -431,3 +437,21 @@ describe("cookies", () => { }); }); }); +describe("getAffiliateCookies", () => { + it("should return an empty array if no affiliate cookies are found", () => { + setupAffiliateCookies({}); + expect(getAffiliateCookies()).toEqual([]); + }); + + it("should return an array of affiliate cookies when they are present", () => { + setupAffiliateCookies({ + CookieName: "affiliate_cookie1_Track", + CookieValue: "value_of_affiliate_cookie1", + }); + + const expectedCookies = [ + { CookieName: "affiliate_cookie1_Track", CookieValue: "value_of_affiliate_cookie1" }, + ]; + expect(getAffiliateCookies()).toEqual(expectedCookies); + }); +}); diff --git a/src/helpers/unit-test-helper.js b/src/helpers/unit-test-helper.js index 2cd580df0..7c3c9ed4e 100644 --- a/src/helpers/unit-test-helper.js +++ b/src/helpers/unit-test-helper.js @@ -139,6 +139,10 @@ export function setupCookies({ funnelCookieValue = "", includeHeritageCookie = t }); } +export function setupAffiliateCookies(cookie) { + setCookieProperties({ [cookie.CookieName]: cookie.CookieValue }, { isSecure: false }); +} + export function setupCrypto(mockValue) { global.crypto = { randomUUID: jest.fn(),