1 files changed, 73 insertions, 0 deletions
diff --git a/test/buildtool/execution_api/bazel/bazel_cas_client.test.cpp b/test/buildtool/execution_api/bazel/bazel_cas_client.test.cpp
new file mode 100644
index 00000000..30866ffe
--- /dev/null
+++ b/test/buildtool/execution_api/bazel/bazel_cas_client.test.cpp
@@ -0,0 +1,73 @@
+#include <string>
+
+#include "catch2/catch.hpp"
+#include "src/buildtool/execution_api/remote/bazel/bazel_cas_client.hpp"
+#include "src/buildtool/execution_api/remote/bazel/bazel_execution_client.hpp"
+#include "src/buildtool/execution_api/remote/config.hpp"
+
+TEST_CASE("Bazel internals: CAS Client", "[execution_api]") {
+    auto const& info = RemoteExecutionConfig::Instance();
+
+    std::string instance_name{"remote-execution"};
+    std::string content("test");
+
+    // Create CAS client
+    BazelCasClient cas_client(info.Host(), info.Port());
+
+    SECTION("Valid digest and blob") {
+        // digest of "test"
+        auto digest = ArtifactDigest::Create(content);
+
+        // Valid blob
+        BazelBlob blob{digest, content};
+
+        // Search blob via digest
+        auto digests = cas_client.FindMissingBlobs(instance_name, {digest});
+        CHECK(digests.size() <= 1);
+
+        if (!digests.empty()) {
+            // Upload blob, if not found
+            std::vector<BazelBlob> to_upload{blob};
+            CHECK(cas_client
+                      .BatchUpdateBlobs(
+                          instance_name, to_upload.begin(), to_upload.end())
+                      .size() == 1);
+        }
+
+        // Read blob
+        std::vector<bazel_re::Digest> to_read{digest};
+        auto blobs = cas_client.BatchReadBlobs(
+            instance_name, to_read.begin(), to_read.end());
+        REQUIRE(blobs.size() == 1);
+        CHECK(std::equal_to<bazel_re::Digest>{}(blobs[0].digest, digest));
+        CHECK(blobs[0].data == content);
+    }
+
+    SECTION("Invalid digest and blob") {
+        // Faulty digest
+        bazel_re::Digest faulty_digest{};
+        faulty_digest.set_hash(
+            "0123456789abcdef0123456789abcdef0123456789abcdef0123456789abcdef");
+        faulty_digest.set_size_bytes(4);
+
+        // Faulty blob
+        BazelBlob faulty_blob{faulty_digest, content};
+
+        // Search faulty digest
+        CHECK(cas_client.FindMissingBlobs(instance_name, {faulty_digest})
+                  .size() == 1);
+
+        // Try upload faulty blob
+        std::vector<BazelBlob> to_upload{faulty_blob};
+        CHECK(cas_client
+                  .BatchUpdateBlobs(
+                      instance_name, to_upload.begin(), to_upload.end())
+                  .empty());
+
+        // Read blob via faulty digest
+        std::vector<bazel_re::Digest> to_read{faulty_digest};
+        CHECK(cas_client
+                  .BatchReadBlobs(instance_name, to_read.begin(), to_read.end())
+                  .empty());
+    }
+}