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