Fix mock tests

Former-commit-id: d52bd8f94e

src/test/java/auctiora/ImageProcessingServiceTest.java

@@ -19,14 +19,14 @@ class ImageProcessingServiceTest {
 
     private DatabaseService mockDb;
     private ObjectDetectionService mockDetector;
-    private RateLimitedHttpClient2 mockHttpClient;
+    private RateLimitedHttpClient mockHttpClient;
     private ImageProcessingService service;
 
     @BeforeEach
     void setUp() {
         mockDb = mock(DatabaseService.class);
         mockDetector = mock(ObjectDetectionService.class);
-        mockHttpClient = mock(RateLimitedHttpClient2.class);
+        mockHttpClient = mock(RateLimitedHttpClient.class);
         service = new ImageProcessingService(mockDb, mockDetector, mockHttpClient);
     }
 
@@ -102,6 +102,7 @@ class ImageProcessingServiceTest {
         ArgumentCaptor<Integer> lotIdCaptor = ArgumentCaptor.forClass(Integer.class);
         ArgumentCaptor<String> urlCaptor = ArgumentCaptor.forClass(String.class);
         ArgumentCaptor<String> filePathCaptor = ArgumentCaptor.forClass(String.class);
+        @SuppressWarnings("unchecked")
         ArgumentCaptor<List<String>> labelsCaptor = ArgumentCaptor.forClass(List.class);
 
         when(mockDetector.detectObjects(anyString()))
@@ -169,8 +170,8 @@ class ImageProcessingServiceTest {
     @DisplayName("Should handle database errors during image save")
     void testDatabaseErrorHandling() throws Exception {
         // Mock successful HTTP download
-        @SuppressWarnings("unchecked")
-        var mockResponse = mock(java.net.http.HttpResponse.class);
+        @SuppressWarnings({"unchecked", "rawtypes"})
+        var mockResponse = (java.net.http.HttpResponse<byte[]>) mock(java.net.http.HttpResponse.class);
         when(mockResponse.statusCode()).thenReturn(200);
         when(mockResponse.body()).thenReturn(new byte[]{1, 2, 3});
         when(mockHttpClient.sendGetBytes(anyString())).thenReturn(mockResponse);

src/test/java/auctiora/IntegrationTest.java

@@ -48,7 +48,7 @@ class IntegrationTest {
             "non_existent.txt"
         );
 
-        RateLimitedHttpClient2 httpClient = new RateLimitedHttpClient2();
+        RateLimitedHttpClient httpClient = new RateLimitedHttpClient();
         imageProcessor = new ImageProcessingService(db, detector, httpClient);
 
         monitor = new TroostwijkMonitor(