Add unit test for TimelineCollapser

Signed-off-by: Davit Yeghshatyan <davo@uber.com>
jaegertracing · Jul 2, 2018 · e75c5de · e75c5de
1 parent 31ebe58
commit e75c5de
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/.../src/components/TracePage/TraceTimelineViewer/TimelineHeaderRow/TimelineHeaderRow.test.js b/.../src/components/TracePage/TraceTimelineViewer/TimelineHeaderRow/TimelineHeaderRow.test.js
@@ -19,6 +19,7 @@ import TimelineHeaderRow from './TimelineHeaderRow';
 import TimelineColumnResizer from './TimelineColumnResizer';
 import TimelineViewingLayer from './TimelineViewingLayer';
 import Ticks from '../Ticks';
+import TimelineCollapser from './TimelineCollapser';
 
 describe('<TimelineHeaderRow>', () => {
   let wrapper;
@@ -28,7 +29,11 @@ describe('<TimelineHeaderRow>', () => {
     nameColumnWidth,
     duration: 1234,
     numTicks: 5,
+    onCollapseAll: () => {},
+    onCollapseOne: () => {},
     onColummWidthChange: () => {},
+    onExpandAll: () => {},
+    onExpandOne: () => {},
     updateNextViewRangeTime: () => {},
     updateViewRangeTime: () => {},
     viewRangeTime: {
@@ -92,4 +97,16 @@ describe('<TimelineHeaderRow>', () => {
     );
     expect(wrapper.containsMatchingElement(elm)).toBe(true);
   });
+
+  it('renders the TimelineCollapser', () => {
+    const elm = (
+      <TimelineCollapser
+        onCollapseAll={props.onCollapseAll}
+        onExpandAll={props.onExpandAll}
+        onCollapseOne={props.onCollapseOne}
+        onExpandOne={props.onExpandOne}
+      />
+    );
+    expect(wrapper.containsMatchingElement(elm)).toBe(true);
+  });
 });