// ArduinoJson - https://arduinojson.org
// Copyright © 2014-2023, Benoit BLANCHON
// MIT License

#include <ArduinoJson/Memory/Alignment.hpp>
#include <ArduinoJson/Memory/ResourceManager.hpp>
#include <ArduinoJson/Memory/VariantPoolImpl.hpp>
#include <catch.hpp>

#include "Allocators.hpp"

using namespace ArduinoJson::detail;

static void fullPreallocatedPools(ResourceManager& resources) {
  for (int i = 0;
       i < ARDUINOJSON_INITIAL_POOL_COUNT * ARDUINOJSON_POOL_CAPACITY; i++)
    resources.allocSlot();
}

TEST_CASE("ResourceManager::swap()") {
  SECTION("Both using preallocated pool list") {
    SpyingAllocator allocator;
    ResourceManager a(&allocator);
    ResourceManager b(&allocator);

    auto a1 = a.allocSlot();
    auto b1 = b.allocSlot();

    swap(a, b);

    REQUIRE(a1->data() == b.getSlot(a1.id())->data());
    REQUIRE(b1->data() == a.getSlot(b1.id())->data());

    REQUIRE(allocator.log() == AllocatorLog()
                                   << AllocatorLog::Allocate(sizeofPool()) * 2);
  }

  SECTION("Only left using preallocated pool list") {
    SpyingAllocator allocator;
    ResourceManager a(&allocator);
    ResourceManager b(&allocator);
    fullPreallocatedPools(b);

    auto a1 = a.allocSlot();
    auto b1 = b.allocSlot();
    swap(a, b);

    REQUIRE(a1->data() == b.getSlot(a1.id())->data());
    REQUIRE(b1->data() == a.getSlot(b1.id())->data());

    REQUIRE(allocator.log() == AllocatorLog()
                                   << AllocatorLog::Allocate(sizeofPool()) *
                                          (ARDUINOJSON_INITIAL_POOL_COUNT + 1)
                                   << AllocatorLog::Allocate(sizeofPoolList(
                                          ARDUINOJSON_INITIAL_POOL_COUNT * 2))
                                   << AllocatorLog::Allocate(sizeofPool()));
  }

  SECTION("Only right using preallocated pool list") {
    SpyingAllocator allocator;
    ResourceManager a(&allocator);
    fullPreallocatedPools(a);
    ResourceManager b(&allocator);

    auto a1 = a.allocSlot();
    auto b1 = b.allocSlot();
    swap(a, b);

    REQUIRE(a1->data() == b.getSlot(a1.id())->data());
    REQUIRE(b1->data() == a.getSlot(b1.id())->data());

    REQUIRE(allocator.log() == AllocatorLog()
                                   << AllocatorLog::Allocate(sizeofPool()) *
                                          ARDUINOJSON_INITIAL_POOL_COUNT
                                   << AllocatorLog::Allocate(sizeofPoolList(
                                          ARDUINOJSON_INITIAL_POOL_COUNT * 2))
                                   << AllocatorLog::Allocate(sizeofPool()) * 2);
  }

  SECTION("None is using preallocated pool list") {
    SpyingAllocator allocator;
    ResourceManager a(&allocator);
    fullPreallocatedPools(a);
    ResourceManager b(&allocator);
    fullPreallocatedPools(b);

    auto a1 = a.allocSlot();
    auto b1 = b.allocSlot();

    swap(a, b);

    REQUIRE(a1->data() == b.getSlot(a1.id())->data());
    REQUIRE(b1->data() == a.getSlot(b1.id())->data());
  }
}