132 lines
		
	
	
		
			3.9 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
			
		
		
	
	
			132 lines
		
	
	
		
			3.9 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 <cstdio>
 | 
						|
#include <gtest/gtest.h>
 | 
						|
#include <signal.h>
 | 
						|
#define private public
 | 
						|
#include "share/scn.h"
 | 
						|
#include "storage/slog/ob_storage_logger.h"
 | 
						|
#include "env/ob_simple_cluster_test_base.h"
 | 
						|
#include "logservice/ob_log_service.h"
 | 
						|
#include "logservice/leader_coordinator/ob_failure_detector.h"
 | 
						|
#undef private
 | 
						|
 | 
						|
const std::string TEST_NAME = "rto_func";
 | 
						|
const int64_t CLOG_HANG_TIME_THRESHOLD_US = 5 * 1000 * 1000;
 | 
						|
int64_t mock_fatal_err_ts = OB_INVALID_TIMESTAMP;
 | 
						|
bool mock_clog_disk_hang = false;
 | 
						|
bool mock_sstable_io_hang = false;
 | 
						|
//slog的路径全部为inline函数,暂时无法mock
 | 
						|
//bool mock_slog_io_hang = false;
 | 
						|
bool mock_clog_disk_full = false;
 | 
						|
using namespace oceanbase;
 | 
						|
namespace oceanbase
 | 
						|
{
 | 
						|
using namespace palf;
 | 
						|
using namespace storage;
 | 
						|
namespace common
 | 
						|
{
 | 
						|
int ObIOFaultDetector::get_device_health_status(ObDeviceHealthStatus &dhs,
 | 
						|
    int64_t &device_abnormal_time)
 | 
						|
{
 | 
						|
  if (mock_sstable_io_hang) {
 | 
						|
    dhs = DEVICE_HEALTH_WARNING;
 | 
						|
    device_abnormal_time = mock_fatal_err_ts - GCONF.data_storage_warning_tolerance_time;
 | 
						|
  } else {
 | 
						|
    dhs = DEVICE_HEALTH_NORMAL;
 | 
						|
    device_abnormal_time = OB_INVALID_TIMESTAMP;
 | 
						|
  }
 | 
						|
  return OB_SUCCESS;
 | 
						|
}
 | 
						|
};
 | 
						|
using namespace common;
 | 
						|
namespace logservice
 | 
						|
{
 | 
						|
int ObLogService::get_io_start_time(int64_t &last_working_time)
 | 
						|
{
 | 
						|
  last_working_time = mock_clog_disk_hang ?
 | 
						|
                      mock_fatal_err_ts - CLOG_HANG_TIME_THRESHOLD_US
 | 
						|
                      : OB_INVALID_TIMESTAMP;
 | 
						|
  return OB_SUCCESS;
 | 
						|
}
 | 
						|
int ObLogService::check_disk_space_enough(bool &is_disk_enough)
 | 
						|
{
 | 
						|
  is_disk_enough = !mock_clog_disk_full;
 | 
						|
  return OB_SUCCESS;
 | 
						|
}
 | 
						|
};
 | 
						|
using namespace logservice;
 | 
						|
using namespace coordinator;
 | 
						|
namespace unittest
 | 
						|
{
 | 
						|
class ObRTOTest : public ObSimpleClusterTestBase
 | 
						|
{
 | 
						|
public:
 | 
						|
  ObRTOTest() : ObSimpleClusterTestBase("test_ob_rto_") {}
 | 
						|
};
 | 
						|
 | 
						|
TEST_F(ObRTOTest, basic_rto)
 | 
						|
{
 | 
						|
  CLOG_LOG(INFO, "test rto begin");
 | 
						|
  share::ObTenantSwitchGuard tguard;
 | 
						|
  ASSERT_EQ(OB_SUCCESS, tguard.switch_to(1));
 | 
						|
  //mock clog disk hang
 | 
						|
  mock_fatal_err_ts = ObTimeUtility::fast_current_time();
 | 
						|
  mock_clog_disk_hang = true;
 | 
						|
  while (!MTL(ObFailureDetector*)->has_add_clog_hang_event_) {
 | 
						|
    CLOG_LOG(INFO, "waiting detect clog disk hang");
 | 
						|
    usleep(100 * 1000);
 | 
						|
  }
 | 
						|
  mock_clog_disk_hang = false;
 | 
						|
  while (MTL(ObFailureDetector*)->has_add_clog_hang_event_) {
 | 
						|
    CLOG_LOG(INFO, "waiting recover clog disk hang");
 | 
						|
    usleep(100 * 1000);
 | 
						|
  }
 | 
						|
 | 
						|
  //mock clog disk full
 | 
						|
  mock_clog_disk_full = true;
 | 
						|
  while (!MTL(ObFailureDetector*)->has_add_clog_full_event_) {
 | 
						|
    CLOG_LOG(INFO, "waiting detect clog disk full");
 | 
						|
    usleep(100 * 1000);
 | 
						|
  }
 | 
						|
  mock_clog_disk_full = false;
 | 
						|
  while (MTL(ObFailureDetector*)->has_add_clog_full_event_) {
 | 
						|
    CLOG_LOG(INFO, "waiting recover clog disk full");
 | 
						|
    usleep(100 * 1000);
 | 
						|
  }
 | 
						|
 | 
						|
  //mock sstable io hang
 | 
						|
  mock_fatal_err_ts = ObTimeUtility::fast_current_time();
 | 
						|
  mock_sstable_io_hang = true;
 | 
						|
  while (!MTL(ObFailureDetector*)->has_add_sstable_hang_event_) {
 | 
						|
    CLOG_LOG(INFO, "waiting detect sstable io hang");
 | 
						|
    usleep(100 * 1000);
 | 
						|
  }
 | 
						|
  mock_sstable_io_hang = false;
 | 
						|
  while (MTL(ObFailureDetector*)->has_add_sstable_hang_event_) {
 | 
						|
    CLOG_LOG(INFO, "waiting recover sstable io hang");
 | 
						|
    usleep(100 * 1000);
 | 
						|
  }
 | 
						|
}
 | 
						|
} // unitest
 | 
						|
} // oceanbase
 | 
						|
 | 
						|
int main(int argc, char **argv)
 | 
						|
{
 | 
						|
  oceanbase::unittest::init_log_and_gtest(argc, argv);
 | 
						|
  OB_LOGGER.set_log_level("INFO");
 | 
						|
  GCONF._enable_defensive_check = false;
 | 
						|
  ::testing::InitGoogleTest(&argc, argv);
 | 
						|
  return RUN_ALL_TESTS();
 | 
						|
}
 |