Test unmounting and calling with undefined

smith · smith · commit 60063d0fca8d · 2021-05-28T09:13:16.000-05:00
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);
     });
   });
 });
