talksik/mediapipe
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Project import generated by Copybara.

Browse Source 
GitOrigin-RevId: d073f8e21be2fcc0e503cb97c6695078b6b75310
This commit is contained in:
MediaPipe Team
2021-02-27 03:30:05 -05:00
committed by chuoling
parent 39309bedba
commit 350fbb2100
755 changed files with 16391 additions and 11075 deletions
Download Patch File Download Diff File Expand all files Collapse all files
mediapipe/framework/tool/status_util_test.cc
+20 -21
View File
@@ -36,24 +36,24 @@ TEST(StatusTest, StatusStopIsNotOk) { EXPECT_FALSE(tool::StatusStop().ok()); }
TEST(StatusTest, Prefix) {
const std::string base_error_message("error_with_this_string");
const std::string prefix_error_message("error_with_prefix: ");
mediapipe::Status base_status = mediapipe::Status(
mediapipe::StatusCode::kInvalidArgument, base_error_message);
mediapipe::Status status =
absl::Status base_status =
absl::Status(absl::StatusCode::kInvalidArgument, base_error_message);
absl::Status status =
tool::AddStatusPrefix(prefix_error_message, base_status);
EXPECT_THAT(status.ToString(), HasSubstr(base_error_message));
EXPECT_THAT(status.ToString(), HasSubstr(prefix_error_message));
EXPECT_EQ(mediapipe::StatusCode::kInvalidArgument, status.code());
EXPECT_EQ(absl::StatusCode::kInvalidArgument, status.code());
}
TEST(StatusTest, CombinedStatus) {
std::vector<mediapipe::Status> errors;
std::vector<absl::Status> errors;
const std::string prefix_error_message("error_with_prefix: ");
mediapipe::Status status;
absl::Status status;
errors.clear();
errors.emplace_back(mediapipe::StatusCode::kInvalidArgument,
errors.emplace_back(absl::StatusCode::kInvalidArgument,
"error_with_this_string");
errors.emplace_back(mediapipe::StatusCode::kInvalidArgument,
errors.emplace_back(absl::StatusCode::kInvalidArgument,
"error_with_that_string");
errors.back().SetPayload("test payload type",
absl::Cord(absl::string_view("hello")));
@@ -61,30 +61,29 @@ TEST(StatusTest, CombinedStatus) {
EXPECT_THAT(status.ToString(), HasSubstr(std::string(errors[0].message())));
EXPECT_THAT(status.ToString(), HasSubstr(std::string(errors[1].message())));
EXPECT_THAT(status.ToString(), HasSubstr(prefix_error_message));
EXPECT_EQ(mediapipe::StatusCode::kInvalidArgument, status.code());
EXPECT_EQ(absl::StatusCode::kInvalidArgument, status.code());
errors.clear();
errors.emplace_back(mediapipe::StatusCode::kNotFound,
"error_with_this_string");
errors.emplace_back(mediapipe::StatusCode::kInvalidArgument,
errors.emplace_back(absl::StatusCode::kNotFound, "error_with_this_string");
errors.emplace_back(absl::StatusCode::kInvalidArgument,
"error_with_that_string");
status = tool::CombinedStatus(prefix_error_message, errors);
EXPECT_THAT(status.ToString(), HasSubstr(std::string(errors[0].message())));
EXPECT_THAT(status.ToString(), HasSubstr(std::string(errors[1].message())));
EXPECT_THAT(status.ToString(), HasSubstr(prefix_error_message));
EXPECT_EQ(mediapipe::StatusCode::kUnknown, status.code());
EXPECT_EQ(absl::StatusCode::kUnknown, status.code());
errors.clear();
errors.emplace_back(mediapipe::StatusCode::kOk, "error_with_this_string");
errors.emplace_back(mediapipe::StatusCode::kInvalidArgument,
errors.emplace_back(absl::StatusCode::kOk, "error_with_this_string");
errors.emplace_back(absl::StatusCode::kInvalidArgument,
"error_with_that_string");
status = tool::CombinedStatus(prefix_error_message, errors);
EXPECT_THAT(status.ToString(), HasSubstr(std::string(errors[1].message())));
EXPECT_THAT(status.ToString(), HasSubstr(prefix_error_message));
EXPECT_EQ(mediapipe::StatusCode::kInvalidArgument, status.code());
EXPECT_EQ(absl::StatusCode::kInvalidArgument, status.code());
errors.clear();
errors.emplace_back(mediapipe::StatusCode::kOk, "error_with_this_string");
errors.emplace_back(mediapipe::StatusCode::kOk, "error_with_that_string");
errors.emplace_back(absl::StatusCode::kOk, "error_with_this_string");
errors.emplace_back(absl::StatusCode::kOk, "error_with_that_string");
MP_EXPECT_OK(tool::CombinedStatus(prefix_error_message, errors));
errors.clear();
@@ -93,13 +92,13 @@ TEST(StatusTest, CombinedStatus) {
// Verify tool::StatusInvalid() and tool::StatusFail() and the alternatives
// recommended by their ABSL_DEPRECATED messages return the same
// mediapipe::Status objects.
// absl::Status objects.
TEST(StatusTest, Deprecated) {
const std::string error_message = "an error message";
EXPECT_EQ(tool::StatusInvalid(error_message), // NOLINT
mediapipe::InvalidArgumentError(error_message));
absl::InvalidArgumentError(error_message));
EXPECT_EQ(tool::StatusFail(error_message), // NOLINT
mediapipe::UnknownError(error_message));
absl::UnknownError(error_message));
}
} // namespace