80 lines
2.3 KiB
C++
80 lines
2.3 KiB
C++
/**
|
|
* Copyright (c) 2021 OceanBase
|
|
* OceanBase CE is licensed under Mulan PubL v2.
|
|
* You can use this software according to the terms and conditions of the Mulan PubL v2.
|
|
* You may obtain a copy of Mulan PubL v2 at:
|
|
* http://license.coscl.org.cn/MulanPubL-2.0
|
|
* THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
|
|
* EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
|
|
* MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
|
|
* See the Mulan PubL v2 for more details.
|
|
*/
|
|
|
|
#include <gtest/gtest.h>
|
|
#include "clog/ob_log_define.h"
|
|
|
|
namespace oceanbase {
|
|
using namespace clog;
|
|
using namespace common;
|
|
namespace unittest {
|
|
struct TestClass {
|
|
union {
|
|
int64_t tmp_val_;
|
|
struct {
|
|
common::ObRole role_;
|
|
int16_t state_;
|
|
bool is_removed_;
|
|
};
|
|
};
|
|
|
|
int get_role_and_state() const
|
|
{
|
|
int ret = OB_SUCCESS;
|
|
const int64_t tmp = ATOMIC_LOAD(&tmp_val_);
|
|
if (tmp >> 48) { // check is_removed_
|
|
ret = OB_NOT_MASTER;
|
|
} else if (((tmp & ((1ll << 48) - 1)) >> 32) != ACTIVE) { // check state_
|
|
ret = OB_NOT_MASTER;
|
|
} else if ((tmp & ((1ll << 32) - 1)) != LEADER) { // check role_
|
|
ret = OB_NOT_MASTER;
|
|
}
|
|
|
|
const bool is_removed = static_cast<bool>(tmp >> 48);
|
|
const int16_t state = static_cast<int16_t>((tmp & ((1ll << 48) - 1)) >> 32);
|
|
const ObRole role = static_cast<ObRole>(tmp & ((1ll << 32) - 1));
|
|
CLOG_LOG(INFO, "info", K(is_removed), K(state), K(role));
|
|
return ret;
|
|
}
|
|
};
|
|
|
|
TEST(test_get_role, test1)
|
|
{
|
|
TestClass a;
|
|
a.tmp_val_ = 0;
|
|
a.role_ = LEADER;
|
|
a.state_ = ACTIVE;
|
|
a.is_removed_ = false;
|
|
EXPECT_EQ(common::OB_SUCCESS, a.get_role_and_state());
|
|
a.role_ = FOLLOWER;
|
|
EXPECT_EQ(common::OB_NOT_MASTER, a.get_role_and_state());
|
|
a.role_ = LEADER;
|
|
a.state_ = RECONFIRM;
|
|
EXPECT_EQ(common::OB_NOT_MASTER, a.get_role_and_state());
|
|
a.state_ = ACTIVE;
|
|
a.is_removed_ = true;
|
|
EXPECT_EQ(common::OB_NOT_MASTER, a.get_role_and_state());
|
|
a.is_removed_ = false;
|
|
EXPECT_EQ(common::OB_SUCCESS, a.get_role_and_state());
|
|
}
|
|
} // namespace unittest
|
|
} // namespace oceanbase
|
|
|
|
int main(int argc, char** argv)
|
|
{
|
|
OB_LOGGER.set_file_name("test_get_role.log", true);
|
|
OB_LOGGER.set_log_level("INFO");
|
|
CLOG_LOG(INFO, "begin unittest::test_get_role");
|
|
testing::InitGoogleTest(&argc, argv);
|
|
return RUN_ALL_TESTS();
|
|
}
|