[Security Solution][Resolver]Enzyme test related events closing (elas…

…tic#74811) (elastic#74999)
Dosant · Aug 14, 2020 · d32d9fd · d32d9fd
1 parent 357e0e0
commit d32d9fd
Showing 1 changed file with 19 additions and 1 deletion.
diff --git a/x-pack/plugins/security_solution/public/resolver/view/clickthrough.test.tsx b/x-pack/plugins/security_solution/public/resolver/view/clickthrough.test.tsx
@@ -162,10 +162,28 @@ describe('Resolver, when analyzing a tree that has two related events for the or
           button.simulate('click');
         }
       });
-      it('should open the submenu', async () => {
+      it('should open the submenu and display exactly one option with the correct count', async () => {
         await expect(
           simulator.map(() => simulator.processNodeSubmenuItems().map((node) => node.text()))
         ).toYieldEqualTo(['2 registry']);
+        await expect(
+          simulator.map(() => simulator.processNodeSubmenuItems().length)
+        ).toYieldEqualTo(1);
+      });
+    });
+    describe('and when the related events button is clicked again', () => {
+      beforeEach(async () => {
+        const button = await simulator.resolveWrapper(() =>
+          simulator.processNodeRelatedEventButton(entityIDs.origin)
+        );
+        if (button) {
+          button.simulate('click');
+        }
+      });
+      it('should close the submenu', async () => {
+        await expect(
+          simulator.map(() => simulator.processNodeSubmenuItems().length)
+        ).toYieldEqualTo(0);
       });
     });
   });