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();
 | |
| }
 | 
