Files
scylladb/test/boost/wrapping_interval_test.cc
Avi Kivity 8815491085 treewide: include boost headers as "system" headers
Boost is external to the project so treat its headers as "system"
headers and include them with angle brackets.

Closes scylladb/scylladb#25619
2025-08-22 17:21:24 +03:00

633 lines
31 KiB
C++

/*
* Copyright (C) 2015-present ScyllaDB
*/
/*
* SPDX-License-Identifier: LicenseRef-ScyllaDB-Source-Available-1.0
*/
#define BOOST_TEST_MODULE core
#include <boost/test/unit_test.hpp>
#include <boost/icl/interval_map.hpp>
#include <fmt/ranges.h>
#include <fmt/std.h>
#include <fmt/ranges.h>
#include "schema/schema_builder.hh"
#include "locator/token_metadata.hh"
#include "utils/to_string.hh"
#include "test/lib/test_utils.hh"
// yuck, but what can one do? needed for BOOST_REQUIRE_EQUAL
namespace std {
ostream&
operator<<(ostream& os, const std::nullopt_t&) {
return os << "{}";
}
}
static
bool includes_token(const schema& s, const dht::partition_range& r, const dht::token& tok) {
dht::ring_position_comparator cmp(s);
return !r.before(dht::ring_position(tok, dht::ring_position::token_bound::end), cmp)
&& !r.after(dht::ring_position(tok, dht::ring_position::token_bound::start), cmp);
}
BOOST_AUTO_TEST_CASE(test_range_with_positions_within_the_same_token) {
auto s = schema_builder("ks", "cf")
.with_column("key", bytes_type, column_kind::partition_key)
.with_column("v", bytes_type)
.build();
dht::token tok = dht::token::get_random_token();
auto key1 = dht::decorated_key{tok,
partition_key::from_single_value(*s, bytes_type->decompose(data_value(bytes("key1"))))};
auto key2 = dht::decorated_key{tok,
partition_key::from_single_value(*s, bytes_type->decompose(data_value(bytes("key2"))))};
{
auto r = dht::partition_range::make(
dht::ring_position(key1),
dht::ring_position(key2));
BOOST_REQUIRE(includes_token(*s, r, tok));
BOOST_REQUIRE(r.contains(key1, dht::ring_position_comparator(*s)));
BOOST_REQUIRE(r.contains(key2, dht::ring_position_comparator(*s)));
}
{
auto r = dht::partition_range::make(
{dht::ring_position(key1), false},
{dht::ring_position(key2), false});
BOOST_REQUIRE(includes_token(*s, r, tok));
BOOST_REQUIRE(!r.contains(key1, dht::ring_position_comparator(*s)));
BOOST_REQUIRE(!r.contains(key2, dht::ring_position_comparator(*s)));
}
{
auto r = dht::partition_range::make(
{dht::ring_position::starting_at(tok), false},
{dht::ring_position(key2), true});
BOOST_REQUIRE(includes_token(*s, r, tok));
BOOST_REQUIRE(r.contains(key1, dht::ring_position_comparator(*s)));
BOOST_REQUIRE(r.contains(key2, dht::ring_position_comparator(*s)));
}
{
auto r = dht::partition_range::make(
{dht::ring_position::starting_at(tok), true},
{dht::ring_position::ending_at(tok), true});
BOOST_REQUIRE(includes_token(*s, r, tok));
BOOST_REQUIRE(r.contains(key1, dht::ring_position_comparator(*s)));
BOOST_REQUIRE(r.contains(key2, dht::ring_position_comparator(*s)));
}
}
BOOST_AUTO_TEST_CASE(test_range_contains) {
auto cmp = [] (int i1, int i2) -> std::strong_ordering { return i1 <=> i2; };
auto check_contains = [&] (wrapping_interval<int> enclosing, wrapping_interval<int> enclosed) {
BOOST_REQUIRE(enclosing.contains(enclosed, cmp));
BOOST_REQUIRE(!enclosed.contains(enclosing, cmp));
};
BOOST_REQUIRE(wrapping_interval<int>({}, {}).contains(wrapping_interval<int>({}, {}), cmp));
check_contains(wrapping_interval<int>({}, {}), wrapping_interval<int>({1}, {2}));
check_contains(wrapping_interval<int>({}, {}), wrapping_interval<int>({}, {2}));
check_contains(wrapping_interval<int>({}, {}), wrapping_interval<int>({1}, {}));
check_contains(wrapping_interval<int>({}, {}), wrapping_interval<int>({2}, {1}));
BOOST_REQUIRE(wrapping_interval<int>({}, {3}).contains(wrapping_interval<int>({}, {3}), cmp));
BOOST_REQUIRE(wrapping_interval<int>({3}, {}).contains(wrapping_interval<int>({3}, {}), cmp));
BOOST_REQUIRE(wrapping_interval<int>({}, {{3, false}}).contains(wrapping_interval<int>({}, {{3, false}}), cmp));
BOOST_REQUIRE(wrapping_interval<int>({{3, false}}, {}).contains(wrapping_interval<int>({{3, false}}, {}), cmp));
BOOST_REQUIRE(wrapping_interval<int>({1}, {3}).contains(wrapping_interval<int>({1}, {3}), cmp));
BOOST_REQUIRE(wrapping_interval<int>({3}, {1}).contains(wrapping_interval<int>({3}, {1}), cmp));
check_contains(wrapping_interval<int>({}, {3}), wrapping_interval<int>({}, {2}));
check_contains(wrapping_interval<int>({}, {3}), wrapping_interval<int>({2}, {{3, false}}));
BOOST_REQUIRE(!wrapping_interval<int>({}, {3}).contains(wrapping_interval<int>({}, {4}), cmp));
BOOST_REQUIRE(!wrapping_interval<int>({}, {3}).contains(wrapping_interval<int>({2}, {{4, false}}), cmp));
BOOST_REQUIRE(!wrapping_interval<int>({}, {3}).contains(wrapping_interval<int>({}, {}), cmp));
check_contains(wrapping_interval<int>({3}, {}), wrapping_interval<int>({4}, {}));
check_contains(wrapping_interval<int>({3}, {}), wrapping_interval<int>({{3, false}}, {}));
check_contains(wrapping_interval<int>({3}, {}), wrapping_interval<int>({3}, {4}));
check_contains(wrapping_interval<int>({3}, {}), wrapping_interval<int>({4}, {5}));
BOOST_REQUIRE(!wrapping_interval<int>({3}, {}).contains(wrapping_interval<int>({2}, {4}), cmp));
BOOST_REQUIRE(!wrapping_interval<int>({3}, {}).contains(wrapping_interval<int>({}, {}), cmp));
check_contains(wrapping_interval<int>({}, {{3, false}}), wrapping_interval<int>({}, {2}));
BOOST_REQUIRE(!wrapping_interval<int>({}, {{3, false}}).contains(wrapping_interval<int>({}, {3}), cmp));
BOOST_REQUIRE(!wrapping_interval<int>({}, {{3, false}}).contains(wrapping_interval<int>({}, {4}), cmp));
check_contains(wrapping_interval<int>({1}, {3}), wrapping_interval<int>({1}, {2}));
check_contains(wrapping_interval<int>({1}, {3}), wrapping_interval<int>({1}, {1}));
BOOST_REQUIRE(!wrapping_interval<int>({1}, {3}).contains(wrapping_interval<int>({2}, {4}), cmp));
BOOST_REQUIRE(!wrapping_interval<int>({1}, {3}).contains(wrapping_interval<int>({0}, {1}), cmp));
BOOST_REQUIRE(!wrapping_interval<int>({1}, {3}).contains(wrapping_interval<int>({0}, {4}), cmp));
check_contains(wrapping_interval<int>({3}, {1}), wrapping_interval<int>({0}, {1}));
check_contains(wrapping_interval<int>({3}, {1}), wrapping_interval<int>({3}, {4}));
check_contains(wrapping_interval<int>({3}, {1}), wrapping_interval<int>({}, {1}));
check_contains(wrapping_interval<int>({3}, {1}), wrapping_interval<int>({}, {{1, false}}));
check_contains(wrapping_interval<int>({3}, {1}), wrapping_interval<int>({3}, {}));
check_contains(wrapping_interval<int>({3}, {1}), wrapping_interval<int>({{3, false}}, {}));
check_contains(wrapping_interval<int>({3}, {1}), wrapping_interval<int>({{3, false}}, {{1, false}}));
check_contains(wrapping_interval<int>({3}, {1}), wrapping_interval<int>({{3, false}}, {1}));
check_contains(wrapping_interval<int>({3}, {1}), wrapping_interval<int>({3}, {{1, false}}));
BOOST_REQUIRE(!wrapping_interval<int>({3}, {1}).contains(wrapping_interval<int>({2}, {2}), cmp));
BOOST_REQUIRE(!wrapping_interval<int>({3}, {1}).contains(wrapping_interval<int>({2}, {{3, false}}), cmp));
BOOST_REQUIRE(!wrapping_interval<int>({3}, {1}).contains(wrapping_interval<int>({{1, false}}, {{3, false}}), cmp));
BOOST_REQUIRE(!wrapping_interval<int>({3}, {1}).contains(wrapping_interval<int>({{1, false}}, {3}), cmp));
}
BOOST_AUTO_TEST_CASE(test_range_subtract) {
auto cmp = [] (int i1, int i2) -> std::strong_ordering { return i1 <=> i2; };
using r = wrapping_interval<int>;
using vec = std::vector<r>;
BOOST_REQUIRE_EQUAL(r({2}, {4}).subtract(r({0}, {1}), cmp), vec({r({2}, {4})}));
BOOST_REQUIRE_EQUAL(r({2}, {4}).subtract(r({}, {1}), cmp), vec({r({2}, {4})}));
BOOST_REQUIRE_EQUAL(r({2}, {4}).subtract(r({}, {2}), cmp), vec({r({{2, false}}, {4})}));
BOOST_REQUIRE_EQUAL(r({2}, {4}).subtract(r({}, {3}), cmp), vec({r({{3, false}}, {4})}));
BOOST_REQUIRE_EQUAL(r({2}, {4}).subtract(r({}, {4}), cmp), vec());
BOOST_REQUIRE_EQUAL(r({2}, {4}).subtract(r({1}, {4}), cmp), vec());
BOOST_REQUIRE_EQUAL(r({2}, {4}).subtract(r({1}, {3}), cmp), vec({r({{3, false}}, {4})}));
BOOST_REQUIRE_EQUAL(r({2}, {4}).subtract(r({1}, {{3, false}}), cmp), vec({r({3}, {4})}));
BOOST_REQUIRE_EQUAL(r({2}, {4}).subtract(r({2}, {4}), cmp), vec());
BOOST_REQUIRE_EQUAL(r({2}, {4}).subtract(r({}, {{4, false}}), cmp), vec({r({4}, {4})}));
BOOST_REQUIRE_EQUAL(r({2}, {4}).subtract(r({}, {}), cmp), vec());
BOOST_REQUIRE_EQUAL(r({2}, {4}).subtract(r({{2, false}}, {}), cmp), vec({r({2}, {2})}));
BOOST_REQUIRE_EQUAL(r({2}, {4}).subtract(r({{2, false}}, {4}), cmp), vec({r({2}, {2})}));
BOOST_REQUIRE_EQUAL(r({2}, {4}).subtract(r({3}, {5}), cmp), vec({r({2}, {{3, false}})}));
BOOST_REQUIRE_EQUAL(r({2}, {4}).subtract(r({3}, {1}), cmp), vec({r({2}, {{3, false}})}));
BOOST_REQUIRE_EQUAL(r({2}, {4}).subtract(r({5}, {1}), cmp), vec({r({2}, {4})}));
BOOST_REQUIRE_EQUAL(r({2}, {4}).subtract(r::make_singular(3), cmp), vec({r({2}, {{3, false}}), r({{3, false}}, {4})}));
BOOST_REQUIRE_EQUAL(r({2}, {4}).subtract(r::make_singular(4), cmp), vec({r({2}, {{4, false}})}));
BOOST_REQUIRE_EQUAL(r({2}, {4}).subtract(r::make_singular(5), cmp), vec({r({2}, {4})}));
BOOST_REQUIRE_EQUAL(r({}, {4}).subtract(r({3}, {5}), cmp), vec({r({}, {{3, false}})}));
BOOST_REQUIRE_EQUAL(r({}, {4}).subtract(r({5}, {}), cmp), vec({r({}, {4})}));
BOOST_REQUIRE_EQUAL(r({}, {4}).subtract(r({5}, {6}), cmp), vec({r({}, {4})}));
BOOST_REQUIRE_EQUAL(r({}, {4}).subtract(r({5}, {2}), cmp), vec({r({{2, false}}, {4})}));
BOOST_REQUIRE_EQUAL(r({4}, {}).subtract(r({3}, {5}), cmp), vec({r({{5, false}}, {})}));
BOOST_REQUIRE_EQUAL(r({4}, {}).subtract(r({1}, {3}), cmp), vec({r({4}, {})}));
BOOST_REQUIRE_EQUAL(r({4}, {}).subtract(r({}, {3}), cmp), vec({r({4}, {})}));
BOOST_REQUIRE_EQUAL(r({4}, {}).subtract(r({7}, {5}), cmp), vec({r({{5, false}}, {{7, false}})}));
BOOST_REQUIRE_EQUAL(r({5}, {1}).subtract(r({6}, {}), cmp), vec({r({}, {1}), r({5}, {{6, false}})}));
BOOST_REQUIRE_EQUAL(r({5}, {1}).subtract(r({6}, {1}), cmp), vec({r({5}, {{6, false}})}));
BOOST_REQUIRE_EQUAL(r({5}, {1}).subtract(r({6}, {2}), cmp), vec({r({5}, {{6, false}})}));
// FIXME: Also accept adjacent ranges merged
BOOST_REQUIRE_EQUAL(r({5}, {1}).subtract(r({4}, {7}), cmp), vec({r({}, {1}), r({{7, false}}, {})}));
// FIXME: Also accept adjacent ranges merged
BOOST_REQUIRE_EQUAL(r({5}, {1}).subtract(r({6}, {7}), cmp), vec({r({}, {1}), r({5}, {{6, false}}), r({{7, false}}, {})}));
BOOST_REQUIRE_EQUAL(r({8}, {3}).subtract(r({2}, {1}), cmp), vec({r({{1, false}}, {{2, false}})}));
BOOST_REQUIRE_EQUAL(r({8}, {3}).subtract(r({10}, {9}), cmp), vec({r({{9, false}}, {{10, false}})}));
BOOST_REQUIRE_EQUAL(r({5}, {1}).subtract(r({6}, {0}), cmp), vec({r({{0, false}}, {1}), r({5}, {{6, false}})}));
BOOST_REQUIRE_EQUAL(r({5}, {1}).subtract(r({}, {0}), cmp), vec({r({{0, false}}, {1}), r({5}, {})}));
}
struct unsigned_comparator {
std::strong_ordering operator()(unsigned u1, unsigned u2) const {
return u1 <=> u2;
}
};
BOOST_AUTO_TEST_CASE(range_overlap_tests) {
unsigned min = 0;
unsigned max = std::numeric_limits<unsigned>::max();
auto range0 = wrapping_interval<unsigned>::make(max, max);
auto range1 = wrapping_interval<unsigned>::make(min, max);
BOOST_REQUIRE(range0.overlaps(range1, unsigned_comparator()) == true);
BOOST_REQUIRE(range1.overlaps(range1, unsigned_comparator()) == true);
BOOST_REQUIRE(range1.overlaps(range0, unsigned_comparator()) == true);
auto range2 = wrapping_interval<unsigned>::make(1, max);
BOOST_REQUIRE(range1.overlaps(range2, unsigned_comparator()) == true);
auto range3 = wrapping_interval<unsigned>::make(min, max-2);
BOOST_REQUIRE(range2.overlaps(range3, unsigned_comparator()) == true);
auto range4 = wrapping_interval<unsigned>::make(2, 10);
auto range5 = wrapping_interval<unsigned>::make(12, 20);
auto range6 = wrapping_interval<unsigned>::make(22, 40);
BOOST_REQUIRE(range4.overlaps(range5, unsigned_comparator()) == false);
BOOST_REQUIRE(range5.overlaps(range4, unsigned_comparator()) == false);
BOOST_REQUIRE(range4.overlaps(range6, unsigned_comparator()) == false);
auto range7 = wrapping_interval<unsigned>::make(2, 10);
auto range8 = wrapping_interval<unsigned>::make(10, 20);
auto range9 = wrapping_interval<unsigned>::make(min, 100);
BOOST_REQUIRE(range7.overlaps(range8, unsigned_comparator()) == true);
BOOST_REQUIRE(range6.overlaps(range8, unsigned_comparator()) == false);
BOOST_REQUIRE(range8.overlaps(range9, unsigned_comparator()) == true);
// wrap around checks
auto range10 = wrapping_interval<unsigned>::make(25, 15);
BOOST_REQUIRE(range9.overlaps(range10, unsigned_comparator()) == true);
BOOST_REQUIRE(range10.overlaps(wrapping_interval<unsigned>({20}, {18}), unsigned_comparator()) == true);
auto range11 = wrapping_interval<unsigned>({}, {2});
BOOST_REQUIRE(range11.overlaps(range10, unsigned_comparator()) == true);
auto range12 = wrapping_interval<unsigned>::make(18, 20);
BOOST_REQUIRE(range12.overlaps(range10, unsigned_comparator()) == false);
BOOST_REQUIRE(wrapping_interval<unsigned>({1}, {{2, false}}).overlaps(wrapping_interval<unsigned>({2}, {3}), unsigned_comparator()) == false);
// open and infinite bound checks
BOOST_REQUIRE(wrapping_interval<unsigned>({1}, {}).overlaps(wrapping_interval<unsigned>({2}, {3}), unsigned_comparator()) == true);
BOOST_REQUIRE(wrapping_interval<unsigned>({5}, {}).overlaps(wrapping_interval<unsigned>({2}, {3}), unsigned_comparator()) == false);
BOOST_REQUIRE(wrapping_interval<unsigned>({}, {{3, false}}).overlaps(wrapping_interval<unsigned>({2}, {3}), unsigned_comparator()) == true);
BOOST_REQUIRE(wrapping_interval<unsigned>({}, {{2, false}}).overlaps(wrapping_interval<unsigned>({2}, {3}), unsigned_comparator()) == false);
BOOST_REQUIRE(wrapping_interval<unsigned>({}, {2}).overlaps(wrapping_interval<unsigned>({2}, {}), unsigned_comparator()) == true);
BOOST_REQUIRE(wrapping_interval<unsigned>({}, {2}).overlaps(wrapping_interval<unsigned>({3}, {4}), unsigned_comparator()) == false);
// [3,4] and [4,5]
BOOST_REQUIRE(wrapping_interval<unsigned>({3}, {4}).overlaps(wrapping_interval<unsigned>({4}, {5}), unsigned_comparator()) == true);
// [3,4) and [4,5]
BOOST_REQUIRE(wrapping_interval<unsigned>({3}, {{4, false}}).overlaps(wrapping_interval<unsigned>({4}, {5}), unsigned_comparator()) == false);
// [3,4] and (4,5]
BOOST_REQUIRE(wrapping_interval<unsigned>({3}, {4}).overlaps(wrapping_interval<unsigned>({{4, false}}, {5}), unsigned_comparator()) == false);
// [3,4) and (4,5]
BOOST_REQUIRE(wrapping_interval<unsigned>({3}, {{4, false}}).overlaps(wrapping_interval<unsigned>({{4, false}}, {5}), unsigned_comparator()) == false);
}
auto get_item(std::string left, std::string right, std::string val) {
using value_type = std::unordered_set<std::string>;
auto l = dht::token::from_sstring(left);
auto r = dht::token::from_sstring(right);
auto rg = wrapping_interval<dht::token>({{l, false}}, {r});
value_type v{val};
return std::make_pair(locator::token_metadata::range_to_interval(rg), v);
}
BOOST_AUTO_TEST_CASE(test_range_interval_map) {
using value_type = std::unordered_set<std::string>;
using token = dht::token;
boost::icl::interval_map<token, value_type> mymap;
mymap += get_item("1", "5", "A");
mymap += get_item("5", "8", "B");
mymap += get_item("1", "3", "C");
mymap += get_item("3", "8", "D");
std::cout << "my map: " << "\n";
for (auto x : mymap) {
std::cout << x.first << " -> ";
for (auto s : x.second) {
std::cout << s << ", ";
}
std::cout << "\n";
}
auto search_item = [&mymap] (std::string val) {
auto tok = dht::token::from_sstring(val);
auto search = wrapping_interval<token>(tok);
auto it = mymap.find(locator::token_metadata::range_to_interval(search));
if (it != mymap.end()) {
std::cout << "Found OK:" << " token = " << tok << " in range: " << it->first << "\n";
return true;
} else {
std::cout << "Found NO:" << " token = " << tok << "\n";
return false;
}
};
BOOST_REQUIRE(search_item("0") == false);
BOOST_REQUIRE(search_item("1") == false);
BOOST_REQUIRE(search_item("2") == true);
BOOST_REQUIRE(search_item("3") == true);
BOOST_REQUIRE(search_item("4") == true);
BOOST_REQUIRE(search_item("5") == true);
BOOST_REQUIRE(search_item("6") == true);
BOOST_REQUIRE(search_item("7") == true);
BOOST_REQUIRE(search_item("8") == true);
BOOST_REQUIRE(search_item("9") == false);
}
BOOST_AUTO_TEST_CASE(test_split_after) {
using b = interval_bound<unsigned>;
using wr = wrapping_interval<unsigned>;
using nwr = interval<unsigned>;
auto cmp = unsigned_comparator();
auto nwr1 = nwr(b(5), b(8));
BOOST_REQUIRE_EQUAL(nwr1.split_after(2, cmp), nwr1);
BOOST_REQUIRE_EQUAL(nwr1.split_after(5, cmp), nwr(b(5, false), b(8)));
BOOST_REQUIRE_EQUAL(nwr1.split_after(6, cmp), nwr(b(6, false), b(8)));
BOOST_REQUIRE_EQUAL(nwr1.split_after(8, cmp), std::nullopt);
BOOST_REQUIRE_EQUAL(nwr1.split_after(9, cmp), std::nullopt);
auto nwr2 = nwr(b(5, false), b(8, false));
BOOST_REQUIRE_EQUAL(nwr2.split_after(2, cmp), nwr2);
BOOST_REQUIRE_EQUAL(nwr2.split_after(5, cmp), nwr(b(5, false), b(8, false)));
BOOST_REQUIRE_EQUAL(nwr2.split_after(6, cmp), nwr(b(6, false), b(8, false)));
BOOST_REQUIRE_EQUAL(nwr2.split_after(8, cmp), std::nullopt);
BOOST_REQUIRE_EQUAL(nwr2.split_after(9, cmp), std::nullopt);
auto nwr3 = nwr(b(5, false), std::nullopt);
BOOST_REQUIRE_EQUAL(nwr3.split_after(2, cmp), nwr3);
BOOST_REQUIRE_EQUAL(nwr3.split_after(5, cmp), nwr3);
BOOST_REQUIRE_EQUAL(nwr3.split_after(6, cmp), nwr(b(6, false), std::nullopt));
auto nwr4 = nwr(std::nullopt, b(5, false));
BOOST_REQUIRE_EQUAL(nwr4.split_after(2, cmp), nwr(b(2, false), b(5, false)));
BOOST_REQUIRE_EQUAL(nwr4.split_after(5, cmp), std::nullopt);
BOOST_REQUIRE_EQUAL(nwr4.split_after(6, cmp), std::nullopt);
auto nwr5 = nwr(std::nullopt, std::nullopt);
BOOST_REQUIRE_EQUAL(nwr5.split_after(2, cmp), nwr(b(2, false), std::nullopt));
auto wr1 = wr(b(5), b(8));
BOOST_REQUIRE_EQUAL(wr1.split_after(2, cmp), wr1);
BOOST_REQUIRE_EQUAL(wr1.split_after(5, cmp), wr(b(5, false), b(8)));
BOOST_REQUIRE_EQUAL(wr1.split_after(6, cmp), wr(b(6, false), b(8)));
BOOST_REQUIRE_EQUAL(wr1.split_after(8, cmp), std::nullopt);
BOOST_REQUIRE_EQUAL(wr1.split_after(9, cmp), std::nullopt);
auto wr2 = wr(b(5, false), b(8, false));
BOOST_REQUIRE_EQUAL(wr2.split_after(2, cmp), wr2);
BOOST_REQUIRE_EQUAL(wr2.split_after(5, cmp), wr(b(5, false), b(8, false)));
BOOST_REQUIRE_EQUAL(wr2.split_after(6, cmp), wr(b(6, false), b(8, false)));
BOOST_REQUIRE_EQUAL(wr2.split_after(8, cmp), std::nullopt);
BOOST_REQUIRE_EQUAL(wr2.split_after(9, cmp), std::nullopt);
auto wr3 = wr(b(5, false), std::nullopt);
BOOST_REQUIRE_EQUAL(wr3.split_after(2, cmp), wr3);
BOOST_REQUIRE_EQUAL(wr3.split_after(5, cmp), wr3);
BOOST_REQUIRE_EQUAL(wr3.split_after(6, cmp), wr(b(6, false), std::nullopt));
auto wr4 = wr(std::nullopt, b(5, false));
BOOST_REQUIRE_EQUAL(wr4.split_after(2, cmp), wr(b(2, false), b(5, false)));
BOOST_REQUIRE_EQUAL(wr4.split_after(5, cmp), std::nullopt);
BOOST_REQUIRE_EQUAL(wr4.split_after(6, cmp), std::nullopt);
auto wr5 = wr(std::nullopt, std::nullopt);
BOOST_REQUIRE_EQUAL(wr5.split_after(2, cmp), wr(b(2, false), std::nullopt));
auto wr6 = wr(b(8), b(5));
BOOST_REQUIRE_EQUAL(wr6.split_after(2, cmp), wr(b(2, false), b(5)));
BOOST_REQUIRE_EQUAL(wr6.split_after(5, cmp), std::nullopt);
BOOST_REQUIRE_EQUAL(wr6.split_after(6, cmp), std::nullopt);
BOOST_REQUIRE_EQUAL(wr6.split_after(8, cmp), wr(b(8, false), b(5)));
BOOST_REQUIRE_EQUAL(wr6.split_after(9, cmp), wr(b(9, false), b(5)));
}
BOOST_AUTO_TEST_CASE(test_intersection) {
using b = interval_bound<unsigned>;
using nwr = interval<unsigned>;
auto cmp = unsigned_comparator();
auto r1 = nwr(b(5), b(10));
auto r2 = nwr(b(1), b(4));
BOOST_REQUIRE_EQUAL(r1.intersection(r2, cmp), std::nullopt);
auto r3 = nwr(b(1), b(5));
BOOST_REQUIRE_EQUAL(r1.intersection(r3, cmp), nwr(b(5), b(5)));
auto r4 = nwr(b(2), b(7));
BOOST_REQUIRE_EQUAL(r1.intersection(r4, cmp), nwr(b(5), b(7)));
auto r5 = nwr(b(5), b(8));
BOOST_REQUIRE_EQUAL(r1.intersection(r5, cmp), nwr(b(5), b(8)));
auto r6 = nwr(b(6), b(8));
BOOST_REQUIRE_EQUAL(r1.intersection(r6, cmp), nwr(b(6), b(8)));
auto r7 = nwr(b(7), b(10));
BOOST_REQUIRE_EQUAL(r1.intersection(r7, cmp), nwr(b(7), b(10)));
auto r8 = nwr(b(8), b(11));
BOOST_REQUIRE_EQUAL(r1.intersection(r8, cmp), nwr(b(8), b(10)));
auto r9 = nwr(b(10), b(12));
BOOST_REQUIRE_EQUAL(r1.intersection(r9, cmp), nwr(b(10), b(10)));
auto r10 = nwr(b(12), b(20));
BOOST_REQUIRE_EQUAL(r1.intersection(r10, cmp), std::nullopt);
auto r11 = nwr(b(1), b(20));
BOOST_REQUIRE_EQUAL(r1.intersection(r11, cmp), nwr(b(5), b(10)));
auto r12 = nwr(b(1), b(3, false));
BOOST_REQUIRE_EQUAL(r12.intersection(nwr(b(3, false), b(5)), cmp), std::nullopt);
BOOST_REQUIRE_EQUAL(r12.intersection(nwr(b(3, false), b(5)), cmp), std::nullopt);
BOOST_REQUIRE_EQUAL(r12.intersection(nwr(b(2), { }), cmp), nwr(b(2), b(3, false)));
BOOST_REQUIRE_EQUAL(r12.intersection(nwr({ }, b(2)), cmp), nwr(b(1), b(2)));
BOOST_REQUIRE_EQUAL(r12.intersection(nwr::make_singular(4), cmp), std::nullopt);
BOOST_REQUIRE_EQUAL(r12.intersection(nwr::make_singular(3), cmp), std::nullopt);
BOOST_REQUIRE_EQUAL(r12.intersection(nwr::make_singular(2), cmp), nwr(b(2), b(2)));
BOOST_REQUIRE_EQUAL(r12.intersection(nwr::make_singular(1), cmp), nwr(b(1), b(1)));
BOOST_REQUIRE_EQUAL(r12.intersection(nwr::make_singular(0), cmp), std::nullopt);
const auto r13 = nwr::make_singular(123);
BOOST_REQUIRE_EQUAL(r13.intersection(r13, cmp), nwr(b(123), b(123)));
BOOST_REQUIRE_EQUAL(r13.intersection(nwr::make_singular(321), cmp), std::nullopt);
BOOST_REQUIRE_EQUAL(r13.intersection(nwr(b(123), b(123)), cmp), nwr(b(123), b(123)));
BOOST_REQUIRE_EQUAL(r13.intersection(nwr(b(123), b(124, false)), cmp), nwr(b(123), b(123)));
BOOST_REQUIRE_EQUAL(r13.intersection(nwr(b(123, false), b(124, false)), cmp), std::nullopt);
BOOST_REQUIRE_EQUAL(r13.intersection(nwr(b(0), b(123, false)), cmp), std::nullopt);
BOOST_REQUIRE_EQUAL(r13.intersection(nwr(b(0), b(1000)), cmp), nwr(b(123), b(123)));
}
BOOST_AUTO_TEST_CASE(test_before_point) {
using r = interval<unsigned>;
using b = r::bound;
auto cmp = unsigned_comparator();
BOOST_REQUIRE_EQUAL(r::make_open_ended_both_sides().before(8, cmp), false);
BOOST_REQUIRE_EQUAL(r::make_ending_with(10).before(8, cmp), false);
BOOST_REQUIRE_EQUAL(r::make_ending_with(10).before(100, cmp), false);
BOOST_REQUIRE_EQUAL(r::make_starting_with(10).before(8, cmp), true);
BOOST_REQUIRE_EQUAL(r::make_starting_with(10).before(10, cmp), false);
BOOST_REQUIRE_EQUAL(r::make_starting_with({10, true}).before(10, cmp), false);
BOOST_REQUIRE_EQUAL(r::make_starting_with({10, false}).before(10, cmp), true);
BOOST_REQUIRE_EQUAL(r::make_starting_with(10).before(100, cmp), false);
BOOST_REQUIRE_EQUAL(r::make_singular(10).before(8, cmp), true);
BOOST_REQUIRE_EQUAL(r::make_singular(10).before(10, cmp), false);
BOOST_REQUIRE_EQUAL(r::make_singular(10).before(100, cmp), false);
BOOST_REQUIRE_EQUAL(r::make(b(10, true), b(11, true)).before(10, cmp), false);
BOOST_REQUIRE_EQUAL(r::make(b(10, false), b(11, true)).before(10, cmp), true);
BOOST_REQUIRE_EQUAL(r::make(b(10, true), b(11, true)).before(11, cmp), false);
BOOST_REQUIRE_EQUAL(r::make(b(10, true), b(11, true)).before(100, cmp), false);
}
BOOST_AUTO_TEST_CASE(test_before_interval) {
using r = interval<unsigned>;
using b = r::bound;
auto cmp = unsigned_comparator();
auto check = [&] (const r& r1, const r& r2, bool is_before, std::source_location sl = std::source_location::current()) {
BOOST_TEST_MESSAGE(fmt::format("check() @ {}:{} {} before {} -> {}", sl.file_name(), sl.line(), r2, r1, is_before));
BOOST_REQUIRE_EQUAL(r1.other_is_before(r2, cmp), is_before);
};
check(r::make_open_ended_both_sides(), r::make_singular(10), false);
check(r::make_open_ended_both_sides(), r::make_starting_with(10), false);
check(r::make_open_ended_both_sides(), r::make_ending_with(10), false);
check(r::make_open_ended_both_sides(), r::make(b(10, false), b(11, true)), false);
check(r::make_open_ended_both_sides(), r::make(b(10, false), b(11, false)), false);
// Check with intervals that has inclusive start bound with value 10
for (const auto& i : {r::make_starting_with(10), r::make_starting_with({10, true}), r::make_singular(10), r::make(b{10, true}, b{11, true})}) {
check(i, r::make_starting_with(1), false);
check(i, r::make_starting_with(10), false);
check(i, r::make_starting_with(100), false);
check(i, r::make(b(10, false), b(11, true)), false);
check(i, r::make(b(10, false), b(11, false)), false);
check(i, r::make_singular(8), true);
check(i, r::make_singular(10), false);
check(i, r::make_singular(10), false);
check(i, r::make_singular(100), false);
check(i, r::make_ending_with(8), true);
check(i, r::make_ending_with(10), false);
check(i, r::make_ending_with({10, true}), false);
check(i, r::make_ending_with({10, false}), true);
check(i, r::make_ending_with(100), false);
check(i, r::make(b{1, false}, b{10, true}), false);
check(i, r::make(b{1, false}, b{10, true}), false);
check(i, r::make(b{1, false}, b{10, false}), true);
}
// Check with intervals that has exclusive start bound with value 10
for (const auto& i : {r::make_starting_with({10, false}), r::make(b{10, false}, b{11, true})}) {
check(i, r::make_singular(10), true);
check(i, r::make_ending_with({10, true}), true);
check(i, r::make_ending_with({10, false}), true);
check(i, r::make(b{1, false}, b{10, true}), true);
check(i, r::make(b{1, false}, b{10, false}), true);
}
}
namespace {
struct exploded {};
// Interval payload which will explode if used incorrectly (in debug mode),
// and detects use of moved-from objects. It simulates exceptions thrown where
// A regular interval payload would throw, and this can be controlled to
// exhaustively test the logical timing of the exception.
class maybe_throwing_interval_payload {
public:
static inline thread_local unsigned s_throw_countdown_counter = 0;
static inline thread_local unsigned s_live_objects = 0;
private:
// The pointer is always engaged, even from moved-from objects, to
// detect move-from-uninitialized.
std::unique_ptr<unsigned> _payload = std::make_unique<unsigned>(-1);
public:
maybe_throwing_interval_payload(unsigned value) {
*_payload = value;
++s_live_objects;
}
maybe_throwing_interval_payload(const maybe_throwing_interval_payload& o) {
*_payload = *o._payload;
if (s_throw_countdown_counter-- == 0) {
throw exploded();
}
++s_live_objects;
}
// We ignore bad_alloc, won't happen for the test
maybe_throwing_interval_payload(maybe_throwing_interval_payload&& o) noexcept {
// Overwrite `o` to simulate destructive move
*_payload = std::exchange(*o._payload, -1u);
++s_live_objects;
}
~maybe_throwing_interval_payload() {
--s_live_objects;
}
maybe_throwing_interval_payload& operator=(const maybe_throwing_interval_payload& o) {
if (this != &o) {
*_payload = *o._payload;
if (s_throw_countdown_counter-- == 0) {
throw exploded();
}
}
return *this;
}
maybe_throwing_interval_payload& operator=(maybe_throwing_interval_payload&&) = default;
unsigned value() const {
if (*_payload == -1u) {
throw std::logic_error("maybe_throwing_interval_payload: accessing moved-from object");
}
return *_payload;
}
std::strong_ordering operator<=>(const maybe_throwing_interval_payload& other) const {
return *_payload <=> * other._payload;
}
};
using exploding_interval = wrapping_interval<maybe_throwing_interval_payload>;
}
static
void
test_with_exploding_payload(std::function<void (exploding_interval)> test_func) {
for (unsigned ctr = 0; ctr < 20; ++ctr) {
for (unsigned bounds_selection = 0; bounds_selection <= 4; ++bounds_selection) {
bool has_left = bounds_selection & 1;
bool has_right = bounds_selection & 2;
bool singular = bounds_selection == 4;
maybe_throwing_interval_payload::s_throw_countdown_counter = ctr;
maybe_throwing_interval_payload::s_live_objects = 0;
try {
auto make_bound = [] (bool has, unsigned val) -> std::optional<exploding_interval::bound> {
if (has) {
return exploding_interval::bound(val);
} else {
return std::nullopt;
}
};
auto input = singular
? exploding_interval(make_bound(true, 5), std::nullopt, true)
: exploding_interval(make_bound(has_left, 4), make_bound(has_right, 9));
test_func(std::move(input));
} catch (exploded&) {
// expected
}
BOOST_REQUIRE_EQUAL(maybe_throwing_interval_payload::s_live_objects, 0);
}
}
}
BOOST_AUTO_TEST_CASE(test_copy_ctor_exploding) {
test_with_exploding_payload([] (exploding_interval i1) {
auto i2 = exploding_interval(i1);
});
}
BOOST_AUTO_TEST_CASE(test_copy_assign_exploding) {
test_with_exploding_payload([] (exploding_interval i1) {
auto i2 = exploding_interval(i1);
i1 = i2;
});
}
BOOST_AUTO_TEST_CASE(test_reverse_exploding) {
test_with_exploding_payload([] (exploding_interval i1) {
i1.reverse();
});
}
BOOST_AUTO_TEST_CASE(test_transorm_rvalue_exploding) {
test_with_exploding_payload([] (exploding_interval i1) {
auto copy = i1;
auto xformed = std::move(i1).transform([] (maybe_throwing_interval_payload p) {
return maybe_throwing_interval_payload(p.value() + 1);
});
BOOST_REQUIRE_EQUAL(bool(copy.start()), bool(xformed.start()));
if (copy.start()) {
BOOST_REQUIRE_EQUAL(copy.start().value().value().value() + 1, xformed.start().value().value().value());
}
BOOST_REQUIRE_EQUAL(bool(copy.end()), bool(xformed.end()));
if (copy.end()) {
BOOST_REQUIRE_EQUAL(copy.end().value().value().value() + 1, xformed.end().value().value().value());
}
});
}