Add a few macros for testing.

PiperOrigin-RevId: 366291052

Author: Neural-Link Team

research/carls/testing/BUILD

@@ -47,11 +47,15 @@ carls_cc_proto_library(
 cc_library(
     name = "test_helper",
     testonly = 1,
+    srcs = ["test_helper.cc"],
     hdrs = ["test_helper.h"],
     deps = [
         "@com_github_google_glog//:glog",
+        "@com_github_grpc_grpc//:grpc++",
+        "@com_google_absl//absl/status",
         "@com_google_googletest//:gtest",
         "@com_google_protobuf//:protobuf",
+        "@tensorflow_includes//:includes",
     ],
 )
 
@@ -61,7 +65,11 @@ cc_test(
     deps = [
         ":test_helper",
         ":test_proto2_cc_proto",
+        "//research/carls/base:status_helper",
+        "@com_github_grpc_grpc//:grpc++",
+        "@com_google_absl//absl/status",
         "@com_google_googletest//:gtest_main",
+        "@tensorflow_includes//:includes",
     ],
 )
 

research/carls/testing/test_helper.cc

@@ -0,0 +1,40 @@
+/* Copyright 2021 Google LLC. All Rights Reserved.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+     https://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+==============================================================================*/
+
+#include "research/carls/testing/test_helper.h"
+
+#include "grpcpp/support/status.h"  // net
+#include "tensorflow/core/platform/status.h"
+
+namespace carls {
+namespace internal {
+
+template <>
+std::string GetErrorMessage(const absl::Status& status) {
+  return std::string(status.message());
+}
+
+template <>
+std::string GetErrorMessage(const grpc::Status& status) {
+  return status.error_message();
+}
+
+template <>
+std::string GetErrorMessage(const tensorflow::Status& status) {
+  return status.error_message();
+}
+
+}  // namespace internal
+}  // namespace carls

research/carls/testing/test_helper.h

@@ -21,8 +21,15 @@ limitations under the License.
 #include "google/protobuf/text_format.h" // proto import
 #include "gmock/gmock.h"
 #include "gtest/gtest.h"
+#include "absl/status/status.h"
 
 namespace carls {
+namespace internal {
+
+template <typename StatusType>
+std::string GetErrorMessage(const StatusType& status);
+
+}  // namespace internal
 
 // A simple implementation of a proto matcher comparing string representations.
 class ProtoStringMatcher {
@@ -57,6 +64,24 @@ inline ::testing::PolymorphicMatcher<ProtoStringMatcher> EqualsProto(
   return ::testing::MakePolymorphicMatcher(ProtoStringMatcher(proto));
 }
 
+// Macros for testing the results of functions that return Status or
+// StatusOr<T> (for any type T).
+#undef EXPECT_OK
+#define EXPECT_OK(expression) EXPECT_TRUE(expression.ok())
+
+#define EXPECT_NOT_OK(expression) EXPECT_FALSE(expression.ok())
+
+#define EXPECT_ERROR(expression, err_msg) \
+  ASSERT_FALSE(expression.ok());          \
+  EXPECT_EQ(err_msg, internal::GetErrorMessage(expression));
+
+#undef ASSERT_OK
+#define ASSERT_OK(expression) ASSERT_TRUE(expression.ok())
+
+#define ASSERT_ERROR(expression, err_msg) \
+  ASSERT_FALSE(expression.ok());          \
+  ASSERT_EQ(err_msg, internal::GetErrorMessage(expression));
+
 }  // namespace carls
 
 #endif  // NEURAL_STRUCTURED_LEARNING_RESEARCH_CARLS_TESTING_TEST_HELPER_H_

research/carls/testing/test_helper_test.cc

@@ -15,7 +15,12 @@ limitations under the License.
 
 #include "research/carls/testing/test_helper.h"
 
+#include "grpcpp/support/status.h"  // net
+#include "absl/status/status.h"
+#include "research/carls/base/status_helper.h"
 #include "research/carls/testing/test_proto2.pb.h"  // proto to pb
+#include "tensorflow/core/platform/status.h"
+#include "tensorflow/core/protobuf/error_codes.pb.h"  // proto to pb
 
 namespace carls {
 
@@ -28,9 +33,41 @@ TEST(TestHelperTest, EqualsProto) {
 TEST(TestHelperTest, EqualsProtoText) {
   TestBaseProto2Def proto;
   proto.set_name("Audrey");
-  EXPECT_THAT(proto, EqualsProto<TestBaseProto2Def>(R"(
+  EXPECT_THAT(proto, EqualsProto<TestBaseProto2Def>(R"pb(
                 name: "Audrey"
-              )"));
+              )pb"));
+}
+
+TEST(TestHelperTest, AbslStatusChecks) {
+  EXPECT_OK(absl::OkStatus());
+  EXPECT_NOT_OK(absl::InternalError("First error."));
+  EXPECT_ERROR(absl::InternalError("First error."), "First error.");
+  ASSERT_OK(absl::OkStatus());
+  ASSERT_ERROR(absl::InternalError("Second error."), "Second error.");
+}
+
+TEST(TestHelperTest, GrpcStatusChecks) {
+  EXPECT_OK(grpc::Status::OK);
+  EXPECT_OK(ToGrpcStatus(absl::OkStatus()));
+  EXPECT_NOT_OK(ToGrpcStatus(absl::InternalError("First error.")));
+  EXPECT_ERROR(ToGrpcStatus(absl::InternalError("First error.")),
+               "First error.");
+  ASSERT_OK(grpc::Status::OK);
+  ASSERT_ERROR(ToGrpcStatus(absl::InternalError("Second error.")),
+               "Second error.");
+}
+
+TEST(TestHelperTest, TensoFlowStatusChecks) {
+  EXPECT_OK(tensorflow::Status::OK());
+  EXPECT_NOT_OK(
+      tensorflow::Status(tensorflow::error::INVALID_ARGUMENT, "First error."));
+  EXPECT_ERROR(
+      tensorflow::Status(tensorflow::error::INVALID_ARGUMENT, "First error."),
+      "First error.");
+  ASSERT_OK(tensorflow::Status::OK());
+  ASSERT_ERROR(
+      tensorflow::Status(tensorflow::error::INVALID_ARGUMENT, "Second error."),
+      "Second error.");
 }
 
 }  // namespace carls