Improve unit test

agama-project · May 2, 2024 · 2f954fa · 2f954fa
1 parent da9ec7d
commit 2f954fa
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/web/src/components/storage/InstallationDeviceField.test.jsx b/web/src/components/storage/InstallationDeviceField.test.jsx
@@ -26,6 +26,15 @@ import { screen, within } from "@testing-library/react";
 import { installerRender } from "~/test-utils";
 import InstallationDeviceField from "~/components/storage/InstallationDeviceField";
 
+jest.mock("@patternfly/react-core", () => {
+  const original = jest.requireActual("@patternfly/react-core");
+
+  return {
+    ...original,
+    Skeleton: () => <div>PF-Skeleton</div>
+  };
+});
+
 /**
  * @typedef {import ("~/components/storage/InstallationDeviceField").InstallationDeviceFieldProps} InstallationDeviceFieldProps
  * @typedef {import ("~/client/storage").StorageDevice} StorageDevice
@@ -97,10 +106,10 @@ describe("when set as loading", () => {
   });
 
   it("renders a loading hint", () => {
-    const { container } = installerRender(<InstallationDeviceField {...props} />);
+    installerRender(<InstallationDeviceField {...props} />);
 
     // a PF skeleton is displayed
-    expect(container.querySelector("div.pf-v5-c-skeleton")).toBeVisible();
+    screen.getByText("PF-Skeleton");
   });
 });