1
diff --git a/apps/web/test/unit-tests/components/views/spaces/SpaceTreeLevel-test.tsx b/apps/web/test/unit-tests/components/views/spaces/SpaceTreeLevel-test.tsx
index 7b3b93c384..c5940c945e 100644
--- a/apps/web/test/unit-tests/components/views/spaces/SpaceTreeLevel-test.tsx
+++ b/apps/web/test/unit-tests/components/views/spaces/SpaceTreeLevel-test.tsx
@@ -127,7 +127,7 @@ describe("SpaceButton", () => {
/>,
);
- expect(container.querySelector(".mx_NotificationBadge_count")).toHaveTextContent("8");
+ expect(getByTestId(container, "notification-badge")).toHaveTextContent("8");
expect(asFragment()).toMatchSnapshot();
});
});
diff --git a/apps/web/test/unit-tests/components/views/spaces/ThreadsActivityCentre-test.tsx b/apps/web/test/unit-tests/components/views/spaces/ThreadsActivityCentre-test.tsx
index 2e722eea3d..271b280d7c 100644
--- a/apps/web/test/unit-tests/components/views/spaces/ThreadsActivityCentre-test.tsx
+++ b/apps/web/test/unit-tests/components/views/spaces/ThreadsActivityCentre-test.tsx
@@ -124,7 +124,7 @@ describe("ThreadsActivityCentre", () => {
expect(tacRows.length).toEqual(1);
getByText(tacRows[0], "A notification");
- expect(tacRows[0].getElementsByClassName("mx_NotificationBadge_level_notification").length).toEqual(1);
+ expect(tacRows[0].querySelector('[data-notification-level="notification"]')).toBeInTheDocument();
});
it("should render a room with a highlight notification in the TAC", async () => {
@@ -136,7 +136,7 @@ describe("ThreadsActivityCentre", () => {
expect(tacRows.length).toEqual(1);
getByText(tacRows[0], "This is a real highlight");
- expect(tacRows[0].getElementsByClassName("mx_NotificationBadge_level_highlight").length).toEqual(1);
+ expect(tacRows[0].querySelector('[data-notification-level="highlight"]')).toBeInTheDocument();
});
it("renders notifications matching the snapshot", async () => {
diff --git a/apps/web/test/unit-tests/components/views/spaces/__snapshots__/SpaceTreeLevel-test.tsx.snap b/apps/web/test/unit-tests/components/views/spaces/__snapshots__/SpaceTreeLevel-test.tsx.snap
index edecf0f04c..6f9a846f1d 100644
--- a/apps/web/test/unit-tests/components/views/spaces/__snapshots__/SpaceTreeLevel-test.tsx.snap
+++ b/apps/web/test/unit-tests/components/views/spaces/__snapshots__/SpaceTreeLevel-test.tsx.snap
@@ -25,17 +25,21 @@ exports[`SpaceButton metaspace should render notificationState if one is provide