memgraph/tests/unit/concurrent_map.cpp

#include <iostream>

#include <glog/logging.h>
#include <gtest/gtest.h>

#include "data_structures/concurrent/concurrent_map.hpp"
#include "utils/assert.hpp"

using concurrent_map_t = ConcurrentMap<int, int>;

void print_skiplist(const concurrent_map_t::Accessor &map) {
  DLOG(INFO) << "Map now has: ";
  for (auto &kv : map)
    DLOG(INFO) << fmt::format("    ({}, {})", kv.first, kv.second);
}

TEST(ConcurrentMapSkiplist, Mix) {
  concurrent_map_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) {
  google::InitGoogleLogging(argv[0]);

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