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

class MockAllocator : public ObIAllocator
{
public:
  MockAllocator(const oceanbase::lib::ObLabel &label = ObModIds::TEST,
      int64_t tenant_id = OB_SERVER_TENANT_ID) : alloc_count_(0)
  {
    UNUSED(label);
    UNUSED(tenant_id);
  }
  MockAllocator(ObIAllocator &allocator)
  {
    UNUSED(allocator);
  }
  void *alloc(const int64_t sz)
  {
    void *ptr = nullptr;
    // only the first memory allocation can succeed
    if (0 == alloc_count_) {
      ptr = ob_malloc(sz, "MockAlloc");
      alloc_count_++;
    }
    return ptr;
  }
  void *alloc(const int64_t sz, const ObMemAttr &attr)
  {
    UNUSEDx(sz, attr);
    return nullptr;
  }
  void free(void *p) { ob_free(p); }

  int64_t alloc_count_;
};

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));
    verify(N, se);

    OK(se.assign(arr2));
    verify(N, se);
  }
}

TEST_F(TestIArray, array_assign_alloc_memory_failed)
{
  ObArray<int32_t, MockAllocator> se;
  int32_t origin_size = 4;
  setup(origin_size, se);

  ObArray<int32_t> arr1;
  ObSEArray<int32_t, 8> arr2;
  int32_t N = 4096;
  setup(N, arr1);
  setup(N, arr2);

  ASSERT_EQ(OB_ALLOCATE_MEMORY_FAILED, se.assign(arr1));
  ASSERT_EQ(OB_ALLOCATE_MEMORY_FAILED, se.assign(arr2));
  verify(origin_size, se);
}

TEST_F(TestIArray, empty_array_assign)
{
  ObArray<int32_t> se;
  int32_t origin_size = 4;
  setup(origin_size, se);

  ObArray<int32_t> arr1;
  OK(se.assign(arr1));
  ASSERT_EQ(0, se.count());
}

bool DECONSTRUCT = false;
struct MockElement
{
public:
  MockElement() : value_(0) {}
  MockElement(int32_t v) : value_(v) {}
  ~MockElement() { if (0 == value_) DECONSTRUCT = true; }
  int assign(const MockElement &other)
  {
    // the function will return error if assign -1
    if (-1 == other.value_) return OB_ERROR;
    this->value_ = other.value_;
    return OB_SUCCESS;
  }
  TO_STRING_KV(K_(value));

  int32_t value_;
};

TEST_F(TestIArray, array_assign_construct_failed)
{
  ObArray<MockElement> se;
  se.push_back(MockElement(1));

  ObArray<MockElement> arr1;
  arr1.push_back(MockElement(2));
  arr1.push_back(MockElement(3));

  arr1[0].value_ = 0;  // DECONSTRUCT will be set to true when deconstruct arr1[0]
  arr1[1].value_ = -1;  // construct_assign will fail when assign arr1[1]

  ASSERT_FALSE(DECONSTRUCT);
  ASSERT_EQ(OB_ERROR, se.assign(arr1));
  ASSERT_TRUE(DECONSTRUCT);
  ASSERT_EQ(1, se.count());
  ASSERT_EQ(1, se[0].value_);
}

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