From aed4267870d91f7f161ea5dc1121f51d261d7a57 Mon Sep 17 00:00:00 2001
From: Dinika Saxena <dinikasaxenas@gmail.com>
Date: Tue, 17 Sep 2024 16:28:30 +0200
Subject: [PATCH] Load resource with tag/revision if it exists in self

Signed-off-by: Dinika Saxena <dinikasaxenas@gmail.com>
---
 .../containers/DataTableContainer.spec.tsx    | 14 ++++++++++-
 src/shared/containers/DataTableContainer.tsx  | 23 +++++++++++++++----
 2 files changed, 31 insertions(+), 6 deletions(-)

diff --git a/src/shared/containers/DataTableContainer.spec.tsx b/src/shared/containers/DataTableContainer.spec.tsx
index 676e71ad3..3879700b2 100644
--- a/src/shared/containers/DataTableContainer.spec.tsx
+++ b/src/shared/containers/DataTableContainer.spec.tsx
@@ -432,6 +432,7 @@ describe('DataTableContainer - Row Click', () => {
   let component: RenderResult;
   let nexus: ReturnType<typeof createNexusClient>;
   let nexusSpy: jest.SpyInstance;
+  let historySpy: jest.SpyInstance;
 
   beforeAll(() => {
     server = setupServer(
@@ -485,6 +486,7 @@ describe('DataTableContainer - Row Click', () => {
           ? Promise.resolve([getMockResource('doesnt-matter', {}, 'agents')])
           : Promise.resolve(getMockResource('doesnt-matter', {}, 'agents'))
       );
+    historySpy = jest.spyOn(history, 'push');
   });
 
   // reset any request handlers that are declared as a part of our tests
@@ -494,6 +496,7 @@ describe('DataTableContainer - Row Click', () => {
     queryClient.clear();
     localStorage.clear();
     nexusSpy.mockClear();
+    historySpy.mockClear();
   });
 
   afterAll(() => {
@@ -529,6 +532,8 @@ describe('DataTableContainer - Row Click', () => {
       path: `${selfWithRevision}&format=expanded`,
       headers: { Accept: 'application/json' },
     });
+    const navigateTo = historySpy.mock.calls[0][0];
+    expect(navigateTo).toContain('rev=30');
   });
 
   it('requests correct resource from delta when user clicks on row with tag in self', async () => {
@@ -547,11 +552,13 @@ describe('DataTableContainer - Row Click', () => {
       path: `${selfWithTag}&format=expanded`,
       headers: { Accept: 'application/json' },
     });
+    const navigateTo = historySpy.mock.calls[0][0];
+    expect(navigateTo).toContain('tag=30');
   });
 
   it('requests correct resource from delta when user clicks on row with tag and revision in self', async () => {
     const selfWithTagAndRev =
-      'https://localhost:3000/resources/bbp/agents/_/persons%2Fc3358e61-7650-4954-99b7-f7572cbf5d5g?tag=30&rev=2-';
+      'https://localhost:3000/resources/bbp/agents/_/persons%2Fc3358e61-7650-4954-99b7-f7572cbf5d5g?tag=30&rev=20';
 
     const resources = [getMockStudioResource('Malory', `${selfWithTagAndRev}`)];
 
@@ -565,6 +572,8 @@ describe('DataTableContainer - Row Click', () => {
       path: `${selfWithTagAndRev}&format=expanded`,
       headers: { Accept: 'application/json' },
     });
+    const navigateTo = historySpy.mock.calls[0][0];
+    expect(navigateTo).toContain('rev=20');
   });
 
   it('requests correct resource from delta when user clicks on row with no tag or revision in self', async () => {
@@ -585,6 +594,9 @@ describe('DataTableContainer - Row Click', () => {
       path: `${selfWithoutTagOrRev}?format=expanded`,
       headers: { Accept: 'application/json' },
     });
+    const navigateTo = historySpy.mock.calls[0][0];
+    expect(navigateTo).not.toContain('rev');
+    expect(navigateTo).not.toContain('tag');
   });
 });
 
diff --git a/src/shared/containers/DataTableContainer.tsx b/src/shared/containers/DataTableContainer.tsx
index 64bea81a9..5be1e24da 100644
--- a/src/shared/containers/DataTableContainer.tsx
+++ b/src/shared/containers/DataTableContainer.tsx
@@ -219,19 +219,32 @@ const DataTableContainer: React.FC<DataTableProps> = ({
         if (resource['@type'] === 'Project') {
           return;
         }
-        const url = new URL(selfUrl);
-        url.searchParams.set('format', 'expanded');
+        const resourceUrl = new URL(selfUrl);
+        resourceUrl.searchParams.set('format', 'expanded');
         nexus
           .httpGet({
-            path: `${url.toString()}`,
+            path: `${resourceUrl.toString()}`,
             headers: { Accept: 'application/json' },
           })
           .then((fullIdResponse: Resource) => {
             const [orgLabel, projectLabel] = parseProjectUrl(resource._project);
-            const hist = `/${orgLabel}/${projectLabel}/resources/${encodeURIComponent(
+
+            let fullResourceId = `/${orgLabel}/${projectLabel}/resources/${encodeURIComponent(
               fullIdResponse[0]['@id']
             )}`;
-            history.push(hist, { background: location });
+
+            const revision = resourceUrl.searchParams.get('rev');
+            const tag = resourceUrl.searchParams.get('tag');
+
+            if (revision) {
+              fullResourceId = `${fullResourceId}?rev=${revision}`;
+            } else if (tag) {
+              fullResourceId = `${fullResourceId}?tag=${tag}`;
+            }
+
+            history.push(`${fullResourceId.toString()}`, {
+              background: location,
+            });
           });
       })
       .catch(error => {