From 2febf98e936f2356a343b7fb8bf8d126ce182a73 Mon Sep 17 00:00:00 2001 From: regexowl Date: Thu, 1 Aug 2024 17:52:36 +0200 Subject: [PATCH] test: Update tests Add mocking to tests where needed. --- src/test/Components/Blueprints/Blueprints.test.tsx | 13 ++++++++++--- .../steps/Details/Details.test.tsx | 13 ++++++++++--- .../FileSystemConfiguration.test.tsx | 13 ++++++++++--- .../steps/ImageOutput/TargetEnvironment.test.tsx | 2 ++ .../CreateImageWizard/steps/Oscap/Oscap.test.tsx | 13 ++++++++++--- .../steps/Registration/Registration.test.tsx | 13 ++++++++++--- .../steps/TargetEnvironment/AwsTarget.test.tsx | 13 ++++++++++--- .../steps/TargetEnvironment/AzureTarget.test.tsx | 13 ++++++++++--- .../steps/TargetEnvironment/GCPTarget.test.tsx | 13 ++++++++++--- 9 files changed, 82 insertions(+), 24 deletions(-) diff --git a/src/test/Components/Blueprints/Blueprints.test.tsx b/src/test/Components/Blueprints/Blueprints.test.tsx index b46f3430..5a1eb687 100644 --- a/src/test/Components/Blueprints/Blueprints.test.tsx +++ b/src/test/Components/Blueprints/Blueprints.test.tsx @@ -24,9 +24,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 selectBlueprintById = async (bpId: string) => { diff --git a/src/test/Components/CreateImageWizard/steps/Details/Details.test.tsx b/src/test/Components/CreateImageWizard/steps/Details/Details.test.tsx index fb471001..6ce518f0 100644 --- a/src/test/Components/CreateImageWizard/steps/Details/Details.test.tsx +++ b/src/test/Components/CreateImageWizard/steps/Details/Details.test.tsx @@ -38,9 +38,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 goToDetailsStep = async () => { diff --git a/src/test/Components/CreateImageWizard/steps/FileSystemConfiguration/FileSystemConfiguration.test.tsx b/src/test/Components/CreateImageWizard/steps/FileSystemConfiguration/FileSystemConfiguration.test.tsx index ca2397f8..ed5eb65c 100644 --- a/src/test/Components/CreateImageWizard/steps/FileSystemConfiguration/FileSystemConfiguration.test.tsx +++ b/src/test/Components/CreateImageWizard/steps/FileSystemConfiguration/FileSystemConfiguration.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.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 goToFileSystemConfigurationStep = async () => { diff --git a/src/test/Components/CreateImageWizard/steps/ImageOutput/TargetEnvironment.test.tsx b/src/test/Components/CreateImageWizard/steps/ImageOutput/TargetEnvironment.test.tsx index 1931c1c4..46ef6757 100644 --- a/src/test/Components/CreateImageWizard/steps/ImageOutput/TargetEnvironment.test.tsx +++ b/src/test/Components/CreateImageWizard/steps/ImageOutput/TargetEnvironment.test.tsx @@ -77,6 +77,8 @@ vi.mock('@unleash/proxy-client-react', () => ({ return true; case 'image-builder.wsl.enabled': return true; + case 'image-builder.snapshots.enabled': + return true; default: return false; } diff --git a/src/test/Components/CreateImageWizard/steps/Oscap/Oscap.test.tsx b/src/test/Components/CreateImageWizard/steps/Oscap/Oscap.test.tsx index 4fed3e63..035376d9 100644 --- a/src/test/Components/CreateImageWizard/steps/Oscap/Oscap.test.tsx +++ b/src/test/Components/CreateImageWizard/steps/Oscap/Oscap.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.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 goToOscapStep = async () => { diff --git a/src/test/Components/CreateImageWizard/steps/Registration/Registration.test.tsx b/src/test/Components/CreateImageWizard/steps/Registration/Registration.test.tsx index cb0a93ba..6ded6439 100644 --- a/src/test/Components/CreateImageWizard/steps/Registration/Registration.test.tsx +++ b/src/test/Components/CreateImageWizard/steps/Registration/Registration.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.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 selectActivationKey = async () => { diff --git a/src/test/Components/CreateImageWizard/steps/TargetEnvironment/AwsTarget.test.tsx b/src/test/Components/CreateImageWizard/steps/TargetEnvironment/AwsTarget.test.tsx index a5132c86..f5722b34 100644 --- a/src/test/Components/CreateImageWizard/steps/TargetEnvironment/AwsTarget.test.tsx +++ b/src/test/Components/CreateImageWizard/steps/TargetEnvironment/AwsTarget.test.tsx @@ -41,9 +41,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 goToAwsStep = async () => { diff --git a/src/test/Components/CreateImageWizard/steps/TargetEnvironment/AzureTarget.test.tsx b/src/test/Components/CreateImageWizard/steps/TargetEnvironment/AzureTarget.test.tsx index bd1fdb4e..8083974e 100644 --- a/src/test/Components/CreateImageWizard/steps/TargetEnvironment/AzureTarget.test.tsx +++ b/src/test/Components/CreateImageWizard/steps/TargetEnvironment/AzureTarget.test.tsx @@ -41,9 +41,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 goToAzureStep = async () => { diff --git a/src/test/Components/CreateImageWizard/steps/TargetEnvironment/GCPTarget.test.tsx b/src/test/Components/CreateImageWizard/steps/TargetEnvironment/GCPTarget.test.tsx index caf3bb16..b2c50069 100644 --- a/src/test/Components/CreateImageWizard/steps/TargetEnvironment/GCPTarget.test.tsx +++ b/src/test/Components/CreateImageWizard/steps/TargetEnvironment/GCPTarget.test.tsx @@ -44,9 +44,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 GCP_ACCOUNT = 'test@gmail.com';