coverage for gcs and s3

Author: lcian

objectstore-service/src/backend/gcs.rs

@@ -1785,4 +1785,46 @@ mod tests {
 
         Ok(())
     }
+
+    #[tokio::test]
+    async fn test_range_request() -> Result<()> {
+        let backend = create_test_backend().await?;
+
+        let id = make_id();
+        let metadata = Metadata::default();
+        backend
+            .put_object(&id, &metadata, stream::single("hello, world"))
+            .await?;
+
+        let range = ByteRange::Bounded(0, 4);
+        let (_, content_range, body) = backend.get_object(&id, Some(range)).await?.unwrap();
+        let payload = stream::read_to_vec(body).await?;
+
+        assert_eq!(payload, b"hello");
+        assert_eq!(content_range.start, 0);
+        assert_eq!(content_range.end, 4);
+        assert_eq!(content_range.total, 12);
+
+        Ok(())
+    }
+
+    #[tokio::test]
+    async fn test_range_unsatisfiable() -> Result<()> {
+        let backend = create_test_backend().await?;
+
+        let id = make_id();
+        let metadata = Metadata::default();
+        backend
+            .put_object(&id, &metadata, stream::single("hello"))
+            .await?;
+
+        let range = ByteRange::Bounded(100, 200);
+        let result = backend.get_object(&id, Some(range)).await;
+        assert!(
+            matches!(result, Err(Error::RangeNotSatisfiable { total }) if total == 5),
+            "expected RangeNotSatisfiable with total=5"
+        );
+
+        Ok(())
+    }
 }

objectstore-service/src/backend/s3_compatible.rs

@@ -416,4 +416,46 @@ mod tests {
         assert!(result.is_none());
         Ok(())
     }
+
+    #[tokio::test]
+    async fn test_range_request() -> Result<()> {
+        let backend = create_test_backend();
+
+        let id = make_id();
+        let metadata = Metadata::default();
+        backend
+            .put_object(&id, &metadata, stream::single("hello, world"))
+            .await?;
+
+        let range = ByteRange::Bounded(0, 4);
+        let (_, content_range, body) = backend.get_object(&id, Some(range)).await?.unwrap();
+        let payload = stream::read_to_vec(body).await?;
+
+        assert_eq!(payload, b"hello");
+        assert_eq!(content_range.start, 0);
+        assert_eq!(content_range.end, 4);
+        assert_eq!(content_range.total, 12);
+
+        Ok(())
+    }
+
+    #[tokio::test]
+    async fn test_range_unsatisfiable() -> Result<()> {
+        let backend = create_test_backend();
+
+        let id = make_id();
+        let metadata = Metadata::default();
+        backend
+            .put_object(&id, &metadata, stream::single("hello"))
+            .await?;
+
+        let range = ByteRange::Bounded(100, 200);
+        let result = backend.get_object(&id, Some(range)).await;
+        assert!(
+            matches!(result, Err(Error::RangeNotSatisfiable { total }) if total == 5),
+            "expected RangeNotSatisfiable with total=5"
+        );
+
+        Ok(())
+    }
 }