oceanbase/unittest/archive/test_archive_overwrite.cpp
2023-07-18 09:42:14 +00:00

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