From c4e42e1bc6183cb9a4af61da77284e50368edb60 Mon Sep 17 00:00:00 2001
From: Kanahathi Mohideen <kpmfazal@gmail.com>
Date: Mon, 6 Feb 2023 16:15:51 -0500
Subject: [PATCH] Add tests for VirtualizedTraceView

Signed-off-by: Kanahathi Mohideen <kpmfazal@gmail.com>
---
 .../VirtualizedTraceView.test.js              | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)

diff --git a/packages/jaeger-ui/src/components/TracePage/TraceTimelineViewer/VirtualizedTraceView.test.js b/packages/jaeger-ui/src/components/TracePage/TraceTimelineViewer/VirtualizedTraceView.test.js
index 62e6d1b773..f7fc949bcf 100644
--- a/packages/jaeger-ui/src/components/TracePage/TraceTimelineViewer/VirtualizedTraceView.test.js
+++ b/packages/jaeger-ui/src/components/TracePage/TraceTimelineViewer/VirtualizedTraceView.test.js
@@ -22,6 +22,7 @@ import { DEFAULT_HEIGHTS, VirtualizedTraceViewImpl } from './VirtualizedTraceVie
 import traceGenerator from '../../../demo/trace-generators';
 import transformTraceData from '../../../model/transform-trace-data';
 import updateUiFindSpy from '../../../utils/update-ui-find';
+import * as getLinks from '../../../model/link-patterns';
 
 jest.mock('./SpanTreeOffset');
 jest.mock('../../../utils/update-ui-find');
@@ -430,4 +431,22 @@ describe('<VirtualizedTraceViewImpl>', () => {
       expect(focusUiFindMatchesMock).toHaveBeenLastCalledWith(trace, spanName, false);
     });
   });
+
+  describe('linksGetter()', () => {
+    let getLinksSpy;
+
+    beforeAll(() => {
+      getLinksSpy = jest.spyOn(getLinks, 'default');
+    });
+
+    afterAll(() => {
+      getLinksSpy.mockRestore();
+    });
+
+    it('calls getLinks with expected params', () => {
+      const span = trace.spans[1];
+      instance.linksGetter(span, span.tags, 0);
+      expect(getLinksSpy).toHaveBeenCalledWith(span, span.tags, 0, trace);
+    });
+  });
 });