145 lines
		
	
	
		
			5.1 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
			
		
		
	
	
			145 lines
		
	
	
		
			5.1 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.
 | |
|  */
 | |
| 
 | |
| #define USING_LOG_PREFIX SHARE
 | |
| 
 | |
| #include <gtest/gtest.h>
 | |
| #include <gmock/gmock.h>
 | |
| #define private public
 | |
| #define protected public
 | |
| #include "share/schema/ob_server_schema_service.h"
 | |
| #include "share/schema/ob_multi_version_schema_service.h"
 | |
| #include "share/schema/ob_schema_service_sql_impl.h"
 | |
| 
 | |
| namespace oceanbase {
 | |
| namespace share {
 | |
| namespace schema {
 | |
| using namespace common;
 | |
| class TestServerSchemaService : public ::testing::Test {
 | |
| public:
 | |
|   virtual void SetUp();
 | |
|   virtual void TearDown()
 | |
|   {}
 | |
| };
 | |
| 
 | |
| void TestServerSchemaService::SetUp()
 | |
| {}
 | |
| 
 | |
| using ::testing::_;
 | |
| using ::testing::Invoke;
 | |
| using ::testing::Return;
 | |
| 
 | |
| TEST_F(TestServerSchemaService, AllSchemaKey)
 | |
| {
 | |
|   int ret = OB_SUCCESS;
 | |
| 
 | |
|   ObServerSchemaService::AllSchemaKeys all_keys;
 | |
|   ret = all_keys.create(50);
 | |
|   ASSERT_EQ(OB_SUCCESS, ret);
 | |
|   SchemaKey key;
 | |
|   ret = all_keys.new_tenant_keys_.set_refactored(key);
 | |
|   ASSERT_EQ(OB_SUCCESS, ret);
 | |
|   ret = all_keys.alter_tenant_keys_.set_refactored(key);
 | |
|   ASSERT_EQ(OB_SUCCESS, ret);
 | |
|   ret = all_keys.del_tenant_keys_.set_refactored(key);
 | |
|   ASSERT_EQ(OB_SUCCESS, ret);
 | |
|   ret = all_keys.new_user_keys_.set_refactored(key);
 | |
|   ASSERT_EQ(OB_SUCCESS, ret);
 | |
|   ret = all_keys.del_user_keys_.set_refactored(key);
 | |
|   ASSERT_EQ(OB_SUCCESS, ret);
 | |
|   ret = all_keys.new_database_keys_.set_refactored(key);
 | |
|   ASSERT_EQ(OB_SUCCESS, ret);
 | |
|   ret = all_keys.del_database_keys_.set_refactored(key);
 | |
|   ASSERT_EQ(OB_SUCCESS, ret);
 | |
|   ret = all_keys.new_tablegroup_keys_.set_refactored(key);
 | |
|   ASSERT_EQ(OB_SUCCESS, ret);
 | |
|   ret = all_keys.del_tablegroup_keys_.set_refactored(key);
 | |
|   ASSERT_EQ(OB_SUCCESS, ret);
 | |
|   ret = all_keys.new_table_keys_.set_refactored(key);
 | |
|   ASSERT_EQ(OB_SUCCESS, ret);
 | |
|   ret = all_keys.del_table_keys_.set_refactored(key);
 | |
|   ASSERT_EQ(OB_SUCCESS, ret);
 | |
|   ret = all_keys.new_outline_keys_.set_refactored(key);
 | |
|   ASSERT_EQ(OB_SUCCESS, ret);
 | |
|   ret = all_keys.del_outline_keys_.set_refactored(key);
 | |
|   ASSERT_EQ(OB_SUCCESS, ret);
 | |
|   ret = all_keys.new_db_priv_keys_.set_refactored(key);
 | |
|   ASSERT_EQ(OB_SUCCESS, ret);
 | |
|   ret = all_keys.del_db_priv_keys_.set_refactored(key);
 | |
|   ASSERT_EQ(OB_SUCCESS, ret);
 | |
|   ret = all_keys.new_table_priv_keys_.set_refactored(key);
 | |
|   ASSERT_EQ(OB_SUCCESS, ret);
 | |
|   ret = all_keys.del_table_priv_keys_.set_refactored(key);
 | |
|   ASSERT_EQ(OB_SUCCESS, ret);
 | |
| 
 | |
|   ASSERT_EQ(1, all_keys.new_tenant_keys_.size());
 | |
|   ASSERT_EQ(1, all_keys.alter_tenant_keys_.size());
 | |
|   ASSERT_EQ(1, all_keys.new_user_keys_.size());
 | |
|   ASSERT_EQ(1, all_keys.new_database_keys_.size());
 | |
|   ASSERT_EQ(1, all_keys.new_tablegroup_keys_.size());
 | |
|   ASSERT_EQ(1, all_keys.new_table_keys_.size());
 | |
|   ASSERT_EQ(1, all_keys.new_outline_keys_.size());
 | |
|   ASSERT_EQ(1, all_keys.new_db_priv_keys_.size());
 | |
|   ASSERT_EQ(1, all_keys.new_table_priv_keys_.size());
 | |
|   ASSERT_EQ(1, all_keys.del_tenant_keys_.size());
 | |
|   ASSERT_EQ(1, all_keys.del_user_keys_.size());
 | |
|   ASSERT_EQ(1, all_keys.del_database_keys_.size());
 | |
|   ASSERT_EQ(1, all_keys.del_tablegroup_keys_.size());
 | |
|   ASSERT_EQ(1, all_keys.del_table_keys_.size());
 | |
|   ASSERT_EQ(1, all_keys.del_outline_keys_.size());
 | |
|   ASSERT_EQ(1, all_keys.del_db_priv_keys_.size());
 | |
|   ASSERT_EQ(1, all_keys.del_table_priv_keys_.size());
 | |
| 
 | |
|   all_keys.reset();
 | |
|   ASSERT_EQ(0, all_keys.new_tenant_keys_.size());
 | |
|   ASSERT_EQ(0, all_keys.alter_tenant_keys_.size());
 | |
|   ASSERT_EQ(0, all_keys.new_user_keys_.size());
 | |
|   ASSERT_EQ(0, all_keys.new_database_keys_.size());
 | |
|   ASSERT_EQ(0, all_keys.new_tablegroup_keys_.size());
 | |
|   ASSERT_EQ(0, all_keys.new_table_keys_.size());
 | |
|   ASSERT_EQ(0, all_keys.new_outline_keys_.size());
 | |
|   ASSERT_EQ(0, all_keys.new_db_priv_keys_.size());
 | |
|   ASSERT_EQ(0, all_keys.new_table_priv_keys_.size());
 | |
|   ASSERT_EQ(0, all_keys.del_tenant_keys_.size());
 | |
|   ASSERT_EQ(0, all_keys.del_user_keys_.size());
 | |
|   ASSERT_EQ(0, all_keys.del_database_keys_.size());
 | |
|   ASSERT_EQ(0, all_keys.del_tablegroup_keys_.size());
 | |
|   ASSERT_EQ(0, all_keys.del_table_keys_.size());
 | |
|   ASSERT_EQ(0, all_keys.del_outline_keys_.size());
 | |
|   ASSERT_EQ(0, all_keys.del_db_priv_keys_.size());
 | |
|   ASSERT_EQ(0, all_keys.del_table_priv_keys_.size());
 | |
| }
 | |
| 
 | |
| TEST_F(TestServerSchemaService, table_priv_equal_to)
 | |
| {
 | |
|   SchemaKey key1;
 | |
|   key1.tenant_id_ = 1;
 | |
|   key1.user_id_ = 1;
 | |
|   key1.database_name_ = "db";
 | |
|   key1.table_name_ = "table";
 | |
|   SchemaKey key2 = key1;
 | |
|   ObServerSchemaService::table_priv_equal_to table_priv_equal_to;
 | |
|   ASSERT_TRUE(table_priv_equal_to(key1, key2));
 | |
| }
 | |
| 
 | |
| }  // end namespace schema
 | |
| }  // end namespace share
 | |
| }  // end namespace oceanbase
 | |
| 
 | |
| int main(int argc, char** argv)
 | |
| {
 | |
|   oceanbase::common::ObLogger::get_logger().set_log_level("INFO");
 | |
|   OB_LOGGER.set_log_level("INFO");
 | |
|   ::testing::InitGoogleTest(&argc, argv);
 | |
|   return RUN_ALL_TESTS();
 | |
| }
 | 
