/*
 * Copyright (c) 2015, 2016, Oracle and/or its affiliates. All rights reserved.
 * ORACLE PROPRIETARY/CONFIDENTIAL.  Use is subject to license terms.
 */

#include "precompiled.hpp"
#include "gc/z/zList.inline.hpp"
#include "unittest.hpp"

#ifndef PRODUCT

class ZTestEntry {
  friend class ZList<ZTestEntry>;

private:
  const int             _id;
  ZListNode<ZTestEntry> _node;

public:
  ZTestEntry(int id) :
      _id(id),
      _node() {}

  int id() const {
    return _id;
  }
};

class ZListTest : public ::testing::Test {
protected:
  static void assert_sorted(ZList<ZTestEntry>* list) {
    // Iterate forward
    {
      int count = list->first()->id();
      ZListIterator<ZTestEntry> iter(list);
      for (ZTestEntry* entry; iter.next(&entry);) {
        ASSERT_EQ(entry->id(), count);
        count++;
      }
    }

    // Iterate backward
    {
      int count = list->last()->id();
      ZListReverseIterator<ZTestEntry> iter(list);
      for (ZTestEntry* entry; iter.next(&entry);) {
        EXPECT_EQ(entry->id(), count);
        count--;
      }
    }
  }
};

TEST_F(ZListTest, test_insert) {
  ZList<ZTestEntry> list;
  ZTestEntry e0(0);
  ZTestEntry e1(1);
  ZTestEntry e2(2);
  ZTestEntry e3(3);
  ZTestEntry e4(4);
  ZTestEntry e5(5);

  list.insert_first(&e2);
  list.insert_before(&e2, &e1);
  list.insert_after(&e2, &e3);
  list.insert_last(&e4);
  list.insert_first(&e0);
  list.insert_last(&e5);

  EXPECT_EQ(list.size(), 6u);
  assert_sorted(&list);
}

TEST_F(ZListTest, test_remove) {
  // Remove first
  {
    ZList<ZTestEntry> list;
    ZTestEntry e0(0);
    ZTestEntry e1(1);
    ZTestEntry e2(2);
    ZTestEntry e3(3);
    ZTestEntry e4(4);
    ZTestEntry e5(5);

    list.insert_last(&e0);
    list.insert_last(&e1);
    list.insert_last(&e2);
    list.insert_last(&e3);
    list.insert_last(&e4);
    list.insert_last(&e5);

    EXPECT_EQ(list.size(), 6u);

    for (int i = 0; i < 6; i++) {
      ZTestEntry* e = list.remove_first();
      EXPECT_EQ(e->id(), i);
    }

    EXPECT_EQ(list.size(), 0u);
  }

  // Remove last
  {
    ZList<ZTestEntry> list;
    ZTestEntry e0(0);
    ZTestEntry e1(1);
    ZTestEntry e2(2);
    ZTestEntry e3(3);
    ZTestEntry e4(4);
    ZTestEntry e5(5);

    list.insert_last(&e0);
    list.insert_last(&e1);
    list.insert_last(&e2);
    list.insert_last(&e3);
    list.insert_last(&e4);
    list.insert_last(&e5);

    EXPECT_EQ(list.size(), 6u);

    for (int i = 5; i >= 0; i--) {
      ZTestEntry* e = list.remove_last();
      EXPECT_EQ(e->id(), i);
    }

    EXPECT_EQ(list.size(), 0u);
  }
}

TEST_F(ZListTest, test_transfer) {
  // Transfer empty to empty
  {
    ZList<ZTestEntry> list0;
    ZList<ZTestEntry> list1;

    EXPECT_TRUE(list0.is_empty());
    EXPECT_TRUE(list1.is_empty());

    list0.transfer(&list1);

    EXPECT_TRUE(list0.is_empty());
    EXPECT_TRUE(list1.is_empty());
  }

  // Transfer non-empty to empty
  {
    ZList<ZTestEntry> list0;
    ZList<ZTestEntry> list1;
    ZTestEntry e0(0);
    ZTestEntry e1(1);
    ZTestEntry e2(2);
    ZTestEntry e3(3);
    ZTestEntry e4(4);
    ZTestEntry e5(5);

    list1.insert_last(&e0);
    list1.insert_last(&e1);
    list1.insert_last(&e2);
    list1.insert_last(&e3);
    list1.insert_last(&e4);
    list1.insert_last(&e5);

    EXPECT_EQ(list0.size(), 0u);
    EXPECT_EQ(list1.size(), 6u);

    list0.transfer(&list1);

    EXPECT_EQ(list0.size(), 6u);
    EXPECT_EQ(list1.size(), 0u);

    assert_sorted(&list0);
  }

  // Transfer non-empty to non-empty
  {
    ZList<ZTestEntry> list0;
    ZList<ZTestEntry> list1;
    ZTestEntry e0(0);
    ZTestEntry e1(1);
    ZTestEntry e2(2);
    ZTestEntry e3(3);
    ZTestEntry e4(4);
    ZTestEntry e5(5);

    list0.insert_last(&e0);
    list0.insert_last(&e1);
    list0.insert_last(&e2);

    list1.insert_last(&e3);
    list1.insert_last(&e4);
    list1.insert_last(&e5);

    EXPECT_EQ(list0.size(), 3u);
    EXPECT_EQ(list1.size(), 3u);

    list0.transfer(&list1);

    EXPECT_EQ(list0.size(), 6u);
    EXPECT_EQ(list1.size(), 0u);

    assert_sorted(&list0);
  }
}

#endif // PRODUCT