101 lines
2.8 KiB
C++
101 lines
2.8 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>
|
|
#define USING_LOG_PREFIX SERVER
|
|
#define protected public
|
|
#define private public
|
|
|
|
#include "env/ob_simple_cluster_test_base.h"
|
|
#include "lib/mysqlclient/ob_mysql_result.h"
|
|
#include "rootserver/ddl_task/ob_ddl_task.h"
|
|
|
|
namespace oceanbase
|
|
{
|
|
namespace unittest
|
|
{
|
|
|
|
using namespace oceanbase::common;
|
|
using namespace oceanbase::storage;
|
|
using namespace oceanbase::rootserver;
|
|
|
|
|
|
class ObDDLTaskTest : public ObSimpleClusterTestBase
|
|
{
|
|
public:
|
|
ObDDLTaskTest() : ObSimpleClusterTestBase("test_ddl_task_") {}
|
|
};
|
|
|
|
TEST_F(ObDDLTaskTest, create_tenant)
|
|
{
|
|
uint64_t tenant_id;
|
|
ASSERT_EQ(OB_SUCCESS, create_tenant());
|
|
ASSERT_EQ(OB_SUCCESS, get_tenant_id(tenant_id));
|
|
ASSERT_NE(0, tenant_id);
|
|
ASSERT_EQ(OB_SUCCESS, get_curr_simple_server().init_sql_proxy2());
|
|
}
|
|
|
|
class MockDDLTask : public ObDDLTask
|
|
{
|
|
public:
|
|
MockDDLTask() : ObDDLTask(DDL_INVALID) { }
|
|
virtual int process() { return OB_SUCCESS; }
|
|
virtual int cleanup_impl() { return OB_SUCCESS; }
|
|
virtual void flt_set_task_span_tag() const { }
|
|
virtual void flt_set_status_span_tag() const { }
|
|
};
|
|
|
|
TEST_F(ObDDLTaskTest, switch_status)
|
|
{
|
|
common::ObMySQLProxy &tenant_sql_proxy = get_curr_simple_server().get_sql_proxy2();
|
|
ObArenaAllocator arena;
|
|
uint64_t tenant_id = 0;
|
|
ASSERT_EQ(OB_SUCCESS, get_tenant_id(tenant_id));
|
|
MockDDLTask task;
|
|
ObDDLTaskRecord task_record;
|
|
task.ddl_stmt_str_ = "create index xxx";
|
|
task.gmt_create_ = ObTimeUtility::current_time();
|
|
task.tenant_id_ = tenant_id;
|
|
task.object_id_ = 1;
|
|
task.target_object_id_ = 1;
|
|
task.schema_version_ = 1;
|
|
task.task_type_ = DDL_CREATE_INDEX;
|
|
task.task_status_ = ObDDLTaskStatus::PREPARE;
|
|
task.task_id_ = 1;
|
|
task.parent_task_id_ = 0;
|
|
task.task_version_ = 1;
|
|
task.execution_id_ = 1;
|
|
task.ret_code_ = OB_SUCCESS;
|
|
task.trace_id_.id_.seq_ = 1;
|
|
ASSERT_EQ(OB_SUCCESS, task.convert_to_record(task_record, arena));
|
|
ASSERT_EQ(OB_SUCCESS, ObDDLTaskRecordOperator::insert_record(*GCTX.sql_proxy_, task_record));
|
|
ASSERT_EQ(OB_SUCCESS, task.switch_status(ObDDLTaskStatus::DROP_SCHEMA, false/*enable_flt*/, OB_SUCCESS));
|
|
}
|
|
|
|
TEST_F(ObDDLTaskTest, end)
|
|
{
|
|
// used for debug
|
|
// ::sleep(3600);
|
|
}
|
|
|
|
}
|
|
}
|
|
|
|
int main(int argc, char **argv)
|
|
{
|
|
oceanbase::unittest::init_log_and_gtest(argc, argv);
|
|
OB_LOGGER.set_log_level("INFO");
|
|
|
|
::testing::InitGoogleTest(&argc, argv);
|
|
return RUN_ALL_TESTS();
|
|
}
|