From 07cda6d4ea30a89a74eb5a42aa98a2ee0fbea1e8 Mon Sep 17 00:00:00 2001
From: Alex Andreev <alex.andreev.email@gmail.com>
Date: Mon, 18 Oct 2021 10:22:07 +0300
Subject: [PATCH] Mocking HotbarToggleMenuItem

Signed-off-by: Alex Andreev <alex.andreev.email@gmail.com>
---
 src/renderer/components/+catalog/catalog.test.tsx           | 6 ++++++
 .../components/+catalog/hotbar-toggle-menu-item.tsx         | 6 +++---
 2 files changed, 9 insertions(+), 3 deletions(-)

diff --git a/src/renderer/components/+catalog/catalog.test.tsx b/src/renderer/components/+catalog/catalog.test.tsx
index bb1ce35a5b..ca09f880c7 100644
--- a/src/renderer/components/+catalog/catalog.test.tsx
+++ b/src/renderer/components/+catalog/catalog.test.tsx
@@ -46,6 +46,12 @@ jest.mock("@electron/remote", () => {
   };
 });
 
+jest.mock("./hotbar-toggle-menu-item", () => {
+  return {
+    HotbarToggleMenuItem: () => <div>menu item</div>
+  };
+});
+
 describe("<Catalog />", () => {
   const history = createMemoryHistory();
   const mockLocation = {
diff --git a/src/renderer/components/+catalog/hotbar-toggle-menu-item.tsx b/src/renderer/components/+catalog/hotbar-toggle-menu-item.tsx
index 67f02a8ff5..7d679e4d81 100644
--- a/src/renderer/components/+catalog/hotbar-toggle-menu-item.tsx
+++ b/src/renderer/components/+catalog/hotbar-toggle-menu-item.tsx
@@ -27,9 +27,9 @@ import type { CatalogEntity } from "../../api/catalog-entity";
 
 export function HotbarToggleMenuItem(props: { entity: CatalogEntity, addContent: ReactNode, removeContent: ReactNode }) {
   const store = HotbarStore.getInstance(false);
-  const add = () => store?.addToHotbar(props.entity);
-  const remove = () => store?.removeFromHotbar(props.entity.getId());
-  const [itemInHotbar, setItemInHotbar] = useState(store?.isAddedToActive(props.entity));
+  const add = () => store.addToHotbar(props.entity);
+  const remove = () => store.removeFromHotbar(props.entity.getId());
+  const [itemInHotbar, setItemInHotbar] = useState(store.isAddedToActive(props.entity));
 
   return (
     <MenuItem onClick={() => {