fix screenshots in steps (via #62)

eroshenkoam · Oct 17, 2023 · e727075 · e727075
1 parent a4f2ee6
commit e727075
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/src/main/java/io/eroshenkoam/xcresults/export/Allure2ExportFormatter.java b/src/main/java/io/eroshenkoam/xcresults/export/Allure2ExportFormatter.java
@@ -180,11 +180,12 @@ private void parseStep(final JsonNode activity,
             return;
         }
 
-        if (activity.has(ATTACHMENTS)) {
-            context.getCurrent().getAttachments().addAll(getAttachments(activity.get(ATTACHMENTS).get(VALUES)));
-        }
+        final Optional<List<Attachment>> attachments = Optional.ofNullable(activity.get(ATTACHMENTS))
+                .map(a -> a.get(VALUES))
+                .map(this::getAttachments);
         if (activityTitle.startsWith("Start Test at") && activity.has(ACTIVITY_START)) {
             context.getResult().setStart(parseDate(activity.get(ACTIVITY_START).get(VALUE).asText()));
+            attachments.ifPresent(context.getCurrent().getAttachments()::addAll);
             return;
         }
 
@@ -193,6 +194,7 @@ private void parseStep(final JsonNode activity,
                 .setStatus(Status.PASSED)
                 .setSteps(new ArrayList<>())
                 .setAttachments(new ArrayList<>());
+        attachments.ifPresent(step.getAttachments()::addAll);
 
         final boolean hasAssertionMessage = activityTitle.startsWith("Assertion Failure")
                 || activityTitle.contains("Test skipped");