//===----------------------------------------------------------------------===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//

// UNSUPPORTED: c++03

// <set>

// class set

// set(set&& s, const allocator_type& a);

#include <set>
#include <cassert>
#include <iterator>

#include "test_macros.h"
#include "MoveOnly.h"
#include "../../../test_compare.h"
#include "test_allocator.h"
#include "Counter.h"

int main(int, char**) {
  {
    typedef MoveOnly V;
    typedef test_less<MoveOnly> C;
    typedef test_allocator<V> A;
    typedef std::set<MoveOnly, C, A> M;
    typedef std::move_iterator<V*> I;
    V a1[] = {V(1), V(1), V(1), V(2), V(2), V(2), V(3), V(3), V(3)};
    M m1(I(a1), I(a1 + sizeof(a1) / sizeof(a1[0])), C(5), A(7));
    V a2[] = {V(1), V(1), V(1), V(2), V(2), V(2), V(3), V(3), V(3)};
    M m2(I(a2), I(a2 + sizeof(a2) / sizeof(a2[0])), C(5), A(7));
    M m3(std::move(m1), A(7));
    assert(m3 == m2);
    assert(m3.get_allocator() == A(7));
    assert(m3.key_comp() == C(5));
    LIBCPP_ASSERT(m1.empty());
  }
  {
    typedef MoveOnly V;
    typedef test_less<MoveOnly> C;
    typedef test_allocator<V> A;
    typedef std::set<MoveOnly, C, A> M;
    typedef std::move_iterator<V*> I;
    V a1[] = {V(1), V(1), V(1), V(2), V(2), V(2), V(3), V(3), V(3)};
    M m1(I(a1), I(a1 + sizeof(a1) / sizeof(a1[0])), C(5), A(7));
    V a2[] = {V(1), V(1), V(1), V(2), V(2), V(2), V(3), V(3), V(3)};
    M m2(I(a2), I(a2 + sizeof(a2) / sizeof(a2[0])), C(5), A(7));
    M m3(std::move(m1), A(5));
    assert(m3 == m2);
    assert(m3.get_allocator() == A(5));
    assert(m3.key_comp() == C(5));
    LIBCPP_ASSERT(m1.empty());
  }
  {
    typedef MoveOnly V;
    typedef test_less<MoveOnly> C;
    typedef other_allocator<V> A;
    typedef std::set<MoveOnly, C, A> M;
    typedef std::move_iterator<V*> I;
    V a1[] = {V(1), V(1), V(1), V(2), V(2), V(2), V(3), V(3), V(3)};
    M m1(I(a1), I(a1 + sizeof(a1) / sizeof(a1[0])), C(5), A(7));
    V a2[] = {V(1), V(1), V(1), V(2), V(2), V(2), V(3), V(3), V(3)};
    M m2(I(a2), I(a2 + sizeof(a2) / sizeof(a2[0])), C(5), A(7));
    M m3(std::move(m1), A(5));
    assert(m3 == m2);
    assert(m3.get_allocator() == A(5));
    assert(m3.key_comp() == C(5));
    LIBCPP_ASSERT(m1.empty());
  }
  {
    typedef Counter<int> V;
    typedef std::less<V> C;
    typedef test_allocator<V> A;
    typedef std::set<V, C, A> M;
    typedef V* I;
    Counter_base::gConstructed = 0;
    {
      V a1[]                = {V(1), V(1), V(1), V(2), V(2), V(2), V(3), V(3), V(3)};
      const std::size_t num = sizeof(a1) / sizeof(a1[0]);
      assert(Counter_base::gConstructed == num);

      M m1(I(a1), I(a1 + num), C(), A());
      assert(Counter_base::gConstructed == 3 + num);

      M m2(m1);
      assert(m2 == m1);
      assert(Counter_base::gConstructed == 6 + num);

      M m3(std::move(m1), A());
      assert(m3 == m2);
      LIBCPP_ASSERT(m1.empty());
      assert(Counter_base::gConstructed >= (int)(6 + num));
      assert(Counter_base::gConstructed <= (int)(m1.size() + 6 + num));

      {
        M m4(std::move(m2), A(5));
        assert(Counter_base::gConstructed >= (int)(6 + num));
        assert(Counter_base::gConstructed <= (int)(m1.size() + m2.size() + 6 + num));
        assert(m4 == m3);
        LIBCPP_ASSERT(m2.empty());
      }
      assert(Counter_base::gConstructed >= (int)(3 + num));
      assert(Counter_base::gConstructed <= (int)(m1.size() + m2.size() + 3 + num));
    }
    assert(Counter_base::gConstructed == 0);
  }

  return 0;
}
