Test size of the blob response

camallo · May 9, 2022 · e720262 · e720262
1 parent e2104ec
commit e720262
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/tests/mock/blobs_download.rs b/tests/mock/blobs_download.rs
@@ -157,10 +157,12 @@ fn get_blobs_stream() -> Fallible<()> {
         .build()
         .unwrap();
 
-    let futcheck = dclient.get_blob_stream(&name, &digest);
 
-    let mut result_stream = runtime.block_on(futcheck)?;
-    let stream_output = result_stream.next().now_or_never();
+    let futcheck = dclient.get_blob_response(&name, &digest);
+
+    let blob_resp = runtime.block_on(futcheck)?;
+    assert_eq!(blob_resp.size(), Some(5));
+    let stream_output = blob_resp.stream().next().now_or_never();
     let output = stream_output.unwrap_or_else(|| panic!("No stream output"));
     let received_blob = output.unwrap_or_else(|| panic!("No blob data"))?;
     assert_eq!(blob.to_vec(), received_blob);