From 989671ba956d96637119f95267b549e388bb9de0 Mon Sep 17 00:00:00 2001
From: regexowl <ksimicko@redhat.com>
Date: Thu, 11 Jul 2024 15:55:07 +0200
Subject: [PATCH] test: Update tests

A custom repo needs to be selected for the correct mocked response.
---
 .../CreateImageWizard.content.test.tsx              | 13 +++++++++++++
 .../steps/Packages/Packages.test.tsx                |  2 ++
 2 files changed, 15 insertions(+)

diff --git a/src/test/Components/CreateImageWizardV2/CreateImageWizard.content.test.tsx b/src/test/Components/CreateImageWizardV2/CreateImageWizard.content.test.tsx
index e6cf95ba..a43eb4ea 100644
--- a/src/test/Components/CreateImageWizardV2/CreateImageWizard.content.test.tsx
+++ b/src/test/Components/CreateImageWizardV2/CreateImageWizard.content.test.tsx
@@ -116,6 +116,16 @@ const checkRecommendationsEmptyState = async () => {
   await screen.findByText('Select packages to generate recommendations.');
 };
 
+export const selectCustomRepo = async () => {
+  await clickBack();
+  const customRepoCheckbox = await screen.findByRole('checkbox', {
+    name: /select row 0/i,
+  });
+
+  await userEvent.click(customRepoCheckbox);
+  await clickNext();
+};
+
 describe('Step Packages', () => {
   const setUp = async () => {
     mockContentSourcesEnabled = false;
@@ -225,6 +235,7 @@ describe('Step Packages', () => {
   test('search results should be sorted with most relevant results first', async () => {
     await setUp();
 
+    await selectCustomRepo();
     await typeIntoSearchBox('test');
 
     const packagesTable = await screen.findByTestId('packages-table');
@@ -246,6 +257,7 @@ describe('Step Packages', () => {
   test('selected packages are sorted the same way as available packages', async () => {
     await setUp();
 
+    await selectCustomRepo();
     await typeIntoSearchBox('test');
 
     const checkboxes = await getAllCheckboxes();
@@ -323,6 +335,7 @@ describe('Step Packages', () => {
   test('should display relevant results in selected first', async () => {
     await setUp();
 
+    await selectCustomRepo();
     await typeIntoSearchBox('test');
 
     const checkboxes = await getAllCheckboxes();
diff --git a/src/test/Components/CreateImageWizardV2/steps/Packages/Packages.test.tsx b/src/test/Components/CreateImageWizardV2/steps/Packages/Packages.test.tsx
index 1e28b399..bc5480f6 100644
--- a/src/test/Components/CreateImageWizardV2/steps/Packages/Packages.test.tsx
+++ b/src/test/Components/CreateImageWizardV2/steps/Packages/Packages.test.tsx
@@ -12,6 +12,7 @@ import {
   packagesCreateBlueprintRequest,
 } from '../../../../fixtures/editMode';
 import { clickNext } from '../../../../testUtils';
+import { selectCustomRepo } from '../../CreateImageWizard.content.test';
 import {
   blueprintRequest,
   clickRegisterLater,
@@ -281,6 +282,7 @@ describe('pagination on packages step', () => {
   test('itemcount correct after search', async () => {
     await renderCreateMode();
     await goToPackagesStep();
+    await selectCustomRepo();
     await searchForPackage();
     await selectFirstPackage();
     // the pagination in the top right