From 832c3165ca0c223590315783ee25bfa14a81816e Mon Sep 17 00:00:00 2001
From: Everett <everett980@gmail.com>
Date: Tue, 10 Mar 2020 11:36:37 -0400
Subject: [PATCH] Fix span links for leading 0s trace ID (#539)

* Fix span links for leading 0s trace ID
* Add tests for deep linking leading 0s
Fix lookback test issue with dst

Signed-off-by: Everett Ross <reverett@uber.com>
---
 .../SearchTracePage/SearchForm.test.js        | 10 +++++--
 .../SearchTracePage/SearchResults/index.js    |  2 +-
 .../SearchResults/index.test.js               | 30 +++++++++++++++++++
 .../components/SearchTracePage/url.test.js    |  2 +-
 4 files changed, 39 insertions(+), 5 deletions(-)

diff --git a/packages/jaeger-ui/src/components/SearchTracePage/SearchForm.test.js b/packages/jaeger-ui/src/components/SearchTracePage/SearchForm.test.js
index 937592ab8a..e2fd2a9b90 100644
--- a/packages/jaeger-ui/src/components/SearchTracePage/SearchForm.test.js
+++ b/packages/jaeger-ui/src/components/SearchTracePage/SearchForm.test.js
@@ -148,9 +148,13 @@ describe('lookback utils', () => {
 
     it('creates timestamp for days ago', () => {
       [1, 2, 4, 7].forEach(lookbackNum => {
-        expect(nowInMicroseconds - lookbackToTimestamp(`${lookbackNum}d`, now)).toBe(
-          lookbackNum * 24 * hourInMicroseconds
-        );
+        const actual = nowInMicroseconds - lookbackToTimestamp(`${lookbackNum}d`, now);
+        const expected = lookbackNum * 24 * hourInMicroseconds;
+        try {
+          expect(actual).toBe(expected);
+        } catch (_e) {
+          expect(Math.abs(actual - expected)).toBe(hourInMicroseconds);
+        }
       });
     });
 
diff --git a/packages/jaeger-ui/src/components/SearchTracePage/SearchResults/index.js b/packages/jaeger-ui/src/components/SearchTracePage/SearchResults/index.js
index 9f4bfd467f..10fb11dc6b 100644
--- a/packages/jaeger-ui/src/components/SearchTracePage/SearchResults/index.js
+++ b/packages/jaeger-ui/src/components/SearchTracePage/SearchResults/index.js
@@ -208,7 +208,7 @@ export class UnconnectedSearchResults extends React.PureComponent<SearchResultsP
                   linkTo={getLocation(
                     trace.traceID,
                     { fromSearch: searchUrl },
-                    spanLinks && spanLinks[trace.traceID]
+                    spanLinks && (spanLinks[trace.traceID] || spanLinks[trace.traceID.replace(/^0*/, '')])
                   )}
                   toggleComparison={this.toggleComparison}
                   trace={trace}
diff --git a/packages/jaeger-ui/src/components/SearchTracePage/SearchResults/index.test.js b/packages/jaeger-ui/src/components/SearchTracePage/SearchResults/index.test.js
index 8f7fbc4c7d..c3dbd51faf 100644
--- a/packages/jaeger-ui/src/components/SearchTracePage/SearchResults/index.test.js
+++ b/packages/jaeger-ui/src/components/SearchTracePage/SearchResults/index.test.js
@@ -74,6 +74,36 @@ describe('<SearchResults>', () => {
       expect(wrapper.find(ResultItem).length).toBe(traces.length);
     });
 
+    it('deep links traces', () => {
+      const uiFind = 'ui-find';
+      const spanLinks = {
+        [traces[0].traceID]: uiFind,
+      };
+      wrapper.setProps({ spanLinks });
+      const results = wrapper.find(ResultItem);
+      expect(results.at(0).prop('linkTo').search).toBe(`uiFind=${uiFind}`);
+      expect(results.at(1).prop('linkTo').search).toBeUndefined();
+    });
+
+    it('deep links traces with leading 0', () => {
+      const uiFind0 = 'ui-find-0';
+      const uiFind1 = 'ui-find-1';
+      const traceID0 = '00traceID0';
+      const traceID1 = 'traceID1';
+      const spanLinks = {
+        [traceID0]: uiFind0,
+        [traceID1]: uiFind1,
+      };
+      const zeroIDTraces = [
+        { traceID: traceID0, spans: [], processes: {} },
+        { traceID: `000${traceID1}`, spans: [], processes: {} },
+      ];
+      wrapper.setProps({ spanLinks, traces: zeroIDTraces });
+      const results = wrapper.find(ResultItem);
+      expect(results.at(0).prop('linkTo').search).toBe(`uiFind=${uiFind0}`);
+      expect(results.at(1).prop('linkTo').search).toBe(`uiFind=${uiFind1}`);
+    });
+
     describe('ddg', () => {
       const searchParam = 'view';
       const viewDdg = 'ddg';
diff --git a/packages/jaeger-ui/src/components/SearchTracePage/url.test.js b/packages/jaeger-ui/src/components/SearchTracePage/url.test.js
index 0710255c2e..af417be216 100644
--- a/packages/jaeger-ui/src/components/SearchTracePage/url.test.js
+++ b/packages/jaeger-ui/src/components/SearchTracePage/url.test.js
@@ -45,7 +45,7 @@ describe('SearchTracePage/url', () => {
       ).toBe(`/search?traceID=${trace0}`);
     });
 
-    it('converts spanLink and traceID to traceID and span', () => {
+    it('converts spanLink and its traceID to just span', () => {
       expect(
         getUrl({
           traceID: trace0,