test uses data-test-subj instead of component reference

elastic · Jul 13, 2020 · 96e642e · 96e642e
1 parent b0b048b
commit 96e642e
Showing 1 changed file with 11 additions and 1 deletion.
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
@@ -18,6 +18,7 @@ import { Sort } from './sort';
 import { wait } from '../../../../common/lib/helpers';
 import { useMountAppended } from '../../../../common/utils/use_mount_appended';
 import { SELECTOR_TIMELINE_BODY_CLASS_NAME, TimelineBody } from '../styles';
+import { ReactWrapper } from '@elastic/eui/node_modules/@types/enzyme';
 
 const testBodyHeight = 700;
 const mockGetNotesByIds = (eventId: string[]) => [];
@@ -162,8 +163,17 @@ describe('Body', () => {
             <Body {...props} />
           </TestProviders>
         );
+
+        // The value returned if `wrapper.find` returns a `TimelineBody` instance.
+        type TimelineBodyEnzymeWrapper = ReactWrapper<React.ComponentProps<typeof TimelineBody>>;
+
+        // The first TimelineBody component
+        const timelineBody: TimelineBodyEnzymeWrapper = wrapper
+          .find('[data-test-subj="timeline-body"]')
+          .first() as TimelineBodyEnzymeWrapper;
+
         // the timeline body still renders, but it gets a `display: none` style via `styled-components`.
-        expect(wrapper.find(TimelineBody).props().visible).toBe(false);
+        expect(timelineBody.props().visible).toBe(false);
       });
     });
   });