test: Add flag mocks

This adds mocked 'image-builder.firstboot.enabled' where needed.

src/test/Components/Blueprints/Blueprints.test.tsx

@@ -24,7 +24,9 @@ vi.mock('@redhat-cloud-services/frontend-components/useChrome', () => ({
 
 vi.mock('@unleash/proxy-client-react', () => ({
   useUnleashContext: () => vi.fn(),
-  useFlag: vi.fn(() => false),
+  useFlag: vi.fn((flag) =>
+    flag === 'image-builder.firstboot.enabled' ? true : false
+  ),
 }));
 
 const selectBlueprintById = async (bpId: string) => {

src/test/Components/CreateImageWizard/EditImageWizard.test.tsx

@@ -17,7 +17,9 @@ vi.mock('@redhat-cloud-services/frontend-components/useChrome', () => ({
 
 vi.mock('@unleash/proxy-client-react', () => ({
   useUnleashContext: () => vi.fn(),
-  useFlag: vi.fn(() => false),
+  useFlag: vi.fn((flag) =>
+    flag === 'image-builder.firstboot.enabled' ? true : false
+  ),
 }));
 
 describe('EditImageWizard', () => {

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

@@ -38,7 +38,9 @@ vi.mock('@redhat-cloud-services/frontend-components/useChrome', () => ({
 
 vi.mock('@unleash/proxy-client-react', () => ({
   useUnleashContext: () => vi.fn(),
-  useFlag: vi.fn(() => false),
+  useFlag: vi.fn((flag) =>
+    flag === 'image-builder.firstboot.enabled' ? true : false
+  ),
 }));
 
 const goToDetailsStep = async () => {

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

@@ -43,7 +43,9 @@ vi.mock('@redhat-cloud-services/frontend-components/useChrome', () => ({
 
 vi.mock('@unleash/proxy-client-react', () => ({
   useUnleashContext: () => vi.fn(),
-  useFlag: vi.fn(() => false),
+  useFlag: vi.fn((flag) =>
+    flag === 'image-builder.firstboot.enabled' ? true : false
+  ),
 }));
 
 const goToFileSystemConfigurationStep = async () => {

src/test/Components/CreateImageWizard/steps/ImageOutput/TargetEnvironment.test.tsx

@@ -71,7 +71,9 @@ vi.mock('@redhat-cloud-services/frontend-components/useChrome', () => ({
 
 vi.mock('@unleash/proxy-client-react', () => ({
   useUnleashContext: () => vi.fn(),
-  useFlag: vi.fn(() => false),
+  useFlag: vi.fn((flag) =>
+    flag === 'image-builder.firstboot.enabled' ? true : false
+  ),
 }));
 
 const openReleaseMenu = async () => {

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

@@ -45,7 +45,9 @@ vi.mock('@redhat-cloud-services/frontend-components/useChrome', () => ({
 
 vi.mock('@unleash/proxy-client-react', () => ({
   useUnleashContext: () => vi.fn(),
-  useFlag: vi.fn(() => false),
+  useFlag: vi.fn((flag) =>
+    flag === 'image-builder.firstboot.enabled' ? true : false
+  ),
 }));
 
 const goToOscapStep = async () => {

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

@@ -45,9 +45,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.firstboot.enabled':
+        return true;
+      default:
+        return false;
+    }
+  }),
 }));
 
 const goToPackagesStep = async () => {

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

@@ -45,7 +45,9 @@ vi.mock('@redhat-cloud-services/frontend-components/useChrome', () => ({
 
 vi.mock('@unleash/proxy-client-react', () => ({
   useUnleashContext: () => vi.fn(),
-  useFlag: vi.fn(() => false),
+  useFlag: vi.fn((flag) =>
+    flag === 'image-builder.firstboot.enabled' ? true : false
+  ),
 }));
 
 const selectActivationKey = async () => {

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

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

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

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

src/test/Components/CreateImageWizard/steps/TargetEnvironment/AwsTarget.test.tsx

@@ -41,7 +41,9 @@ vi.mock('@redhat-cloud-services/frontend-components/useChrome', () => ({
 
 vi.mock('@unleash/proxy-client-react', () => ({
   useUnleashContext: () => vi.fn(),
-  useFlag: vi.fn(() => false),
+  useFlag: vi.fn((flag) =>
+    flag === 'image-builder.firstboot.enabled' ? true : false
+  ),
 }));
 
 const goToAwsStep = async () => {

src/test/Components/CreateImageWizard/steps/TargetEnvironment/AzureTarget.test.tsx

@@ -41,7 +41,9 @@ vi.mock('@redhat-cloud-services/frontend-components/useChrome', () => ({
 
 vi.mock('@unleash/proxy-client-react', () => ({
   useUnleashContext: () => vi.fn(),
-  useFlag: vi.fn(() => false),
+  useFlag: vi.fn((flag) =>
+    flag === 'image-builder.firstboot.enabled' ? true : false
+  ),
 }));
 
 const goToAzureStep = async () => {

src/test/Components/CreateImageWizard/steps/TargetEnvironment/GCPTarget.test.tsx

@@ -44,7 +44,9 @@ vi.mock('@redhat-cloud-services/frontend-components/useChrome', () => ({
 
 vi.mock('@unleash/proxy-client-react', () => ({
   useUnleashContext: () => vi.fn(),
-  useFlag: vi.fn(() => false),
+  useFlag: vi.fn((flag) =>
+    flag === 'image-builder.firstboot.enabled' ? true : false
+  ),
 }));
 
 const GCP_ACCOUNT = 'test@gmail.com';