From 7ea37913b03565dbed86ab912f431ae58d0d1a15 Mon Sep 17 00:00:00 2001
From: Alex Andreev <alex.andreev.email@gmail.com>
Date: Wed, 8 Dec 2021 14:02:23 +0300
Subject: [PATCH] Fix tests

Signed-off-by: Alex Andreev <alex.andreev.email@gmail.com>
---
 .../layout/__tests__/topbar-linux.test.tsx    |  6 +++++
 .../layout/__tests__/topbar-win.test.tsx      | 23 ++++++++++++++++---
 .../layout/__tests__/topbar.test.tsx          |  1 +
 3 files changed, 27 insertions(+), 3 deletions(-)

diff --git a/src/renderer/components/layout/__tests__/topbar-linux.test.tsx b/src/renderer/components/layout/__tests__/topbar-linux.test.tsx
index 00d327e318..f3ed0b0a04 100644
--- a/src/renderer/components/layout/__tests__/topbar-linux.test.tsx
+++ b/src/renderer/components/layout/__tests__/topbar-linux.test.tsx
@@ -32,6 +32,12 @@ jest.mock("../../../../common/vars", () => {
   };
 });
 
+jest.mock("@electron/remote", () => {
+  return {
+    getCurrentWindow: () => jest.fn(),
+  };
+});
+
 describe("<Tobar/> in Linux", () => {
   beforeEach(() => {
     TopBarRegistry.createInstance();
diff --git a/src/renderer/components/layout/__tests__/topbar-win.test.tsx b/src/renderer/components/layout/__tests__/topbar-win.test.tsx
index c3be9d2826..6a464f8bef 100644
--- a/src/renderer/components/layout/__tests__/topbar-win.test.tsx
+++ b/src/renderer/components/layout/__tests__/topbar-win.test.tsx
@@ -36,6 +36,23 @@ jest.mock("../../../../common/vars", () => {
   };
 });
 
+const mockMinimize = jest.fn();
+const mockMaximize = jest.fn();
+const mockUnmaximize = jest.fn();
+const mockClose = jest.fn();
+
+jest.mock("@electron/remote", () => {
+  return {
+    getCurrentWindow: () => ({
+      minimize: () => mockMinimize(),
+      maximize: () => mockMaximize(),
+      unmaximize: () => mockUnmaximize(),
+      close: () => mockClose(),
+      isMaximized: () => false,
+    }),
+  };
+});
+
 describe("<Tobar/> in Windows", () => {
   beforeEach(() => {
     TopBarRegistry.createInstance();
@@ -66,12 +83,12 @@ describe("<Tobar/> in Windows", () => {
     expect(broadcastMessage).toHaveBeenCalledWith(IpcMainWindowEvents.OPEN_CONTEXT_MENU);
 
     fireEvent.click(minimize);
-    expect(broadcastMessage).toHaveBeenCalledWith(IpcMainWindowEvents.MINIMIZE);
+    expect(mockMinimize).toHaveBeenCalled();
 
     fireEvent.click(maximize);
-    expect(broadcastMessage).toHaveBeenCalledWith(IpcMainWindowEvents.MAXIMIZE);
+    expect(mockMaximize).toHaveBeenCalled();
 
     fireEvent.click(close);
-    expect(broadcastMessage).toHaveBeenCalledWith(IpcMainWindowEvents.CLOSE);
+    expect(mockClose).toHaveBeenCalled();
   });
 });
diff --git a/src/renderer/components/layout/__tests__/topbar.test.tsx b/src/renderer/components/layout/__tests__/topbar.test.tsx
index 17a9655ec1..7f19c38d3f 100644
--- a/src/renderer/components/layout/__tests__/topbar.test.tsx
+++ b/src/renderer/components/layout/__tests__/topbar.test.tsx
@@ -65,6 +65,7 @@ jest.mock("@electron/remote", () => {
         }];
       },
     },
+    getCurrentWindow: () => jest.fn(),
   };
 });