1 files changed, 10 insertions, 0 deletions
diff --git a/src/main/java/com/android/vts/entity/TestSuiteResultEntity.java b/src/main/java/com/android/vts/entity/TestSuiteResultEntity.java
index b3d24df..2e12f54 100644
--- a/src/main/java/com/android/vts/entity/TestSuiteResultEntity.java
+++ b/src/main/java/com/android/vts/entity/TestSuiteResultEntity.java
@@ -391,6 +391,16 @@ public class TestSuiteResultEntity {
         return deviceName;
     }
 
+    public String getScreenResultLogPath() {
+        String gcsBucketName = systemConfigProp.getProperty("gcs.bucketName");
+        return resultPath.replace("gs://" + gcsBucketName + "/", "");
+    }
+
+    public String getScreenInfraLogPath() {
+        String gcsInfraLogBucketName = systemConfigProp.getProperty("gcs.infraLogBucketName");
+        return infraLogPath.replace("gs://" + gcsInfraLogBucketName + "/", "");
+    }
+
     private String getNormalizedVersion(String fingerprint) {
         Map<String, Pattern> partternMap =
                 new HashMap<String, Pattern>() {