From 9ab21a7f3d7e349b05ceaad5862d7e72a6d7e7b4 Mon Sep 17 00:00:00 2001
From: Paul Cristian Sarbu <paul.cristian.sarbu@huawei.com>
Date: Mon, 1 Jul 2024 14:56:32 +0200
Subject: Pass Auth::TLS instance to BazelApi and ServeApi

---
 test/buildtool/execution_api/bazel/bazel_api.test.cpp | 13 +++++++++++--
 1 file changed, 11 insertions(+), 2 deletions(-)

(limited to 'test/buildtool/execution_api/bazel/bazel_api.test.cpp')

diff --git a/test/buildtool/execution_api/bazel/bazel_api.test.cpp b/test/buildtool/execution_api/bazel/bazel_api.test.cpp
index 268edb1e..af457066 100644
--- a/test/buildtool/execution_api/bazel/bazel_api.test.cpp
+++ b/test/buildtool/execution_api/bazel/bazel_api.test.cpp
@@ -13,9 +13,11 @@
 // limitations under the License.
 
 #include <cstdlib>
+#include <optional>
 #include <string>
 
 #include "catch2/catch_test_macros.hpp"
+#include "src/buildtool/auth/authentication.hpp"
 #include "src/buildtool/execution_api/remote/bazel/bazel_api.hpp"
 #include "src/buildtool/execution_api/remote/config.hpp"
 #include "test/buildtool/execution_api/common/api_test.hpp"
@@ -25,8 +27,15 @@ namespace {
 
 auto const kApiFactory = []() {
     static auto const& server = RemoteExecutionConfig::RemoteAddress();
-    return IExecutionApi::Ptr{
-        new BazelApi{"remote-execution", server->host, server->port, {}}};
+    std::optional<Auth::TLS> auth = {};
+    if (Auth::Instance().GetAuthMethod() == AuthMethod::kTLS) {
+        auth = Auth::TLS::Instance();
+    }
+    return IExecutionApi::Ptr{new BazelApi{"remote-execution",
+                                           server->host,
+                                           server->port,
+                                           auth ? &*auth : nullptr,
+                                           {}}};
 };
 
 }  // namespace
-- 
cgit v1.2.3