feat: Added logic for handling unavailable images

Refs: #307
SchweizerischeBundesbahnen · Dec 19, 2024 · cd69c6d · cd69c6d
1 parent 32ce943
commit cd69c6d
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 1 deletion.
diff --git a/src/main/java/ch/sbb/polarion/extension/pdf_exporter/rest/model/conversion/ExportParams.java b/src/main/java/ch/sbb/polarion/extension/pdf_exporter/rest/model/conversion/ExportParams.java
@@ -118,7 +118,7 @@ public int getUnavailableWorkItemAttachmentsCount() {
     public List<String> getWorkItemsIdContainUnavailableAttachments() {
         return unavailableWorkItemAttachments.stream()
                 .distinct()
-                .collect(Collectors.toList());
+                .toList();
     }
 
     public @NotNull DocumentType getDocumentType() {

diff --git a/...java/ch/sbb/polarion/extension/pdf_exporter/util/PdfExporterFileResourceProviderTest.java b/...java/ch/sbb/polarion/extension/pdf_exporter/util/PdfExporterFileResourceProviderTest.java
@@ -119,6 +119,20 @@ void getResourceAsBytesExceptionHandling() {
         assertTrue(unavailableAttachments.isEmpty());
     }
 
+    @Test
+    void getWorkItemIdFromAttachmentUrlValidUrl() {
+        String url = "http://localhost/polarion/wi-attachment/elibrary/EL-14852/attachment.png";
+        String result = resourceProvider.getWorkItemIdFromAttachmentUrl(url);
+        assertEquals("EL-14852", result);
+    }
+
+    @Test
+    void getWorkItemIdFromAttachmentUrl_InvalidUrl() {
+        String url = "http://example.com/invalid/url";
+        String result = resourceProvider.getWorkItemIdFromAttachmentUrl(url);
+        assertNull(result);
+    }
+
     @Test
     void isMediaTypeMismatchMatchingMimeTypes() {
         String resource = "image.png";