tests: offline test

Signed-off-by: Alexandre Milesi <[email protected]>

Author: milesial

lib/llm/src/preprocessor/media/decoders/image.rs

@@ -25,8 +25,11 @@ impl Decoder for ImageDecoder {
         let n_channels = img.color().channel_count();
 
         let max_pixels = self.max_pixels.unwrap_or(usize::MAX);
+        let pixel_count = (width as usize)
+            .checked_mul(height as usize)
+            .ok_or_else(|| anyhow::anyhow!("Image dimensions {width}x{height} overflow usize"))?;
         anyhow::ensure!(
-            (width as usize) * (height as usize) <= max_pixels,
+            pixel_count <= max_pixels,
             "Image dimensions {width}x{height} exceed max pixels {max_pixels}"
         );
         let data = match n_channels {

lib/llm/src/preprocessor/media/loader.rs

@@ -120,15 +120,30 @@ mod tests {
     use super::*;
     use dynamo_async_openai::types::{ChatCompletionRequestMessageContentPartImage, ImageUrl};
 
-    // warning: non-airgap test
     #[tokio::test]
     async fn test_fetch_and_decode() {
+        let test_image_bytes =
+            include_bytes!("../../../tests/data/media/llm-optimize-deploy-graphic.png");
+
+        let mut server = mockito::Server::new_async().await;
+        let mock = server
+            .mock("GET", "/llm-optimize-deploy-graphic.png")
+            .with_status(200)
+            .with_header("content-type", "image/png")
+            .with_body(&test_image_bytes[..])
+            .create_async()
+            .await;
+
         let media_decoder = MediaDecoder::default();
-        let loader = MediaLoader::new(media_decoder, MediaFetcher::default()).unwrap();
+        let fetcher = MediaFetcher {
+            allow_direct_ip: true,
+            allow_direct_port: true,
+            ..Default::default()
+        };
 
-        let image_url = ImageUrl::from(
-            "https://developer-blogs.nvidia.com/wp-content/uploads/2023/11/llm-optimize-deploy-graphic.png",
-        );
+        let loader = MediaLoader::new(media_decoder, fetcher).unwrap();
+
+        let image_url = ImageUrl::from(format!("{}/llm-optimize-deploy-graphic.png", server.url()));
         let content_part = ChatCompletionRequestUserMessageContentPart::ImageUrl(
             ChatCompletionRequestMessageContentPartImage { image_url },
         );
@@ -145,10 +160,12 @@ mod tests {
 
         // Verify image dimensions: 1,999px × 1,125px (width × height)
         // Shape format is [height, width, channels]
-        assert!(!data.shape.is_empty(), "Shape should not be empty");
+        assert_eq!(data.shape.len(), 3);
         assert_eq!(data.shape[0], 1125, "Height should be 1125");
         assert_eq!(data.shape[1], 1999, "Width should be 1999");
         assert_eq!(data.shape[2], 4, "RGBA channels should be 4");
+
+        mock.assert_async().await;
     }
 
     #[test]

lib/llm/tests/data/media/.gitattributes

@@ -0,0 +1 @@
+llm-optimize-deploy-graphic.png filter=lfs diff=lfs merge=lfs -text