diff options
Diffstat (limited to 'test/buildtool/execution_api')
-rw-r--r-- | test/buildtool/execution_api/bazel/bytestream_client.test.cpp | 12 |
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") { |