#include <iostream>

#include "gtest/gtest.h"

#include "logging/default.hpp"
#include "logging/streams/stdout.hpp"
#include "data_structures/concurrent/concurrent_map.hpp"
#include "logging/default.hpp"
#include "logging/streams/stdout.hpp"
#include "utils/assert.hpp"

using skiplist_t = ConcurrentMap<int, int>;

void print_skiplist(const skiplist_t::Accessor &skiplist)
{
    logging::info("Skiplist now has: ");

    for (auto &kv : skiplist)
        logging::info("    ({}, {})", kv.first, kv.second);
}

TEST(ConcurrentMapSkiplist, Mix)
{
    skiplist_t skiplist;
    auto accessor = skiplist.access();

    // insert 10
    permanent_assert(accessor.insert(1, 10).second == true,
                     "add first element");

    // try insert 10 again (should fail)
    permanent_assert(accessor.insert(1, 10).second == false,
                     "add the same element, should fail");

    // insert 20
    permanent_assert(accessor.insert(2, 20).second == true,
                     "insert new unique element");

    print_skiplist(accessor);

    // value at key 3 shouldn't exist
    permanent_assert((accessor.find(3) == accessor.end()) == true,
                     "try to find element which doesn't exist");

    // value at key 2 should exist
    permanent_assert((accessor.find(2) != accessor.end()) == true,
                     "find iterator");

    // at key 2 is 20 (true)
    permanent_assert(accessor.find(2)->second == 20, "find element");

    // removed existing (1)
    permanent_assert(accessor.remove(1) == true, "try to remove element");

    // removed non-existing (3)
    permanent_assert(accessor.remove(3) == false,
                     "try to remove element which doesn't exist");

    // insert (1, 10)
    permanent_assert(accessor.insert(1, 10).second == true,
                     "insert unique element");

    // insert (4, 40)
    permanent_assert(accessor.insert(4, 40).second == true,
                     "insert unique element");

    print_skiplist(accessor);
}

int main(int argc, char **argv)
{
    logging::init_async();
    logging::log->pipe(std::make_unique<Stdout>());

    ::testing::InitGoogleTest(&argc, argv);
    return RUN_ALL_TESTS();
}