chore: add ut for s3 (#36)

Co-authored-by: GitHub Copilot <[email protected]>

Author: frankliu20

asset-manager/worker/src/test/java/com/microsoft/migration/assets/worker/service/S3FileProcessingServiceTest.java

@@ -0,0 +1,102 @@
+package com.microsoft.migration.assets.worker.service;
+
+import com.microsoft.migration.assets.worker.repository.ImageMetadataRepository;
+import org.junit.jupiter.api.BeforeEach;
+import org.junit.jupiter.api.Test;
+import org.junit.jupiter.api.extension.ExtendWith;
+import org.mockito.InjectMocks;
+import org.mockito.Mock;
+import org.mockito.junit.jupiter.MockitoExtension;
+import org.springframework.test.util.ReflectionTestUtils;
+import software.amazon.awssdk.core.ResponseInputStream;
+import software.amazon.awssdk.core.sync.RequestBody;
+import software.amazon.awssdk.services.s3.S3Client;
+import software.amazon.awssdk.services.s3.model.GetObjectRequest;
+import software.amazon.awssdk.services.s3.model.GetObjectResponse;
+import software.amazon.awssdk.services.s3.model.PutObjectRequest;
+
+import java.nio.file.Files;
+import java.nio.file.Path;
+import java.util.Collections;
+
+import static org.junit.jupiter.api.Assertions.assertEquals;
+import static org.mockito.ArgumentMatchers.any;
+import static org.mockito.Mockito.*;
+
+@ExtendWith(MockitoExtension.class)
+public class S3FileProcessingServiceTest {
+
+    @Mock
+    private S3Client s3Client;
+
+    @Mock
+    private ImageMetadataRepository imageMetadataRepository;
+
+    @InjectMocks
+    private S3FileProcessingService s3FileProcessingService;
+
+    private final String bucketName = "test-bucket";
+    private final String testKey = "test-image.jpg";
+    private final String thumbnailKey = "test-image_thumbnail.jpg";
+
+    @BeforeEach
+    void setUp() {
+        ReflectionTestUtils.setField(s3FileProcessingService, "bucketName", bucketName);
+    }
+
+    @Test
+    void getStorageTypeReturnsS3() {
+        // Act
+        String result = s3FileProcessingService.getStorageType();
+
+        // Assert
+        assertEquals("s3", result);
+    }
+
+    @Test
+    void downloadOriginalCopiesFileFromS3() throws Exception {
+        // Arrange
+        Path tempFile = Files.createTempFile("download-", ".tmp");
+        @SuppressWarnings("unchecked")
+        ResponseInputStream<GetObjectResponse> mockResponse = mock(ResponseInputStream.class);
+
+        when(s3Client.getObject(any(GetObjectRequest.class))).thenReturn(mockResponse);
+
+        // Act
+        s3FileProcessingService.downloadOriginal(testKey, tempFile);
+
+        // Assert
+        verify(s3Client).getObject(any(GetObjectRequest.class));
+
+        // Clean up
+        Files.deleteIfExists(tempFile);
+    }
+
+    @Test
+    void uploadThumbnailPutsFileToS3() throws Exception {
+        // Arrange
+        Path tempFile = Files.createTempFile("thumbnail-", ".tmp");
+        when(imageMetadataRepository.findAll()).thenReturn(Collections.emptyList());
+
+        // Act
+        s3FileProcessingService.uploadThumbnail(tempFile, thumbnailKey, "image/jpeg");
+
+        // Assert
+        verify(s3Client).putObject(any(PutObjectRequest.class), any(RequestBody.class));
+
+        // Clean up
+        Files.deleteIfExists(tempFile);
+    }
+
+    @Test
+    void testExtractOriginalKey() throws Exception {
+        // Use reflection to test private method
+        String result = (String) ReflectionTestUtils.invokeMethod(
+                s3FileProcessingService,
+                "extractOriginalKey",
+                "image_thumbnail.jpg");
+
+        // Assert
+        assertEquals("image.jpg", result);
+    }
+}