From 80acdce41b5ff9bf4c0b4602d646a250bfaede85 Mon Sep 17 00:00:00 2001 From: Maksim Denisov Date: Wed, 29 Jan 2025 14:59:19 +0100 Subject: BazelCasClient: Use ArtifactDigest in BatchReadBlobs ...and drop testing of invalid digests since ArtifactDigest always guarantees to hold a valid digest. --- .../execution_api/bazel/bazel_cas_client.test.cpp | 17 +++++++++-------- 1 file changed, 9 insertions(+), 8 deletions(-) (limited to 'test/buildtool/execution_api/bazel/bazel_cas_client.test.cpp') diff --git a/test/buildtool/execution_api/bazel/bazel_cas_client.test.cpp b/test/buildtool/execution_api/bazel/bazel_cas_client.test.cpp index 4054b32a..ad2e0678 100644 --- a/test/buildtool/execution_api/bazel/bazel_cas_client.test.cpp +++ b/test/buildtool/execution_api/bazel/bazel_cas_client.test.cpp @@ -22,6 +22,8 @@ #include "catch2/catch_test_macros.hpp" #include "gsl/gsl" +#include "src/buildtool/common/artifact_digest.hpp" +#include "src/buildtool/common/artifact_digest_factory.hpp" #include "src/buildtool/common/bazel_digest_factory.hpp" #include "src/buildtool/common/bazel_types.hpp" #include "src/buildtool/common/remote/remote_common.hpp" @@ -55,14 +57,17 @@ TEST_CASE("Bazel internals: CAS Client", "[execution_api]") { SECTION("Valid digest and blob") { // digest of "test" HashFunction const hash_function{TestHashType::ReadFromEnvironment()}; - auto digest = BazelDigestFactory::HashDataAs( + auto const digest = ArtifactDigestFactory::HashDataAs( + hash_function, content); + auto bazel_digest = BazelDigestFactory::HashDataAs( hash_function, content); // Valid blob - BazelBlob blob{digest, content, /*is_exec=*/false}; + BazelBlob blob{bazel_digest, content, /*is_exec=*/false}; // Search blob via digest - auto digests = cas_client.FindMissingBlobs(instance_name, {digest}); + auto digests = + cas_client.FindMissingBlobs(instance_name, {bazel_digest}); CHECK(digests.size() <= 1); if (not digests.empty()) { @@ -73,7 +78,7 @@ TEST_CASE("Bazel internals: CAS Client", "[execution_api]") { // Read blob auto blobs = cas_client.BatchReadBlobs(instance_name, {digest}); REQUIRE(blobs.size() == 1); - CHECK(std::equal_to{}(blobs.begin()->digest, digest)); + CHECK(blobs.begin()->digest == digest); CHECK(*blobs.begin()->data == content); } @@ -93,9 +98,5 @@ TEST_CASE("Bazel internals: CAS Client", "[execution_api]") { // Try upload faulty blob CHECK(cas_client.BatchUpdateBlobs(instance_name, {faulty_blob}) == 0U); - - // Read blob via faulty digest - CHECK( - cas_client.BatchReadBlobs(instance_name, {faulty_digest}).empty()); } } -- cgit v1.2.3