third_party/brpc
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases 1 Wiki Activity
0cf6c335b8
brpc/test/status_unittest.cpp
yangzuhao 0cf6c335b8 brpc源码
2022-12-14 20:13:26 +08:00

273 lines
8.6 KiB
C++
Raw Blame History

// 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());
}
}
Reference in New Issue View Git Blame Copy Permalink