Fix tsl/platform/cloud:curl_http_request_test after breakage

PiperOrigin-RevId: 678538860
google · Sep 25, 2024 · 8c1310a · 8c1310a
1 parent 7c709a2
commit 8c1310a
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/tsl/platform/cloud/BUILD b/tsl/platform/cloud/BUILD
@@ -437,12 +437,14 @@ tsl_cc_test(
     srcs = ["curl_http_request_test.cc"],
     deps = [
         ":curl_http_request",
+        "//tsl/platform",
         "//tsl/platform:env_impl",
         "//tsl/platform:path",
         "//tsl/platform:platform_port",
         "//tsl/platform:test",
         "//tsl/platform:test_main",
         "@com_google_absl//absl/status",
+        "@com_google_absl//absl/strings",
         "@xla//xla/tsl/lib/core:status_test_util",
     ],
 )

diff --git a/tsl/platform/cloud/curl_http_request_test.cc b/tsl/platform/cloud/curl_http_request_test.cc
@@ -19,9 +19,11 @@ limitations under the License.
 #include <string>
 
 #include "absl/status/status.h"
+#include "absl/strings/str_cat.h"
 #include "xla/tsl/lib/core/status_test_util.h"
 #include "tsl/platform/mem.h"
 #include "tsl/platform/path.h"
+#include "tsl/platform/platform.h"
 #include "tsl/platform/test.h"
 
 namespace tsl {
@@ -495,9 +497,11 @@ TEST(CurlHttpRequestTest, GetRequest_CouldntResolveHost) {
   const auto& status = http_request.Send();
   EXPECT_EQ(error::FAILED_PRECONDITION, status.code());
   EXPECT_EQ(
-      "Error executing an HTTP request: libcurl code 6 meaning "
-      "'Could not resolve hostname', error details: Could not resolve host "
-      "'metadata'",
+      absl::StrCat(
+          "Error executing an HTTP request: libcurl code 6 meaning ",
+          (kIsOpenSource ? "'Couldn't resolve host name', error details: "
+                         : "'Could not resolve hostname', error details: "),
+          "Could not resolve host ", "'metadata'"),
       status.message());
   EXPECT_EQ(0, http_request.GetResponseCode());
 }