From e0833fcedfdb7233047615650900ef2f285a80eb Mon Sep 17 00:00:00 2001
From: Maksim Denisov <denisov.maksim@huawei.com>
Date: Mon, 10 Feb 2025 10:56:56 +0100
Subject: Store BazelCapabilitiesClient in BazelNetwork

...and pass it to BazelCasClient
---
 test/utils/remote_execution/bazel_action_creator.hpp | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

(limited to 'test/utils/remote_execution/bazel_action_creator.hpp')

diff --git a/test/utils/remote_execution/bazel_action_creator.hpp b/test/utils/remote_execution/bazel_action_creator.hpp
index 3ffcf827..49f29e9c 100644
--- a/test/utils/remote_execution/bazel_action_creator.hpp
+++ b/test/utils/remote_execution/bazel_action_creator.hpp
@@ -33,6 +33,7 @@
 #include "src/buildtool/common/remote/retry_config.hpp"
 #include "src/buildtool/crypto/hash_function.hpp"
 #include "src/buildtool/execution_api/common/artifact_blob.hpp"
+#include "src/buildtool/execution_api/remote/bazel/bazel_capabilities_client.hpp"
 #include "src/buildtool/execution_api/remote/bazel/bazel_cas_client.hpp"
 #include "src/buildtool/execution_api/remote/config.hpp"
 #include "src/buildtool/file_system/object_type.hpp"
@@ -104,10 +105,15 @@
 
     RetryConfig retry_config{};  // default retry config
 
+    BazelCapabilitiesClient capabilities(remote_config->remote_address->host,
+                                         remote_config->remote_address->port,
+                                         &*auth_config,
+                                         &retry_config);
     BazelCasClient cas_client(remote_config->remote_address->host,
                               remote_config->remote_address->port,
                               &*auth_config,
-                              &retry_config);
+                              &retry_config,
+                              &capabilities);
 
     if (cas_client.BatchUpdateBlobs(instance_name, blobs) == blobs.size()) {
         return std::make_unique<bazel_re::Digest>(
-- 
cgit v1.2.3