ByteStreamClient: Return ArtifactBlob from Read.

1 files changed, 7 insertions, 5 deletions

diff --git a/test/buildtool/execution_api/bazel/bytestream_client.test.cpp b/test/buildtool/execution_api/bazel/bytestream_client.test.cpp
index f607bc3a..20e00aee 100644
--- a/test/buildtool/execution_api/bazel/bytestream_client.test.cpp
+++ b/test/buildtool/execution_api/bazel/bytestream_client.test.cpp
@@ -15,6 +15,7 @@
 #include "src/buildtool/execution_api/remote/bazel/bytestream_client.hpp"
 
 #include <cstddef>
+#include <memory>
 #include <optional>
 #include <string>
 
@@ -58,9 +59,9 @@ TEST_CASE("ByteStream Client: Transfer single blob", "[execution_api]") {
         ArtifactBlob const blob{digest, content, /*is_exec=*/false};
         CHECK(stream.Write(instance_name, blob));
 
-        auto const data = stream.Read(instance_name, digest);
-
-        CHECK(data == content);
+        auto const downloaded_blob = stream.Read(instance_name, digest);
+        REQUIRE(downloaded_blob.has_value());
+        CHECK(*downloaded_blob->data == content);
     }
 
     SECTION("Small blob with wrong digest") {
@@ -94,8 +95,9 @@ TEST_CASE("ByteStream Client: Transfer single blob", "[execution_api]") {
         CHECK(stream.Write(instance_name, blob));
 
         SECTION("Download large blob") {
-            auto const data = stream.Read(instance_name, digest);
-            CHECK(data == content);
+            auto const downloaded_blob = stream.Read(instance_name, digest);
+            REQUIRE(downloaded_blob.has_value());
+            CHECK(*downloaded_blob->data == content);
         }
 
         SECTION("Incrementally download large blob") {