160 lines
3.4 KiB
C++
160 lines
3.4 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/container/ob_se_array.h"
|
|
#include <gtest/gtest.h>
|
|
using namespace oceanbase::common;
|
|
|
|
class ObSEArrayTest: public ::testing::Test
|
|
{
|
|
public:
|
|
ObSEArrayTest();
|
|
virtual ~ObSEArrayTest();
|
|
virtual void SetUp();
|
|
virtual void TearDown();
|
|
private:
|
|
// disallow copy
|
|
ObSEArrayTest(const ObSEArrayTest &other);
|
|
ObSEArrayTest& operator=(const ObSEArrayTest &other);
|
|
private:
|
|
// data members
|
|
};
|
|
|
|
ObSEArrayTest::ObSEArrayTest()
|
|
{
|
|
}
|
|
|
|
ObSEArrayTest::~ObSEArrayTest()
|
|
{
|
|
}
|
|
|
|
void ObSEArrayTest::SetUp()
|
|
{
|
|
}
|
|
|
|
void ObSEArrayTest::TearDown()
|
|
{
|
|
}
|
|
|
|
struct A
|
|
{
|
|
int64_t a_;
|
|
TO_STRING_EMPTY();
|
|
};
|
|
|
|
TEST_F(ObSEArrayTest, basic_test)
|
|
{
|
|
ObSEArray<A, 256> searray, searray2;
|
|
A a;
|
|
int NUMS[] = {256, 1024};
|
|
for (int k = 0; k < (int)ARRAYSIZEOF(NUMS); ++k)
|
|
{
|
|
int num = NUMS[k];
|
|
searray.reset();
|
|
searray2.reset();
|
|
for (int64_t i = 0; i < num; ++i)
|
|
{
|
|
a.a_ = i;
|
|
ASSERT_EQ(OB_SUCCESS, searray.push_back(a));
|
|
ASSERT_EQ(i+1, searray.count());
|
|
}
|
|
searray2 = searray;
|
|
for (int64_t i = 0; i < num; ++i)
|
|
{
|
|
ASSERT_EQ(i, searray2.at(i).a_);
|
|
}
|
|
searray.reset();
|
|
ASSERT_EQ(0, searray.count());
|
|
}
|
|
}
|
|
|
|
TEST_F(ObSEArrayTest, destroy)
|
|
{
|
|
ObSEArray<A, 8> searray;
|
|
for (int i = 0; i < 9; ++i)
|
|
{
|
|
A a;
|
|
a.a_ = i;
|
|
ASSERT_EQ(OB_SUCCESS, searray.push_back(a));
|
|
}
|
|
searray.destroy();
|
|
}
|
|
|
|
TEST(ObSEArrayTest2, array_remove)
|
|
{
|
|
ObSEArray<int, 10> ai;
|
|
for (int i = 0; i < 32; ++i)
|
|
{
|
|
ASSERT_EQ(OB_SUCCESS, ai.push_back(i));
|
|
}
|
|
ASSERT_EQ(32, ai.count());
|
|
|
|
ASSERT_EQ(OB_ARRAY_OUT_OF_RANGE, ai.remove(-1));
|
|
ASSERT_EQ(OB_ARRAY_OUT_OF_RANGE, ai.remove(32));
|
|
ASSERT_EQ(32, ai.count());
|
|
|
|
ASSERT_EQ(OB_SUCCESS, ai.remove(10));
|
|
int v = 0;
|
|
ASSERT_EQ(31, static_cast<int>(ai.count()));
|
|
for (int i = 0; i < 31; ++i)
|
|
{
|
|
ASSERT_EQ(OB_SUCCESS, ai.at(static_cast<int64_t>(i), v));
|
|
ASSERT_NE(10, v);
|
|
}
|
|
}
|
|
|
|
TEST(ObSEArrayTest2, serialize)
|
|
{
|
|
char buf[1024];
|
|
ObSEArray<int64_t, 100> array;
|
|
const int64_t NUM = 20;
|
|
for (int64_t i = 0; i < NUM; i ++)
|
|
{
|
|
array.push_back(i);
|
|
}
|
|
|
|
int64_t pos = 0;
|
|
array.serialize(buf, 1024, pos);
|
|
|
|
ObSEArray<int64_t, 100> array2;
|
|
int64_t data_len = pos;
|
|
pos = 0;
|
|
array2.push_back(30);
|
|
array2.deserialize(buf, data_len, pos);
|
|
|
|
ASSERT_EQ(array2.count(), NUM);
|
|
for (int64_t i = 0; i < NUM; i ++)
|
|
{
|
|
ASSERT_EQ(i, array2.at(i));
|
|
}
|
|
}
|
|
|
|
TEST_F(ObSEArrayTest, test_null_allocator_array)
|
|
{
|
|
const int64_t count = 10;
|
|
int64_t value = 0;
|
|
ObSEArray<int64_t, count, ObNullAllocator> array;
|
|
for (int i = 0; i < count; ++i) {
|
|
ASSERT_EQ(OB_SUCCESS, array.push_back(1));
|
|
}
|
|
ASSERT_EQ(OB_ALLOCATE_MEMORY_FAILED, array.push_back(1));
|
|
ASSERT_EQ(OB_SUCCESS, array.pop_back(value));
|
|
ASSERT_EQ(OB_SUCCESS, array.push_back(1));
|
|
|
|
}
|
|
|
|
int main(int argc, char **argv)
|
|
{
|
|
::testing::InitGoogleTest(&argc,argv);
|
|
return RUN_ALL_TESTS();
|
|
}
|