diff --git a/src/renderer/components/update-button/__tests__/update-button.test.tsx b/src/renderer/components/update-button/__tests__/update-button.test.tsx
index 98a86aa318..ff52035a48 100644
--- a/src/renderer/components/update-button/__tests__/update-button.test.tsx
+++ b/src/renderer/components/update-button/__tests__/update-button.test.tsx
@@ -3,7 +3,7 @@
* Licensed under MIT License. See LICENSE in root directory for more information.
*/
-import { render } from "@testing-library/react";
+import { render, act } from "@testing-library/react";
import React from "react";
import { UpdateButton } from "../update-button";
import "@testing-library/jest-dom/extend-expect";
@@ -27,12 +27,12 @@ describe("", () => {
expect(getByTestId("update-button")).toMatchSnapshot();
});
- it("should open menu when clicked", () => {
+ it("should open menu when clicked", async () => {
const { getByTestId } = render();
const button = getByTestId("update-button");
- button.click();
+ act(() => button.click());
expect(getByTestId("update-lens-menu-item")).toBeInTheDocument();
});
@@ -42,7 +42,7 @@ describe("", () => {
const button = getByTestId("update-button");
- button.click();
+ act(() => button.click());
const menuItem = getByTestId("update-lens-menu-item");
diff --git a/src/renderer/components/update-button/update-button.tsx b/src/renderer/components/update-button/update-button.tsx
index 6da453de3d..f2e19b7979 100644
--- a/src/renderer/components/update-button/update-button.tsx
+++ b/src/renderer/components/update-button/update-button.tsx
@@ -50,6 +50,7 @@ export function UpdateButton({ warningLevel, update, id }: UpdateButtonProps) {
isOpen={opened}
close={toggle}
open={toggle}
+ aria-expanded={opened}
>