From 611a6ca84f5d49c0aed3875b6b288a5c80c53ed4 Mon Sep 17 00:00:00 2001
From: Pyifan <Pyifan@users.noreply.github.com>
Date: Wed, 18 Dec 2024 14:22:25 +0800
Subject: [PATCH] Testplan result importer to best effort fix attachment path

---
 testplan/importers/testplan.py | 40 ++++++++++++++++++++++++++++++++++
 1 file changed, 40 insertions(+)

diff --git a/testplan/importers/testplan.py b/testplan/importers/testplan.py
index 1acaa9743..f1ce75a8b 100644
--- a/testplan/importers/testplan.py
+++ b/testplan/importers/testplan.py
@@ -1,9 +1,12 @@
 """
 Implements one-phase importer for Testplan JSON format.
 """
+import os
+import pathlib
 from typing import List
 
 from testplan.common.utils.json import json_loads
+from testplan.defaults import ATTACHMENTS
 from testplan.importers import ImportedResult, ResultImporter
 from testplan.report import ReportCategories, TestGroupReport, TestReport
 from testplan.report.testing.schemas import TestReportSchema
@@ -42,6 +45,43 @@ def import_result(self) -> ImportedResult:
         """ """
         with open(self.path) as fp:
             result_json = json_loads(fp.read())
+            self.fix_attachments_path(result_json)
             result = self.schema.load(result_json)
 
             return TestplanImportedResult(result)
+
+    def fix_attachments_path(
+        self,
+        report: dict,
+        attachment_dir: pathlib.Path = None,
+    ):
+        """
+        Best effort fix attachment path in case report.json and _attachments are copied around
+        """
+        attachment_dir = (
+            str(attachment_dir)
+            if attachment_dir
+            else os.path.join(os.path.dirname(self.path), ATTACHMENTS)
+        )
+        if report.get("attachments"):
+            for dst, src in report["attachments"].items():
+                if os.path.isfile(src):
+                    # attachment path is correct
+                    break
+                else:
+                    # attempt to fix attachment path
+                    alt_src = os.path.join(attachment_dir, dst)
+                    if os.path.isfile(alt_src):
+                        report["attachments"][dst] = alt_src
+
+        # recursively fix entries
+        def _fix_attachments_path(report):
+            if report.get("entries"):
+                for entry in report["entries"]:
+                    _fix_attachments_path(entry)
+            elif report.get("source_path") and report.get("dst_path"):
+                alt_src = os.path.join(attachment_dir, report["dst_path"])
+                report["source_path"] = alt_src
+
+        _fix_attachments_path(report)
+        return report