From 78fc533d28a0c90e9fbe27c09c1f697ef687624c Mon Sep 17 00:00:00 2001
From: Patryk Kopycinski <contact@patrykkopycinski.com>
Date: Tue, 2 Jun 2020 09:19:30 +0200
Subject: [PATCH] Add unit tests

---
 .../timeline/properties/index.test.tsx        | 56 +++++++++++++++++++
 1 file changed, 56 insertions(+)

diff --git a/x-pack/plugins/siem/public/timelines/components/timeline/properties/index.test.tsx b/x-pack/plugins/siem/public/timelines/components/timeline/properties/index.test.tsx
index a8fb9110babe7..8bdec78ec8da2 100644
--- a/x-pack/plugins/siem/public/timelines/components/timeline/properties/index.test.tsx
+++ b/x-pack/plugins/siem/public/timelines/components/timeline/properties/index.test.tsx
@@ -26,6 +26,24 @@ jest.mock('../../../../common/components/utils');
   width: mockedWidth,
 }));
 
+jest.mock('react-redux', () => {
+  const originalModule = jest.requireActual('react-redux');
+
+  return {
+    ...originalModule,
+    useSelector: jest.fn().mockReturnValue({ savedObjectId: '1' }),
+  };
+});
+
+jest.mock('react-router-dom', () => {
+  const originalModule = jest.requireActual('react-router-dom');
+
+  return {
+    ...originalModule,
+    useHistory: jest.fn(),
+  };
+});
+
 describe('Properties', () => {
   const usersViewing = ['elastic'];
 
@@ -62,7 +80,45 @@ describe('Properties', () => {
         />
       </ReduxStoreProvider>
     );
+
+    wrapper.find('[data-test-subj="settings-gear"]').at(0).simulate('click');
+
     expect(wrapper.find('[data-test-subj="timeline-properties"]').exists()).toEqual(true);
+    expect(wrapper.find('button[data-test-subj="attach-timeline-case"]').prop('disabled')).toEqual(
+      false
+    );
+  });
+
+  test('renders correctly draft timeline', () => {
+    const wrapper = mount(
+      <ReduxStoreProvider store={store}>
+        <Properties
+          associateNote={jest.fn()}
+          createTimeline={jest.fn()}
+          isDataInTimeline={false}
+          isDatepickerLocked={false}
+          isFavorite={false}
+          title=""
+          description=""
+          getNotesByIds={jest.fn()}
+          noteIds={[]}
+          status={TimelineStatus.draft}
+          timelineId="abc"
+          toggleLock={jest.fn()}
+          updateDescription={jest.fn()}
+          updateIsFavorite={jest.fn()}
+          updateTitle={jest.fn()}
+          updateNote={jest.fn()}
+          usersViewing={usersViewing}
+        />
+      </ReduxStoreProvider>
+    );
+
+    wrapper.find('[data-test-subj="settings-gear"]').at(0).simulate('click');
+
+    expect(wrapper.find('button[data-test-subj="attach-timeline-case"]').prop('disabled')).toEqual(
+      true
+    );
   });
 
   test('it renders an empty star icon when it is NOT a favorite', () => {