From 25d0c901f666ce20e4da293a091e591e0bde076e Mon Sep 17 00:00:00 2001
From: patrykkopycinski <patryk.kopycinski@elastic.co>
Date: Tue, 4 Feb 2020 17:28:09 +0100
Subject: [PATCH] [SIEM] Enable flow_target_select_connected unit tests
 (#55618) (#56738)

---
 .../index.test.tsx                            | 22 ++++++++++++++-----
 .../flow_target_select_connected/index.tsx    |  2 +-
 2 files changed, 17 insertions(+), 7 deletions(-)

diff --git a/x-pack/legacy/plugins/siem/public/components/page/network/flow_target_select_connected/index.test.tsx b/x-pack/legacy/plugins/siem/public/components/page/network/flow_target_select_connected/index.test.tsx
index 006587d8fc294..e71be5a51e505 100644
--- a/x-pack/legacy/plugins/siem/public/components/page/network/flow_target_select_connected/index.test.tsx
+++ b/x-pack/legacy/plugins/siem/public/components/page/network/flow_target_select_connected/index.test.tsx
@@ -6,27 +6,37 @@
 
 import { mount } from 'enzyme';
 import React from 'react';
+import { MemoryRouter } from 'react-router-dom';
 
 import { TestProviders } from '../../../../mock';
-import { FlowTargetSelectConnected } from './index';
+import { FlowTargetSelectConnectedComponent } from './index';
 import { FlowTarget } from '../../../../graphql/types';
 
-describe.skip('Flow Target Select Connected', () => {
+describe('Flow Target Select Connected', () => {
   test('renders correctly against snapshot flowTarget source', () => {
     const wrapper = mount(
       <TestProviders>
-        <FlowTargetSelectConnected flowTarget={FlowTarget.source} />
+        <MemoryRouter>
+          <FlowTargetSelectConnectedComponent flowTarget={FlowTarget.source} />
+        </MemoryRouter>
       </TestProviders>
     );
-    expect(wrapper.find('FlowTargetSelectConnected')).toMatchSnapshot();
+    expect(wrapper.find('Memo(FlowTargetSelectComponent)').prop('selectedTarget')).toEqual(
+      FlowTarget.source
+    );
   });
 
   test('renders correctly against snapshot flowTarget destination', () => {
     const wrapper = mount(
       <TestProviders>
-        <FlowTargetSelectConnected flowTarget={FlowTarget.destination} />
+        <MemoryRouter>
+          <FlowTargetSelectConnectedComponent flowTarget={FlowTarget.destination} />
+        </MemoryRouter>
       </TestProviders>
     );
-    expect(wrapper.find('FlowTargetSelectConnected')).toMatchSnapshot();
+
+    expect(wrapper.find('Memo(FlowTargetSelectComponent)').prop('selectedTarget')).toEqual(
+      FlowTarget.destination
+    );
   });
 });
diff --git a/x-pack/legacy/plugins/siem/public/components/page/network/flow_target_select_connected/index.tsx b/x-pack/legacy/plugins/siem/public/components/page/network/flow_target_select_connected/index.tsx
index 1b87c36902159..2651c31e0a2c9 100644
--- a/x-pack/legacy/plugins/siem/public/components/page/network/flow_target_select_connected/index.tsx
+++ b/x-pack/legacy/plugins/siem/public/components/page/network/flow_target_select_connected/index.tsx
@@ -36,7 +36,7 @@ const getUpdatedFlowTargetPath = (
   return `${newPathame}${location.search}`;
 };
 
-const FlowTargetSelectConnectedComponent: React.FC<Props> = ({ flowTarget }) => {
+export const FlowTargetSelectConnectedComponent: React.FC<Props> = ({ flowTarget }) => {
   const history = useHistory();
   const location = useLocation();