test: Add flag mocks

This mocks 'image-builder.snapshots.enabled' where needed.

src/test/Components/CreateImageWizard/CreateImageWizard.compliance.test.tsx

@@ -43,9 +43,16 @@ vi.mock('@redhat-cloud-services/frontend-components/useChrome', () => ({
 
 vi.mock('@unleash/proxy-client-react', () => ({
   useUnleashContext: () => vi.fn(),
-  useFlag: vi.fn((flag) =>
-    flag === 'image-builder.wsl.enabled' ? true : false
-  ),
+  useFlag: vi.fn((flag) => {
+    switch (flag) {
+      case 'image-builder.wsl.enabled':
+        return true;
+      case 'image-builder.snapshots.enabled':
+        return true;
+      default:
+        return false;
+    }
+  }),
 }));
 
 const selectRhel8 = async () => {

src/test/Components/CreateImageWizard/CreateImageWizard.content.test.tsx

@@ -47,9 +47,16 @@ vi.mock('@redhat-cloud-services/frontend-components/useChrome', () => ({
 
 vi.mock('@unleash/proxy-client-react', () => ({
   useUnleashContext: () => vi.fn(),
-  useFlag: vi.fn((flag) =>
-    flag === 'image-builder.pkgrecs.enabled' ? true : false
-  ),
+  useFlag: vi.fn((flag) => {
+    switch (flag) {
+      case 'image-builder.pkgrecs.enabled':
+        return true;
+      case 'image-builder.snapshots.enabled':
+        return true;
+      default:
+        return false;
+    }
+  }),
 }));
 
 const typeIntoSearchBox = async (searchTerm: string) => {

src/test/Components/CreateImageWizard/steps/FirstBoot/Firstboot.test.tsx

@@ -46,9 +46,16 @@ vi.mock('@redhat-cloud-services/frontend-components/useChrome', () => ({
 
 vi.mock('@unleash/proxy-client-react', () => ({
   useUnleashContext: () => vi.fn(),
-  useFlag: vi.fn((flag) =>
-    flag === 'image-builder.firstboot.enabled' ? true : false
-  ),
+  useFlag: vi.fn((flag) => {
+    switch (flag) {
+      case 'image-builder.firstboot.enabled':
+        return true;
+      case 'image-builder.snapshots.enabled':
+        return true;
+      default:
+        return false;
+    }
+  }),
 }));
 
 const goToFirstBootStep = async (): Promise<void> => {

src/test/Components/CreateImageWizard/steps/Packages/Packages.test.tsx

@@ -51,6 +51,8 @@ vi.mock('@unleash/proxy-client-react', () => ({
         return true;
       case 'image-builder.firstboot.enabled':
         return true;
+      case 'image-builder.snapshots.enabled':
+        return true;
       default:
         return false;
     }

src/test/Components/CreateImageWizard/steps/Repositories/Repositories.test.tsx

@@ -46,9 +46,16 @@ vi.mock('@redhat-cloud-services/frontend-components/useChrome', () => ({
 
 vi.mock('@unleash/proxy-client-react', () => ({
   useUnleashContext: () => vi.fn(),
-  useFlag: vi.fn((flag) =>
-    flag === 'image-builder.firstboot.enabled' ? true : false
-  ),
+  useFlag: vi.fn((flag) => {
+    switch (flag) {
+      case 'image-builder.firstboot.enabled':
+        return true;
+      case 'image-builder.snapshots.enabled':
+        return true;
+      default:
+        return false;
+    }
+  }),
 }));
 
 const goToRepositoriesStep = async () => {

src/test/Components/CreateImageWizard/steps/Snapshot/Snapshot.test.tsx

@@ -42,9 +42,16 @@ vi.mock('@redhat-cloud-services/frontend-components/useChrome', () => ({
 
 vi.mock('@unleash/proxy-client-react', () => ({
   useUnleashContext: () => vi.fn(),
-  useFlag: vi.fn((flag) =>
-    flag === 'image-builder.firstboot.enabled' ? true : false
-  ),
+  useFlag: vi.fn((flag) => {
+    switch (flag) {
+      case 'image-builder.firstboot.enabled':
+        return true;
+      case 'image-builder.snapshots.enabled':
+        return true;
+      default:
+        return false;
+    }
+  }),
 }));
 
 const goToSnapshotStep = async () => {