init push

deps/oblib/unittest/lib/container/test_iarray.cpp

@@ -0,0 +1,146 @@
+/**
+ * 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 <gtest/gtest.h>
+#include "lib/utility/ob_test_util.h"
+#include "lib/container/ob_array.h"
+#include "lib/container/ob_se_array.h"
+#include "lib/container/ob_2d_array.h"
+using namespace oceanbase::common;
+
+class TestIArray : public ::testing::Test {
+  public:
+  TestIArray();
+  virtual ~TestIArray();
+  virtual void SetUp();
+  virtual void TearDown();
+
+  private:
+  // disallow copy
+  DISALLOW_COPY_AND_ASSIGN(TestIArray);
+
+  protected:
+  // function members
+  template <typename T>
+  void setup(int32_t N, T& arr);
+  void verify(int32_t N, const ObIArray<int32_t>& arr);
+
+  protected:
+  // data members
+};
+
+TestIArray::TestIArray()
+{}
+
+TestIArray::~TestIArray()
+{}
+
+void TestIArray::SetUp()
+{}
+
+void TestIArray::TearDown()
+{}
+
+template <typename T>
+void TestIArray::setup(int32_t N, T& arr)
+{
+  arr.reset();
+  for (int32_t i = 0; i < N; ++i) {
+    OK(arr.push_back(i));
+  }  // end for
+}
+
+void TestIArray::verify(int32_t N, const ObIArray<int32_t>& arr)
+{
+  ASSERT_EQ(N, arr.count());
+  for (int32_t i = 0; i < N; ++i) {
+    ASSERT_EQ(i, arr.at(i));
+  }
+}
+
+TEST_F(TestIArray, array_assign)
+{
+  ObArray<int32_t> se;
+
+  ObArray<int32_t> arr1;
+  ObSEArray<int32_t, 8> arr2;
+
+  int32_t sizes[2] = {8, 4096};
+
+  for (int32_t i = 0; i < 2; ++i) {
+    int32_t N = sizes[i];
+    setup(N, arr1);
+    setup(N, arr2);
+    OK(se.assign(arr1));
+    OK(se.assign(arr2));
+  }
+}
+
+TEST_F(TestIArray, se_array_assign)
+{
+  ObSEArray<int32_t, 8> se;
+
+  ObArray<int32_t> arr1;
+  ObSEArray<int32_t, 4> arr2;
+  ObSEArray<int32_t, 8> arr4;
+
+  int32_t sizes[3] = {4, 8, 4096};
+
+  for (int32_t i = 0; i < 3; ++i) {
+    int32_t N = sizes[i];
+    setup(N, arr1);
+    setup(N, arr2);
+    setup(N, arr4);
+    OK(se.assign(arr1));
+    OK(se.assign(arr2));
+    OK(se.assign(arr4));
+  }
+}
+
+TEST_F(TestIArray, 2d_array_assign)
+{
+  Ob2DArray<int32_t> se;
+
+  ObArray<int32_t> arr1;
+  ObSEArray<int32_t, 8> arr2;
+  Ob2DArray<int32_t> arr3;
+
+  int32_t sizes[2] = {8, 4096};
+
+  for (int32_t i = 0; i < 2; ++i) {
+    int32_t N = sizes[i];
+    setup(N, arr1);
+    setup(N, arr2);
+    setup(N, arr3);
+    OK(se.assign(arr1));
+    OK(se.assign(arr2));
+    OK(se.assign(arr3));
+  }
+}
+TEST_F(TestIArray, array_append)
+{
+  ObArray<int32_t> arr1;
+  ObSEArray<int32_t, 8> arr2;
+  OK(arr1.push_back(0));
+  OK(arr2.push_back(1));
+  OK(append(arr1, arr2));
+  ASSERT_EQ(2, arr1.count());
+  for (int32_t i = 0; i < 2; ++i) {
+    ASSERT_EQ(i, arr1.at(i));
+  }
+}
+
+int main(int argc, char** argv)
+{
+  ::testing::InitGoogleTest(&argc, argv);
+  return RUN_ALL_TESTS();
+}