From 3b7a1338ec16154a1ff00df3379fce38aea54a5a Mon Sep 17 00:00:00 2001
From: regexowl <ksimicko@redhat.com>
Date: Tue, 23 Apr 2024 15:06:44 +0200
Subject: [PATCH] test: Add unleash flag mocking to the tests

The functionality is currently gated behind an unleash flag. This adds the mocked flag to the tests.
---
 .../CreateImageWizard.content.test.tsx            | 15 ++++++++++-----
 .../steps/Packages/Packages.test.tsx              |  7 +++++++
 2 files changed, 17 insertions(+), 5 deletions(-)

diff --git a/src/test/Components/CreateImageWizardV2/CreateImageWizard.content.test.tsx b/src/test/Components/CreateImageWizardV2/CreateImageWizard.content.test.tsx
index 15d5c8e4..f20dee7b 100644
--- a/src/test/Components/CreateImageWizardV2/CreateImageWizard.content.test.tsx
+++ b/src/test/Components/CreateImageWizardV2/CreateImageWizard.content.test.tsx
@@ -50,11 +50,16 @@ jest.mock('@redhat-cloud-services/frontend-components/useChrome', () => ({
 let mockContentSourcesEnabled: boolean;
 jest.mock('@unleash/proxy-client-react', () => ({
   useUnleashContext: () => jest.fn(),
-  useFlag: jest.fn((flag) =>
-    flag === 'image-builder.enable-content-sources'
-      ? mockContentSourcesEnabled
-      : false
-  ),
+  useFlag: jest.fn((flag) => {
+    switch (flag) {
+      case 'image-builder.enable-content-sources':
+        return mockContentSourcesEnabled;
+      case 'image-builder.pkgrecs.enabled':
+        return true;
+      default:
+        return false;
+    }
+  }),
 }));
 
 beforeAll(() => {
diff --git a/src/test/Components/CreateImageWizardV2/steps/Packages/Packages.test.tsx b/src/test/Components/CreateImageWizardV2/steps/Packages/Packages.test.tsx
index 447e95de..95657fe1 100644
--- a/src/test/Components/CreateImageWizardV2/steps/Packages/Packages.test.tsx
+++ b/src/test/Components/CreateImageWizardV2/steps/Packages/Packages.test.tsx
@@ -31,6 +31,13 @@ jest.mock('@redhat-cloud-services/frontend-components/useChrome', () => ({
   }),
 }));
 
+jest.mock('@unleash/proxy-client-react', () => ({
+  useUnleashContext: () => jest.fn(),
+  useFlag: jest.fn((flag) =>
+    flag === 'image-builder.pkgrecs.enabled' ? true : false
+  ),
+}));
+
 const goToPackagesStep = async () => {
   const bareMetalCheckBox = await screen.findByRole('checkbox', {
     name: /bare metal installer checkbox/i,