// Licensed to the Apache Software Foundation (ASF) under one // or more contributor license agreements. See the NOTICE file // distributed with this work for additional information // regarding copyright ownership. The ASF licenses this file // to you 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 // // http://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 <errno.h> #include <gtest/gtest.h> #include "butil/status.h" namespace { class StatusTest : public ::testing::Test{ protected: StatusTest(){ }; virtual ~StatusTest(){}; virtual void SetUp() { srand(time(0)); }; virtual void TearDown() { }; }; TEST_F(StatusTest, success_status) { std::ostringstream oss; butil::Status st; ASSERT_TRUE(st.ok()); ASSERT_EQ(0, st.error_code()); ASSERT_STREQ("OK", st.error_cstr()); ASSERT_EQ("OK", st.error_str()); oss.str(""); oss << st; ASSERT_EQ("OK", oss.str()); butil::Status st2(0, "blahblah"); ASSERT_TRUE(st2.ok()); ASSERT_EQ(0, st2.error_code()); ASSERT_STREQ("OK", st2.error_cstr()); ASSERT_EQ("OK", st2.error_str()); oss.str(""); oss << st2; ASSERT_EQ("OK", oss.str()); butil::Status st3 = butil::Status::OK(); ASSERT_TRUE(st3.ok()); ASSERT_EQ(0, st3.error_code()); ASSERT_STREQ("OK", st3.error_cstr()); ASSERT_EQ("OK", st3.error_str()); oss.str(""); oss << st3; ASSERT_EQ("OK", oss.str()); } #define NO_MEMORY_STR "No memory" #define NO_CPU_STR "No CPU" TEST_F(StatusTest, failed_status) { std::ostringstream oss; butil::Status st1(ENOMEM, NO_MEMORY_STR); ASSERT_FALSE(st1.ok()); ASSERT_EQ(ENOMEM, st1.error_code()); ASSERT_STREQ(NO_MEMORY_STR, st1.error_cstr()); ASSERT_EQ(NO_MEMORY_STR, st1.error_str()); oss.str(""); oss << st1; ASSERT_EQ(NO_MEMORY_STR, oss.str()); butil::Status st2(EINVAL, "%s%s", NO_MEMORY_STR, NO_CPU_STR); ASSERT_FALSE(st2.ok()); ASSERT_EQ(EINVAL, st2.error_code()); ASSERT_STREQ(NO_MEMORY_STR NO_CPU_STR, st2.error_cstr()); ASSERT_EQ(NO_MEMORY_STR NO_CPU_STR, st2.error_str()); oss.str(""); oss << st2; ASSERT_EQ(NO_MEMORY_STR NO_CPU_STR, oss.str()); butil::Status st3(ENOMEM, NO_MEMORY_STR); ASSERT_FALSE(st3.ok()); ASSERT_EQ(ENOMEM, st3.error_code()); ASSERT_STREQ(NO_MEMORY_STR, st3.error_cstr()); ASSERT_EQ(NO_MEMORY_STR, st3.error_str()); oss.str(""); oss << st3; ASSERT_EQ(NO_MEMORY_STR, oss.str()); butil::Status st4(EINVAL, "%s%s", NO_MEMORY_STR, NO_CPU_STR); ASSERT_FALSE(st4.ok()); ASSERT_EQ(EINVAL, st4.error_code()); ASSERT_STREQ(NO_MEMORY_STR NO_CPU_STR, st4.error_cstr()); ASSERT_EQ(NO_MEMORY_STR NO_CPU_STR, st4.error_str()); oss.str(""); oss << st4; ASSERT_EQ(NO_MEMORY_STR NO_CPU_STR, oss.str()); butil::Status st5(EINVAL, "Blah"); ASSERT_FALSE(st5.ok()); ASSERT_EQ(EINVAL, st5.error_code()); ASSERT_STREQ("Blah", st5.error_cstr()); ASSERT_EQ(std::string("Blah"), st5.error_str()); oss.str(""); oss << st5; ASSERT_EQ(std::string("Blah"), oss.str()); } #define VERYLONGERROR \ "verylongverylongverylongverylongverylongverylongverylongverylong" \ "verylongverylongverylongverylongverylongverylongverylongverylong" \ "verylongverylongverylongverylongverylongverylongverylongverylong" \ "verylongverylongverylongverylongverylongverylongverylongverylong" \ "verylongverylongverylongverylongverylongverylongverylongverylong" \ "verylongverylongverylongverylongverylongverylongverylongverylong" \ "verylongverylongverylongverylongverylongverylongverylongverylong" \ "verylongverylongverylongverylongverylongverylongverylongverylong" \ "verylongverylongverylongverylongverylongverylongverylongverylong" \ "verylongverylongverylongverylongverylongverylongverylongverylong" \ "verylongverylongverylongverylongverylongverylongverylongverylong" \ " error" TEST_F(StatusTest, reset) { std::ostringstream oss; butil::Status st1(ENOMEM, NO_MEMORY_STR); ASSERT_FALSE(st1.ok()); ASSERT_EQ(ENOMEM, st1.error_code()); ASSERT_STREQ(NO_MEMORY_STR, st1.error_cstr()); ASSERT_EQ(NO_MEMORY_STR, st1.error_str()); oss.str(""); oss << st1; ASSERT_EQ(NO_MEMORY_STR, oss.str()); ASSERT_EQ(0, st1.set_error(EINVAL, "%s%s", NO_MEMORY_STR, NO_CPU_STR)); ASSERT_FALSE(st1.ok()); ASSERT_EQ(EINVAL, st1.error_code()); ASSERT_STREQ(NO_MEMORY_STR NO_CPU_STR, st1.error_cstr()); ASSERT_EQ(NO_MEMORY_STR NO_CPU_STR, st1.error_str()); oss.str(""); oss << st1; ASSERT_EQ(NO_MEMORY_STR NO_CPU_STR, oss.str()); ASSERT_EQ(0, st1.set_error(ENOMEM, NO_MEMORY_STR)); ASSERT_FALSE(st1.ok()); ASSERT_EQ(ENOMEM, st1.error_code()); ASSERT_STREQ(NO_MEMORY_STR, st1.error_cstr()); ASSERT_EQ(NO_MEMORY_STR, st1.error_str()); oss.str(""); oss << st1; ASSERT_EQ(NO_MEMORY_STR, oss.str()); st1.reset(); ASSERT_TRUE(st1.ok()); ASSERT_EQ(0, st1.error_code()); ASSERT_STREQ("OK", st1.error_cstr()); ASSERT_EQ("OK", st1.error_str()); oss.str(""); oss << st1; ASSERT_EQ("OK", oss.str()); ASSERT_EQ(0, st1.set_error(ENOMEM, "%s", VERYLONGERROR)); ASSERT_FALSE(st1.ok()); ASSERT_EQ(ENOMEM, st1.error_code()); ASSERT_STREQ(VERYLONGERROR, st1.error_cstr()); ASSERT_EQ(VERYLONGERROR, st1.error_str()); oss.str(""); oss << st1; ASSERT_EQ(VERYLONGERROR, oss.str()); } TEST_F(StatusTest, copy) { std::ostringstream oss; butil::Status st1(ENOMEM, NO_MEMORY_STR); ASSERT_FALSE(st1.ok()); ASSERT_EQ(ENOMEM, st1.error_code()); ASSERT_STREQ(NO_MEMORY_STR, st1.error_cstr()); ASSERT_EQ(NO_MEMORY_STR, st1.error_str()); oss.str(""); oss << st1; ASSERT_EQ(NO_MEMORY_STR, oss.str()); butil::Status st2; ASSERT_TRUE(st2.ok()); ASSERT_EQ(0, st2.error_code()); ASSERT_STREQ("OK", st2.error_cstr()); ASSERT_EQ("OK", st2.error_str()); oss.str(""); oss << st2; ASSERT_EQ("OK", oss.str()); st2 = st1; ASSERT_FALSE(st2.ok()); ASSERT_EQ(ENOMEM, st2.error_code()); ASSERT_STREQ(NO_MEMORY_STR, st2.error_cstr()); ASSERT_EQ(NO_MEMORY_STR, st2.error_str()); oss.str(""); oss << st2; ASSERT_EQ(NO_MEMORY_STR, oss.str()); ASSERT_EQ(0, st1.set_error(EINVAL, "%s%s", NO_MEMORY_STR, NO_CPU_STR)); ASSERT_FALSE(st1.ok()); ASSERT_EQ(EINVAL, st1.error_code()); ASSERT_STREQ(NO_MEMORY_STR NO_CPU_STR, st1.error_cstr()); ASSERT_EQ(NO_MEMORY_STR NO_CPU_STR, st1.error_str()); oss.str(""); oss << st1; ASSERT_EQ(NO_MEMORY_STR NO_CPU_STR, oss.str()); ASSERT_FALSE(st2.ok()); ASSERT_EQ(ENOMEM, st2.error_code()); ASSERT_STREQ(NO_MEMORY_STR, st2.error_cstr()); ASSERT_EQ(NO_MEMORY_STR, st2.error_str()); oss.str(""); oss << st2; ASSERT_EQ(NO_MEMORY_STR, oss.str()); st2 = st1; ASSERT_FALSE(st2.ok()); ASSERT_EQ(EINVAL, st2.error_code()); ASSERT_STREQ(NO_MEMORY_STR NO_CPU_STR, st2.error_cstr()); ASSERT_EQ(NO_MEMORY_STR NO_CPU_STR, st2.error_str()); oss.str(""); oss << st2; ASSERT_EQ(NO_MEMORY_STR NO_CPU_STR, oss.str()); ASSERT_EQ(0, st1.set_error(ENOMEM, NO_MEMORY_STR)); ASSERT_FALSE(st1.ok()); ASSERT_EQ(ENOMEM, st1.error_code()); ASSERT_STREQ(NO_MEMORY_STR, st1.error_cstr()); ASSERT_EQ(NO_MEMORY_STR, st1.error_str()); oss.str(""); oss << st1; ASSERT_EQ(NO_MEMORY_STR, oss.str()); // assign shorter to longer, no memory allocation. st2 = st1; ASSERT_FALSE(st2.ok()); ASSERT_EQ(ENOMEM, st2.error_code()); ASSERT_STREQ(NO_MEMORY_STR, st2.error_cstr()); ASSERT_EQ(NO_MEMORY_STR, st2.error_str()); oss.str(""); oss << st2; ASSERT_EQ(NO_MEMORY_STR, oss.str()); } TEST_F(StatusTest, message_has_zero) { std::ostringstream oss; char str[32] = "hello world"; butil::StringPiece slice(str); ASSERT_EQ(11UL, slice.as_string().size()); str[5] = '\0'; ASSERT_EQ(11UL, slice.as_string().size()); butil::Status st1(ENOMEM, slice); ASSERT_FALSE(st1.ok()); ASSERT_EQ(ENOMEM, st1.error_code()); ASSERT_STREQ("hello", st1.error_cstr()); oss.str(""); oss << st1; ASSERT_EQ(st1.error_str(), oss.str()); } }