110 lines
		
	
	
		
			2.6 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
			
		
		
	
	
			110 lines
		
	
	
		
			2.6 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 "db_initializer.h"
 | 
						|
 | 
						|
namespace oceanbase
 | 
						|
{
 | 
						|
namespace share
 | 
						|
{
 | 
						|
namespace schema
 | 
						|
{
 | 
						|
using namespace common;
 | 
						|
 | 
						|
TEST(TestDBInitializer, init)
 | 
						|
{
 | 
						|
  DBInitializer initer;
 | 
						|
  int ret = initer.init();
 | 
						|
  ASSERT_EQ(OB_SUCCESS, ret);
 | 
						|
}
 | 
						|
 | 
						|
// TEST(TestDBInitializer, create_system_table)
 | 
						|
// {
 | 
						|
//   {
 | 
						|
//     DBInitializer initer;
 | 
						|
//     int ret = initer.init();
 | 
						|
//     ASSERT_EQ(OB_SUCCESS, ret);
 | 
						|
 | 
						|
//     const bool only_core_tables = true;
 | 
						|
//     ret = initer.create_system_table(only_core_tables);
 | 
						|
//     ASSERT_EQ(OB_SUCCESS, ret);
 | 
						|
//   }
 | 
						|
 | 
						|
//   {
 | 
						|
//     DBInitializer initer;
 | 
						|
//     int ret = initer.init();
 | 
						|
//     ASSERT_EQ(OB_SUCCESS, ret);
 | 
						|
 | 
						|
//     const bool only_core_tables = false;
 | 
						|
//     ret = initer.create_system_table(only_core_tables);
 | 
						|
//     ASSERT_EQ(OB_SUCCESS, ret);
 | 
						|
 | 
						|
//     ret = initer.fill_sys_stat_table();
 | 
						|
//     ASSERT_EQ(OB_SUCCESS, ret);
 | 
						|
 | 
						|
//     ret = initer.create_tenant_space(2);
 | 
						|
//     ASSERT_EQ(OB_SUCCESS, ret);
 | 
						|
//   }
 | 
						|
// }
 | 
						|
 | 
						|
TEST(TestDBInitializer, inactive_sql_client)
 | 
						|
{
 | 
						|
  const uint64_t tenant_id = OB_SYS_TENANT_ID;
 | 
						|
  DBInitializer initer;
 | 
						|
  int ret = initer.init();
 | 
						|
  ASSERT_EQ(OB_SUCCESS, ret);
 | 
						|
 | 
						|
  ObMySQLProxy &proxy = initer.get_sql_proxy();
 | 
						|
  proxy.set_inactive();
 | 
						|
 | 
						|
  ObSqlString sql;
 | 
						|
  sql.assign("select 1");
 | 
						|
  int64_t ar = false;
 | 
						|
  ret = proxy.write(sql.ptr(), ar);
 | 
						|
  ASSERT_EQ(OB_INACTIVE_SQL_CLIENT, ret);
 | 
						|
 | 
						|
  {
 | 
						|
    ObISQLClient::ReadResult rs;
 | 
						|
    ret = proxy.read(rs, sql.ptr());
 | 
						|
    ASSERT_EQ(OB_INACTIVE_SQL_CLIENT, ret);
 | 
						|
  }
 | 
						|
 | 
						|
  {
 | 
						|
    ObMySQLTransaction trans;
 | 
						|
    ret = trans.start(&proxy, tenant_id);
 | 
						|
    ASSERT_EQ(OB_INACTIVE_SQL_CLIENT, ret);
 | 
						|
  }
 | 
						|
 | 
						|
  {
 | 
						|
    proxy.set_active();
 | 
						|
    ObMySQLTransaction trans;
 | 
						|
    ret = trans.start(&proxy, tenant_id);
 | 
						|
    ASSERT_EQ(OB_SUCCESS, ret);
 | 
						|
    proxy.set_inactive();
 | 
						|
    ret = proxy.write(sql.ptr(), ar);
 | 
						|
    ASSERT_EQ(OB_INACTIVE_SQL_CLIENT, ret);
 | 
						|
  }
 | 
						|
}
 | 
						|
 | 
						|
} // end namespace schema
 | 
						|
} // end namespace share
 | 
						|
} // end namespace oceanbase
 | 
						|
 | 
						|
int main(int argc, char **argv)
 | 
						|
{
 | 
						|
  testing::InitGoogleTest(&argc, argv);
 | 
						|
  oceanbase::common::ObLogger::get_logger().set_log_level("INFO");
 | 
						|
  return RUN_ALL_TESTS();
 | 
						|
}
 | 
						|
 |