tdesktop/Telegram/SourceFiles/base/flat_set_tests.cpp

/*
This file is part of Telegram Desktop,
the official desktop application for the Telegram messaging service.

For license and copyright information please follow this link:
https://github.com/telegramdesktop/tdesktop/blob/master/LEGAL
*/
#include "catch.hpp"

#include "base/flat_set.h"

struct int_wrap {
	int value;
};
struct int_wrap_comparator {
	using is_transparent = void;
	inline bool operator()(const int &a, const int_wrap &b) const {
		return a < b.value;
	}
	inline bool operator()(const int_wrap &a, const int_wrap &b) const {
		return a.value < b.value;
	}
	inline bool operator()(const int_wrap &a, const int &b) const {
		return a.value < b;
	}
	inline bool operator()(const int &a, const int &b) const {
		return a < b;
	}
};

TEST_CASE("flat_sets should keep items sorted", "[flat_set]") {

	base::flat_set<int> v;
	v.insert(0);
	v.insert(5);
	v.insert(4);
	v.insert(2);

	REQUIRE(v.contains(4));

	auto checkSorted = [&] {
		auto prev = v.begin();
		REQUIRE(prev != v.end());
		for (auto i = prev + 1; i != v.end(); prev = i, ++i) {
			REQUIRE(*prev < *i);
		}
	};
	REQUIRE(v.size() == 4);
	checkSorted();

	SECTION("adding item puts it in the right position") {
		v.insert(3);
		REQUIRE(v.size() == 5);
		REQUIRE(v.find(3) != v.end());
		checkSorted();
	}
}

TEST_CASE("flat_sets with custom comparators", "[flat_set]") {
	base::flat_set<int_wrap, int_wrap_comparator> v;
	v.insert({ 0 });
	v.insert({ 5 });
	v.insert({ 4 });
	v.insert({ 2 });

	REQUIRE(v.find(4) != v.end());

	auto checkSorted = [&] {
		auto prev = v.begin();
		REQUIRE(prev != v.end());
		for (auto i = prev + 1; i != v.end(); prev = i, ++i) {
			REQUIRE(prev->value < i->value);
		}
	};
	REQUIRE(v.size() == 4);
	checkSorted();

	SECTION("adding item puts it in the right position") {
		v.insert({ 3 });
		REQUIRE(v.size() == 5);
		REQUIRE(v.find(3) != v.end());
		checkSorted();
	}
}