diff --git a/src/renderer/components/+catalog/__tests__/catalog-category-label.test.tsx b/src/renderer/components/+catalog/__tests__/catalog-category-label.test.tsx index 015929b643..41796e3226 100644 --- a/src/renderer/components/+catalog/__tests__/catalog-category-label.test.tsx +++ b/src/renderer/components/+catalog/__tests__/catalog-category-label.test.tsx @@ -37,7 +37,7 @@ describe("CatalogCategoryLabel", () => { it("renders without a badge", async () => { const category = new CatalogCategoryWithoutBadge(); - render(); + render(); expect(await screen.findByText("Test Category")).toBeInTheDocument(); }); @@ -45,7 +45,7 @@ describe("CatalogCategoryLabel", () => { it("renders with a badge", async () => { const category = new CatalogCategoryWithBadge(); - render(); + render(); expect(await screen.findByText("Test Category")).toBeInTheDocument(); expect(await screen.findByText("Test Badge")).toBeInTheDocument(); diff --git a/src/renderer/components/+catalog/catalog-category-label.tsx b/src/renderer/components/+catalog/catalog-category-label.tsx index f5a82eda27..46222490c8 100644 --- a/src/renderer/components/+catalog/catalog-category-label.tsx +++ b/src/renderer/components/+catalog/catalog-category-label.tsx @@ -3,19 +3,19 @@ * Licensed under MIT License. See LICENSE in root directory for more information. */ import React from "react"; -import type { CatalogCategory } from "../../../common/catalog/catalog-entity"; interface CatalogCategoryLabelProps { - category: CatalogCategory; + label: string | React.ReactNode; + badge?: React.ReactNode; } /** * Display label for Catalog Category for the Catalog menu */ -export const CatalogCategoryLabel = ({ category }: CatalogCategoryLabelProps) => +export const CatalogCategoryLabel = ({ label, badge }: CatalogCategoryLabelProps) => (
-
{category.metadata.name}
-
{category.getBadge()}
+
{label}
+ {badge ? (
{badge}
) : null}
); diff --git a/src/renderer/components/+catalog/catalog-menu.tsx b/src/renderer/components/+catalog/catalog-menu.tsx index d2942d72cf..eea6e012b1 100644 --- a/src/renderer/components/+catalog/catalog-menu.tsx +++ b/src/renderer/components/+catalog/catalog-menu.tsx @@ -67,7 +67,7 @@ export const CatalogMenu = observer((props: CatalogMenuProps) => { icon={getCategoryIcon(category)} key={category.getId()} nodeId={category.getId()} - label={} + label={} data-testid={`${category.getId()}-tab`} onClick={() => props.onItemClick(category.getId())} />