88 lines
		
	
	
		
			3.1 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
			
		
		
	
	
			88 lines
		
	
	
		
			3.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.
 | 
						|
 */
 | 
						|
 | 
						|
#include "lib/allocator/ob_malloc.h"
 | 
						|
#include "lib/ob_errno.h"
 | 
						|
#include "lib/oblog/ob_log.h"
 | 
						|
#include "lib/string/ob_string.h"
 | 
						|
#include "share/backup/ob_backup_io_adapter.h"
 | 
						|
#include "share/backup/ob_backup_struct.h"
 | 
						|
#include "logservice/archiveservice/ob_archive_io.h"
 | 
						|
#include <cstdlib>
 | 
						|
#include <gtest/gtest.h>
 | 
						|
#include <string>
 | 
						|
 | 
						|
using namespace oceanbase;
 | 
						|
using namespace common;
 | 
						|
using namespace share;
 | 
						|
using namespace archive;
 | 
						|
using namespace std;
 | 
						|
 | 
						|
TEST(TestObArchiveOverwrite, oss_push_log)
 | 
						|
{
 | 
						|
  int64_t rand_num = rand();
 | 
						|
  ObBackupDest dest;
 | 
						|
  ObBackupIoAdapter util;
 | 
						|
  ObArchiveIO archive_io;
 | 
						|
  const int64_t part_size = 5 * 1024 * 1024L;
 | 
						|
  const int64_t total_size = 10 * 1024 * 1024L;
 | 
						|
  char *data = (char*)ob_malloc(total_size, "ArchiveTest");
 | 
						|
  string path = "oss://antsys-oceanbasebackup/test_archive/test_archive_overwrite/" + to_string(rand_num) + "?host=cn-hangzhou-alipay-b.oss-cdn.aliyun-inc.com&access_id=LTAI4Fdwx9iFgZso4CqyHPs7&" + "access_" + "key=ER51kMn" + "lmu3zXwcxczJ" + "MbYzJIgrY9O";
 | 
						|
 | 
						|
  EXPECT_EQ(false, NULL == data);
 | 
						|
 | 
						|
  EXPECT_EQ(0, dest.set(path.c_str()));
 | 
						|
 | 
						|
  const ObString &uri = dest.get_root_path();
 | 
						|
  const ObBackupStorageInfo *storage_info = dest.get_storage_info();
 | 
						|
 | 
						|
  // delete file
 | 
						|
  EXPECT_EQ(0, util.del_file(uri, storage_info));
 | 
						|
 | 
						|
  // scene 1: append & append
 | 
						|
  EXPECT_EQ(0, archive_io.push_log(uri, storage_info, data, part_size, 0, false /*append*/));
 | 
						|
  EXPECT_EQ(0, archive_io.push_log(uri, storage_info, data, total_size, 0, false /*append*/));
 | 
						|
  // delete file
 | 
						|
  EXPECT_EQ(0, util.del_file(uri, storage_info));
 | 
						|
 | 
						|
  // scene 2: append & put
 | 
						|
  EXPECT_EQ(0, archive_io.push_log(uri, storage_info, data, part_size, 0, false /*append*/));
 | 
						|
  EXPECT_EQ(0, archive_io.push_log(uri, storage_info, data, total_size, 0, true /*put*/));
 | 
						|
  // delete file
 | 
						|
  EXPECT_EQ(0, util.del_file(uri, storage_info));
 | 
						|
 | 
						|
  // scene 3: put & put
 | 
						|
  EXPECT_EQ(0, archive_io.push_log(uri, storage_info, data, total_size, 0, true /*put*/));
 | 
						|
  EXPECT_EQ(0, archive_io.push_log(uri, storage_info, data, total_size, 0, true /*put*/));
 | 
						|
  // delete file
 | 
						|
  EXPECT_EQ(0, util.del_file(uri, storage_info));
 | 
						|
 | 
						|
  // scene 4: put & append
 | 
						|
  EXPECT_EQ(0, archive_io.push_log(uri, storage_info, data, total_size, 0, true /*put*/));
 | 
						|
  EXPECT_EQ(0, archive_io.push_log(uri, storage_info, data, part_size, 0, false /*append*/));
 | 
						|
  // delete file
 | 
						|
  EXPECT_EQ(0, util.del_file(uri, storage_info));
 | 
						|
 | 
						|
  if (NULL != data) {
 | 
						|
    ob_free(data);
 | 
						|
    data = NULL;
 | 
						|
  }
 | 
						|
}
 | 
						|
 | 
						|
int main(int argc, char **argv)
 | 
						|
{
 | 
						|
  ObLogger::get_logger().set_file_name("test_archive_overwrite.log", true);
 | 
						|
  ObLogger::get_logger().set_log_level("INFO");
 | 
						|
  ::testing::InitGoogleTest(&argc, argv);
 | 
						|
  return RUN_ALL_TESTS();
 | 
						|
}
 |