From 72e7a876840642ad8ae7295abcd86e46a537fe3a Mon Sep 17 00:00:00 2001
From: Jeremy Guarini <jguarini@paloaltonetworks.com>
Date: Wed, 27 Oct 2021 09:36:30 -0700
Subject: [PATCH] add more testing for FileExplorer tests

Signed-off-by: Jeremy Guarini <jguarini@paloaltonetworks.com>
---
 .../FileExplorer/FileExplorer.test.tsx        | 144 +++++++++++++++++-
 .../components/FileExplorer/FileExplorer.tsx  |   2 +-
 2 files changed, 144 insertions(+), 2 deletions(-)

diff --git a/plugins/code-coverage/src/components/FileExplorer/FileExplorer.test.tsx b/plugins/code-coverage/src/components/FileExplorer/FileExplorer.test.tsx
index 948d9c9722..4163b22ac1 100644
--- a/plugins/code-coverage/src/components/FileExplorer/FileExplorer.test.tsx
+++ b/plugins/code-coverage/src/components/FileExplorer/FileExplorer.test.tsx
@@ -13,7 +13,7 @@
  * See the License for the specific language governing permissions and
  * limitations under the License.
  */
-import { groupByPath } from './FileExplorer';
+import { groupByPath, buildFileStructure } from './FileExplorer';
 
 const dummyFiles = [
   {
@@ -40,6 +40,22 @@ const dummyFiles = [
     tracked: 1,
     path: '',
   },
+  {
+    filename: 'dir3/dir2/dir1/file4',
+    files: [],
+    coverage: 1,
+    missing: 1,
+    tracked: 1,
+    path: '',
+  },
+  {
+    filename: 'dir3/dir2/dir3/file4',
+    files: [],
+    coverage: 1,
+    missing: 1,
+    tracked: 1,
+    path: '',
+  },
 ];
 
 const dummyDataGroupedByPath = {
@@ -71,6 +87,124 @@ const dummyDataGroupedByPath = {
       path: '',
     },
   ],
+  dir3: [
+    {
+      filename: 'dir3/dir2/dir1/file4',
+      files: [],
+      coverage: 1,
+      missing: 1,
+      tracked: 1,
+      path: '',
+    },
+    {
+      filename: 'dir3/dir2/dir3/file4',
+      files: [],
+      coverage: 1,
+      missing: 1,
+      tracked: 1,
+      path: '',
+    },
+  ],
+};
+
+const coverageTableRow = {
+  files: dummyFiles,
+  coverage: 1,
+  missing: 1,
+  tracked: 1,
+  path: '',
+};
+
+const coverageTableRowResults = {
+  files: [
+    {
+      path: 'dir1',
+      files: [
+        {
+          path: 'file1',
+          files: [],
+          coverage: 1,
+          missing: 1,
+          tracked: 1,
+        },
+        {
+          path: 'file2',
+          files: [],
+          coverage: 1,
+          missing: 1,
+          tracked: 1,
+        },
+      ],
+      coverage: 1,
+      missing: 2,
+      tracked: 2,
+    },
+    {
+      path: 'dir2',
+      files: [
+        {
+          path: 'file3',
+          files: [],
+          coverage: 1,
+          missing: 1,
+          tracked: 1,
+        },
+      ],
+      coverage: 1,
+      missing: 1,
+      tracked: 1,
+    },
+    {
+      path: 'dir3',
+      files: [
+        {
+          path: 'dir2',
+          files: [
+            {
+              path: 'dir1',
+              files: [
+                {
+                  path: 'file4',
+                  files: [],
+                  coverage: 1,
+                  missing: 1,
+                  tracked: 1,
+                },
+              ],
+              coverage: 1,
+              missing: 1,
+              tracked: 1,
+            },
+            {
+              path: 'dir3',
+              files: [
+                {
+                  path: 'file4',
+                  files: [],
+                  coverage: 1,
+                  missing: 1,
+                  tracked: 1,
+                },
+              ],
+              coverage: 1,
+              missing: 1,
+              tracked: 1,
+            },
+          ],
+          coverage: 1,
+          missing: 2,
+          tracked: 2,
+        },
+      ],
+      coverage: 1,
+      missing: 2,
+      tracked: 2,
+    },
+  ],
+  coverage: 1,
+  missing: 1,
+  tracked: 1,
+  path: '',
 };
 
 describe('groupByPath function', () => {
@@ -78,3 +212,11 @@ describe('groupByPath function', () => {
     expect(groupByPath(dummyFiles)).toStrictEqual(dummyDataGroupedByPath);
   });
 });
+
+describe('buildFileStructure function', () => {
+  it('should group files by their root directory,as per their filename', () => {
+    expect(buildFileStructure(coverageTableRow)).toStrictEqual(
+      coverageTableRowResults,
+    );
+  });
+});
diff --git a/plugins/code-coverage/src/components/FileExplorer/FileExplorer.tsx b/plugins/code-coverage/src/components/FileExplorer/FileExplorer.tsx
index ed31cb9c62..49379551ad 100644
--- a/plugins/code-coverage/src/components/FileExplorer/FileExplorer.tsx
+++ b/plugins/code-coverage/src/components/FileExplorer/FileExplorer.tsx
@@ -82,7 +82,7 @@ const removeVisitedPathGroup = (
   });
 };
 
-const buildFileStructure = (row: CoverageTableRow) => {
+export const buildFileStructure = (row: CoverageTableRow) => {
   const dataGroupedByPath: FileStructureObject = groupByPath(row.files);
   row.files = Object.keys(dataGroupedByPath).map(pathGroup => {
     return buildFileStructure({