Test unmounting and calling with undefined

elastic · May 28, 2021 · 60063d0 · 60063d0
1 parent fb86497
commit 60063d0
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/x-pack/plugins/observability/public/components/shared/header_menu_portal.test.tsx b/x-pack/plugins/observability/public/components/shared/header_menu_portal.test.tsx
@@ -10,13 +10,17 @@ import React from 'react';
 import HeaderMenuPortal from './header_menu_portal';
 
 describe('HeaderMenuPortal', () => {
-  describe('render', () => {
-    it('sets the header action menu', () => {
+  describe('when unmounted', () => {
+    it('calls setHeaderActionMenu with undefined', () => {
       const setHeaderActionMenu = jest.fn();
 
-      render(<HeaderMenuPortal setHeaderActionMenu={setHeaderActionMenu}>test</HeaderMenuPortal>);
+      const { unmount } = render(
+        <HeaderMenuPortal setHeaderActionMenu={setHeaderActionMenu}>test</HeaderMenuPortal>
+      );
 
-      expect(setHeaderActionMenu).toHaveBeenCalled();
+      unmount();
+
+      expect(setHeaderActionMenu).toHaveBeenCalledWith(undefined);
     });
   });
 });