[Security Solution] Enable investigate in timeline for all events, no…

…t just signals (#119375) (#119555) * Enable investigate in timeline for all events, not just signals * Update mocks * Lint Co-authored-by: Kevin Qualters <[email protected]>
elastic · Nov 24, 2021 · d3a15d0 · d3a15d0
1 parent 419c311
commit d3a15d0
Show file tree

Hide file tree

Showing 4 changed files with 14 additions and 4 deletions.
diff --git a/x-pack/plugins/security_solution/public/hosts/pages/navigation/events_query_tab_body.tsx b/x-pack/plugins/security_solution/public/hosts/pages/navigation/events_query_tab_body.tsx
@@ -70,7 +70,7 @@ const EventsQueryTabBodyComponent: React.FC<HostsComponentsQueryProps> = ({
 }) => {
   const dispatch = useDispatch();
   const { globalFullScreen } = useGlobalFullScreen();
-  const ACTION_BUTTON_COUNT = 3;
+  const ACTION_BUTTON_COUNT = 4;
   const tGridEnabled = useIsExperimentalFeatureEnabled('tGridEnabled');
 
   useEffect(() => {

diff --git a/x-pack/plugins/security_solution/public/timelines/components/timeline/body/actions/index.tsx b/x-pack/plugins/security_solution/public/timelines/components/timeline/body/actions/index.tsx
@@ -165,7 +165,7 @@ const ActionsComponent: React.FC<ActionProps> = ({
         </EventsTdContent>
       </div>
       <>
-        {timelineId !== TimelineId.active && eventType === 'signal' && (
+        {timelineId !== TimelineId.active && (
           <InvestigateInTimelineAction
             ariaLabel={i18n.SEND_ALERT_TO_TIMELINE_FOR_ROW({ ariaRowindex, columnValues })}
             key="investigate-in-timeline"

diff --git a/...rity_solution/public/timelines/components/timeline/body/events/event_column_view.test.tsx b/...rity_solution/public/timelines/components/timeline/body/events/event_column_view.test.tsx
@@ -23,11 +23,18 @@ import { getActionsColumnWidth } from '../../../../../../../timelines/public';
 
 jest.mock('../../../../../common/hooks/use_experimental_features');
 const useIsExperimentalFeatureEnabledMock = useIsExperimentalFeatureEnabled as jest.Mock;
-jest.mock('../../../../../common/hooks/use_selector');
+jest.mock('../../../../../common/hooks/use_selector', () => ({
+  useShallowEqualSelector: jest.fn(),
+  useDeepEqualSelector: jest.fn(),
+}));
 jest.mock('../../../../../common/lib/kibana', () => ({
   useKibana: () => ({
     services: {
       timelines: { ...mockTimelines },
+      data: {
+        search: jest.fn(),
+        query: jest.fn(),
+      },
       application: {
         capabilities: {
           siem: { crud_alerts: true, read_alerts: true },
@@ -146,7 +153,6 @@ describe('EventColumnView', () => {
     const wrapper = mount(
       <EventColumnView
         {...props}
-        timelineId="timeline-test"
         leadingControlColumns={[testLeadingControlColumn, ...leadingControlColumns]}
       />,
       {

diff --git a/x-pack/plugins/security_solution/public/timelines/components/timeline/body/index.test.tsx b/x-pack/plugins/security_solution/public/timelines/components/timeline/body/index.test.tsx
@@ -41,6 +41,10 @@ jest.mock('../../../../common/lib/kibana', () => {
             siem: { crud_alerts: true, read_alerts: true },
           },
         },
+        data: {
+          search: jest.fn(),
+          query: jest.fn(),
+        },
         uiSettings: {
           get: jest.fn(),
         },