| // Copyright 2012 The Chromium Authors |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #include "base/containers/enum_set.h" |
| |
| #include <stddef.h> |
| |
| #include <optional> |
| |
| #include "base/containers/to_vector.h" |
| #include "base/test/gtest_util.h" |
| #include "testing/gmock/include/gmock/gmock.h" |
| #include "testing/gtest/include/gtest/gtest-death-test.h" |
| #include "testing/gtest/include/gtest/gtest.h" |
| |
| namespace base { |
| namespace { |
| |
| enum class TestEnum { |
| kTestBelowMinNegative = -1, |
| kTestBelowMin = 0, |
| kTest1 = 1, |
| kTestMin = kTest1, |
| kTest2, |
| kTest3, |
| kTest4, |
| kTest5, |
| kTestMax = kTest5, |
| kTest6OutOfBounds, |
| kTest7OutOfBounds |
| }; |
| using TestEnumSet = EnumSet<TestEnum, TestEnum::kTestMin, TestEnum::kTestMax>; |
| |
| enum class TestEnumExtreme { |
| kTest0 = 0, |
| kTestMin = kTest0, |
| kTest63 = 63, |
| kTestMax = kTest63, |
| kTest64OutOfBounds, |
| }; |
| using TestEnumExtremeSet = EnumSet<TestEnumExtreme, |
| TestEnumExtreme::kTestMin, |
| TestEnumExtreme::kTestMax>; |
| |
| class EnumSetTest : public ::testing::Test {}; |
| class EnumSetDeathTest : public ::testing::Test {}; |
| |
| TEST_F(EnumSetTest, ClassConstants) { |
| EXPECT_EQ(TestEnum::kTestMin, TestEnumSet::kMinValue); |
| EXPECT_EQ(TestEnum::kTestMax, TestEnumSet::kMaxValue); |
| EXPECT_EQ(5u, TestEnumSet::kValueCount); |
| } |
| |
| // Use static_assert to check that functions we expect to be compile time |
| // evaluatable are really that way. |
| TEST_F(EnumSetTest, ConstexprsAreValid) { |
| static_assert(TestEnumSet::All().Has(TestEnum::kTest2), |
| "Expected All() to be integral constant expression"); |
| static_assert(TestEnumSet::FromRange(TestEnum::kTest2, TestEnum::kTest4) |
| .Has(TestEnum::kTest2), |
| "Expected FromRange() to be integral constant expression"); |
| static_assert(TestEnumSet{TestEnum::kTest2}.Has(TestEnum::kTest2), |
| "Expected TestEnumSet() to be integral constant expression"); |
| static_assert( |
| TestEnumSet::FromEnumBitmask(1 << static_cast<uint64_t>(TestEnum::kTest2)) |
| .Has(TestEnum::kTest2), |
| "Expected TestEnumSet() to be integral constant expression"); |
| } |
| |
| TEST_F(EnumSetTest, DefaultConstructor) { |
| const TestEnumSet enums; |
| EXPECT_TRUE(enums.empty()); |
| EXPECT_EQ(0u, enums.size()); |
| EXPECT_FALSE(enums.Has(TestEnum::kTest1)); |
| EXPECT_FALSE(enums.Has(TestEnum::kTest2)); |
| EXPECT_FALSE(enums.Has(TestEnum::kTest3)); |
| EXPECT_FALSE(enums.Has(TestEnum::kTest4)); |
| EXPECT_FALSE(enums.Has(TestEnum::kTest5)); |
| } |
| |
| TEST_F(EnumSetTest, OneArgConstructor) { |
| const TestEnumSet enums = {TestEnum::kTest4}; |
| EXPECT_FALSE(enums.empty()); |
| EXPECT_EQ(1u, enums.size()); |
| EXPECT_FALSE(enums.Has(TestEnum::kTest1)); |
| EXPECT_FALSE(enums.Has(TestEnum::kTest2)); |
| EXPECT_FALSE(enums.Has(TestEnum::kTest3)); |
| EXPECT_TRUE(enums.Has(TestEnum::kTest4)); |
| EXPECT_FALSE(enums.Has(TestEnum::kTest5)); |
| } |
| |
| TEST_F(EnumSetTest, OneArgConstructorSize) { |
| TestEnumExtremeSet enums = {TestEnumExtreme::kTest0}; |
| EXPECT_TRUE(enums.Has(TestEnumExtreme::kTest0)); |
| } |
| |
| TEST_F(EnumSetTest, TwoArgConstructor) { |
| const TestEnumSet enums = {TestEnum::kTest4, TestEnum::kTest2}; |
| EXPECT_FALSE(enums.empty()); |
| EXPECT_EQ(2u, enums.size()); |
| EXPECT_FALSE(enums.Has(TestEnum::kTest1)); |
| EXPECT_TRUE(enums.Has(TestEnum::kTest2)); |
| EXPECT_FALSE(enums.Has(TestEnum::kTest3)); |
| EXPECT_TRUE(enums.Has(TestEnum::kTest4)); |
| EXPECT_FALSE(enums.Has(TestEnum::kTest5)); |
| } |
| |
| TEST_F(EnumSetTest, ThreeArgConstructor) { |
| const TestEnumSet enums = {TestEnum::kTest4, TestEnum::kTest2, |
| TestEnum::kTest1}; |
| EXPECT_FALSE(enums.empty()); |
| EXPECT_EQ(3u, enums.size()); |
| EXPECT_TRUE(enums.Has(TestEnum::kTest1)); |
| EXPECT_TRUE(enums.Has(TestEnum::kTest2)); |
| EXPECT_FALSE(enums.Has(TestEnum::kTest3)); |
| EXPECT_TRUE(enums.Has(TestEnum::kTest4)); |
| EXPECT_FALSE(enums.Has(TestEnum::kTest5)); |
| } |
| |
| TEST_F(EnumSetTest, DuplicatesInConstructor) { |
| EXPECT_EQ( |
| TestEnumSet({TestEnum::kTest4, TestEnum::kTest2, TestEnum::kTest1, |
| TestEnum::kTest4, TestEnum::kTest2, TestEnum::kTest4}), |
| TestEnumSet({TestEnum::kTest1, TestEnum::kTest2, TestEnum::kTest4})); |
| } |
| |
| TEST_F(EnumSetTest, All) { |
| const TestEnumSet enums(TestEnumSet::All()); |
| EXPECT_FALSE(enums.empty()); |
| EXPECT_EQ(5u, enums.size()); |
| EXPECT_TRUE(enums.Has(TestEnum::kTest1)); |
| EXPECT_TRUE(enums.Has(TestEnum::kTest2)); |
| EXPECT_TRUE(enums.Has(TestEnum::kTest3)); |
| EXPECT_TRUE(enums.Has(TestEnum::kTest4)); |
| EXPECT_TRUE(enums.Has(TestEnum::kTest5)); |
| } |
| |
| TEST_F(EnumSetTest, AllExtreme) { |
| const TestEnumExtremeSet enums(TestEnumExtremeSet::All()); |
| EXPECT_FALSE(enums.empty()); |
| EXPECT_EQ(64u, enums.size()); |
| EXPECT_TRUE(enums.Has(TestEnumExtreme::kTest0)); |
| EXPECT_TRUE(enums.Has(TestEnumExtreme::kTest63)); |
| EXPECT_FALSE(enums.Has(TestEnumExtreme::kTest64OutOfBounds)); |
| } |
| |
| TEST_F(EnumSetTest, FromRange) { |
| EXPECT_EQ(TestEnumSet({TestEnum::kTest2, TestEnum::kTest3, TestEnum::kTest4}), |
| TestEnumSet::FromRange(TestEnum::kTest2, TestEnum::kTest4)); |
| EXPECT_EQ(TestEnumSet::All(), |
| TestEnumSet::FromRange(TestEnum::kTest1, TestEnum::kTest5)); |
| EXPECT_EQ(TestEnumSet({TestEnum::kTest2}), |
| TestEnumSet::FromRange(TestEnum::kTest2, TestEnum::kTest2)); |
| |
| using RestrictedRangeSet = |
| EnumSet<TestEnum, TestEnum::kTest2, TestEnum::kTestMax>; |
| EXPECT_EQ(RestrictedRangeSet( |
| {TestEnum::kTest2, TestEnum::kTest3, TestEnum::kTest4}), |
| RestrictedRangeSet::FromRange(TestEnum::kTest2, TestEnum::kTest4)); |
| EXPECT_EQ(RestrictedRangeSet::All(), |
| RestrictedRangeSet::FromRange(TestEnum::kTest2, TestEnum::kTest5)); |
| } |
| |
| TEST_F(EnumSetTest, Put) { |
| TestEnumSet enums = {TestEnum::kTest4}; |
| enums.Put(TestEnum::kTest3); |
| EXPECT_EQ(TestEnumSet({TestEnum::kTest3, TestEnum::kTest4}), enums); |
| enums.Put(TestEnum::kTest5); |
| EXPECT_EQ(TestEnumSet({TestEnum::kTest3, TestEnum::kTest4, TestEnum::kTest5}), |
| enums); |
| } |
| |
| TEST_F(EnumSetTest, PutAll) { |
| TestEnumSet enums = {TestEnum::kTest4, TestEnum::kTest5}; |
| enums.PutAll({TestEnum::kTest3, TestEnum::kTest4}); |
| EXPECT_EQ(TestEnumSet({TestEnum::kTest3, TestEnum::kTest4, TestEnum::kTest5}), |
| enums); |
| } |
| |
| TEST_F(EnumSetTest, PutRange) { |
| TestEnumSet enums; |
| enums.PutRange(TestEnum::kTest2, TestEnum::kTest4); |
| EXPECT_EQ(TestEnumSet({TestEnum::kTest2, TestEnum::kTest3, TestEnum::kTest4}), |
| enums); |
| } |
| |
| TEST_F(EnumSetTest, RetainAll) { |
| TestEnumSet enums = {TestEnum::kTest4, TestEnum::kTest5}; |
| enums.RetainAll(TestEnumSet({TestEnum::kTest3, TestEnum::kTest4})); |
| EXPECT_EQ(TestEnumSet({TestEnum::kTest4}), enums); |
| } |
| |
| TEST_F(EnumSetTest, Remove) { |
| TestEnumSet enums = {TestEnum::kTest4, TestEnum::kTest5}; |
| enums.Remove(TestEnum::kTest1); |
| enums.Remove(TestEnum::kTest3); |
| EXPECT_EQ(TestEnumSet({TestEnum::kTest4, TestEnum::kTest5}), enums); |
| enums.Remove(TestEnum::kTest4); |
| EXPECT_EQ(TestEnumSet({TestEnum::kTest5}), enums); |
| enums.Remove(TestEnum::kTest5); |
| enums.Remove(TestEnum::kTest6OutOfBounds); |
| EXPECT_TRUE(enums.empty()); |
| } |
| |
| TEST_F(EnumSetTest, RemoveAll) { |
| TestEnumSet enums = {TestEnum::kTest4, TestEnum::kTest5}; |
| enums.RemoveAll(TestEnumSet({TestEnum::kTest3, TestEnum::kTest4})); |
| EXPECT_EQ(TestEnumSet({TestEnum::kTest5}), enums); |
| } |
| |
| TEST_F(EnumSetTest, Clear) { |
| TestEnumSet enums = {TestEnum::kTest4, TestEnum::kTest5}; |
| enums.Clear(); |
| EXPECT_TRUE(enums.empty()); |
| } |
| |
| TEST_F(EnumSetTest, Set) { |
| TestEnumSet enums; |
| EXPECT_TRUE(enums.empty()); |
| |
| enums.PutOrRemove(TestEnum::kTest3, false); |
| EXPECT_TRUE(enums.empty()); |
| |
| enums.PutOrRemove(TestEnum::kTest4, true); |
| EXPECT_EQ(enums, TestEnumSet({TestEnum::kTest4})); |
| |
| enums.PutOrRemove(TestEnum::kTest5, true); |
| EXPECT_EQ(enums, TestEnumSet({TestEnum::kTest4, TestEnum::kTest5})); |
| enums.PutOrRemove(TestEnum::kTest5, true); |
| EXPECT_EQ(enums, TestEnumSet({TestEnum::kTest4, TestEnum::kTest5})); |
| |
| enums.PutOrRemove(TestEnum::kTest4, false); |
| EXPECT_EQ(enums, TestEnumSet({TestEnum::kTest5})); |
| } |
| |
| TEST_F(EnumSetTest, Has) { |
| const TestEnumSet enums = {TestEnum::kTest4, TestEnum::kTest5}; |
| EXPECT_FALSE(enums.Has(TestEnum::kTest1)); |
| EXPECT_FALSE(enums.Has(TestEnum::kTest2)); |
| EXPECT_FALSE(enums.Has(TestEnum::kTest3)); |
| EXPECT_TRUE(enums.Has(TestEnum::kTest4)); |
| EXPECT_TRUE(enums.Has(TestEnum::kTest5)); |
| EXPECT_FALSE(enums.Has(TestEnum::kTest6OutOfBounds)); |
| } |
| |
| TEST_F(EnumSetTest, HasAll) { |
| const TestEnumSet enums1 = {TestEnum::kTest4, TestEnum::kTest5}; |
| const TestEnumSet enums2 = {TestEnum::kTest3, TestEnum::kTest4}; |
| const TestEnumSet enums3 = Union(enums1, enums2); |
| EXPECT_TRUE(enums1.HasAll(enums1)); |
| EXPECT_FALSE(enums1.HasAll(enums2)); |
| EXPECT_FALSE(enums1.HasAll(enums3)); |
| |
| EXPECT_FALSE(enums2.HasAll(enums1)); |
| EXPECT_TRUE(enums2.HasAll(enums2)); |
| EXPECT_FALSE(enums2.HasAll(enums3)); |
| |
| EXPECT_TRUE(enums3.HasAll(enums1)); |
| EXPECT_TRUE(enums3.HasAll(enums2)); |
| EXPECT_TRUE(enums3.HasAll(enums3)); |
| } |
| |
| TEST_F(EnumSetTest, HasAny) { |
| const TestEnumSet enums1 = {TestEnum::kTest4, TestEnum::kTest5}; |
| const TestEnumSet enums2 = {TestEnum::kTest3, TestEnum::kTest4}; |
| const TestEnumSet enums3 = {TestEnum::kTest1, TestEnum::kTest2}; |
| EXPECT_TRUE(enums1.HasAny(enums1)); |
| EXPECT_TRUE(enums1.HasAny(enums2)); |
| EXPECT_FALSE(enums1.HasAny(enums3)); |
| |
| EXPECT_TRUE(enums2.HasAny(enums1)); |
| EXPECT_TRUE(enums2.HasAny(enums2)); |
| EXPECT_FALSE(enums2.HasAny(enums3)); |
| |
| EXPECT_FALSE(enums3.HasAny(enums1)); |
| EXPECT_FALSE(enums3.HasAny(enums2)); |
| EXPECT_TRUE(enums3.HasAny(enums3)); |
| } |
| |
| TEST_F(EnumSetTest, Iterators) { |
| const TestEnumSet enums1 = {TestEnum::kTest4, TestEnum::kTest5}; |
| TestEnumSet enums2; |
| for (TestEnum e : enums1) { |
| enums2.Put(e); |
| } |
| EXPECT_EQ(enums2, enums1); |
| } |
| |
| TEST_F(EnumSetTest, RangeBasedForLoop) { |
| const TestEnumSet enums1 = {TestEnum::kTest2, TestEnum::kTest5}; |
| TestEnumSet enums2; |
| for (TestEnum e : enums1) { |
| enums2.Put(e); |
| } |
| EXPECT_EQ(enums2, enums1); |
| } |
| |
| TEST_F(EnumSetTest, IteratorComparisonOperators) { |
| const TestEnumSet enums = {TestEnum::kTest2, TestEnum::kTest4}; |
| const auto first_it = enums.begin(); |
| const auto second_it = ++enums.begin(); |
| |
| // Copy for equality testing. |
| const auto first_it_copy = first_it; |
| |
| // Sanity check, as the rest of the test relies on |first_it| and |
| // |first_it_copy| pointing to the same element and |first_it| and |second_it| |
| // pointing to different elements. |
| ASSERT_EQ(*first_it, *first_it_copy); |
| ASSERT_NE(*first_it, *second_it); |
| |
| EXPECT_TRUE(first_it == first_it_copy); |
| EXPECT_FALSE(first_it != first_it_copy); |
| |
| EXPECT_TRUE(first_it != second_it); |
| EXPECT_FALSE(first_it == second_it); |
| } |
| |
| TEST_F(EnumSetTest, IteratorIncrementOperators) { |
| const TestEnumSet enums = {TestEnum::kTest2, TestEnum::kTest4}; |
| const auto begin = enums.begin(); |
| |
| auto post_inc_it = begin; |
| auto pre_inc_it = begin; |
| |
| auto post_inc_return_it = post_inc_it++; |
| auto pre_inc_return_it = ++pre_inc_it; |
| |
| // |pre_inc_it| and |post_inc_it| should point to the same element. |
| EXPECT_EQ(pre_inc_it, post_inc_it); |
| EXPECT_EQ(*pre_inc_it, *post_inc_it); |
| |
| // |pre_inc_it| should NOT point to the first element. |
| EXPECT_NE(begin, pre_inc_it); |
| EXPECT_NE(*begin, *pre_inc_it); |
| |
| // |post_inc_it| should NOT point to the first element. |
| EXPECT_NE(begin, post_inc_it); |
| EXPECT_NE(*begin, *post_inc_it); |
| |
| // Prefix increment should return new iterator. |
| EXPECT_EQ(pre_inc_return_it, post_inc_it); |
| EXPECT_EQ(*pre_inc_return_it, *post_inc_it); |
| |
| // Postfix increment should return original iterator. |
| EXPECT_EQ(post_inc_return_it, begin); |
| EXPECT_EQ(*post_inc_return_it, *begin); |
| } |
| |
| TEST_F(EnumSetTest, Union) { |
| const TestEnumSet enums1 = {TestEnum::kTest4, TestEnum::kTest5}; |
| const TestEnumSet enums2 = {TestEnum::kTest3, TestEnum::kTest4}; |
| const TestEnumSet enums3 = Union(enums1, enums2); |
| |
| EXPECT_EQ(TestEnumSet({TestEnum::kTest3, TestEnum::kTest4, TestEnum::kTest5}), |
| enums3); |
| } |
| |
| TEST_F(EnumSetTest, Intersection) { |
| const TestEnumSet enums1 = {TestEnum::kTest4, TestEnum::kTest5}; |
| const TestEnumSet enums2 = {TestEnum::kTest3, TestEnum::kTest4}; |
| const TestEnumSet enums3 = Intersection(enums1, enums2); |
| |
| EXPECT_EQ(TestEnumSet({TestEnum::kTest4}), enums3); |
| } |
| |
| TEST_F(EnumSetTest, Difference) { |
| const TestEnumSet enums1 = {TestEnum::kTest4, TestEnum::kTest5}; |
| const TestEnumSet enums2 = {TestEnum::kTest3, TestEnum::kTest4}; |
| const TestEnumSet enums3 = Difference(enums1, enums2); |
| |
| EXPECT_EQ(TestEnumSet({TestEnum::kTest5}), enums3); |
| } |
| |
| TEST_F(EnumSetTest, ToFromEnumBitmask) { |
| const TestEnumSet empty; |
| EXPECT_EQ(empty.ToEnumBitmask(), 0ULL); |
| EXPECT_EQ(TestEnumSet::FromEnumBitmask(0), empty); |
| |
| const TestEnumSet enums1 = {TestEnum::kTest2}; |
| const uint64_t val1 = 1ULL << static_cast<uint64_t>(TestEnum::kTest2); |
| EXPECT_EQ(enums1.ToEnumBitmask(), val1); |
| EXPECT_EQ(TestEnumSet::FromEnumBitmask(val1), enums1); |
| |
| const TestEnumSet enums2 = {TestEnum::kTest3, TestEnum::kTest4}; |
| const uint64_t val2 = 1ULL << static_cast<uint64_t>(TestEnum::kTest3) | |
| 1ULL << static_cast<uint64_t>(TestEnum::kTest4); |
| EXPECT_EQ(enums2.ToEnumBitmask(), val2); |
| EXPECT_EQ(TestEnumSet::FromEnumBitmask(val2), enums2); |
| } |
| |
| TEST_F(EnumSetTest, ToFromEnumBitmaskExtreme) { |
| const TestEnumExtremeSet empty; |
| EXPECT_EQ(empty.ToEnumBitmask(), 0ULL); |
| EXPECT_EQ(TestEnumExtremeSet::FromEnumBitmask(0ULL), empty); |
| |
| const TestEnumExtremeSet enums1 = {TestEnumExtreme::kTest63}; |
| const uint64_t val1 = 1ULL << static_cast<uint64_t>(TestEnumExtreme::kTest63); |
| EXPECT_EQ(enums1.ToEnumBitmask(), val1); |
| EXPECT_EQ(TestEnumExtremeSet::FromEnumBitmask(val1), enums1); |
| } |
| |
| TEST_F(EnumSetTest, FromEnumBitmaskIgnoresExtraBits) { |
| const TestEnumSet kSets[] = { |
| {}, |
| {TestEnum::kTestMin}, |
| {TestEnum::kTestMax}, |
| {TestEnum::kTestMin, TestEnum::kTestMax}, |
| {TestEnum::kTestMin, TestEnum::kTestMax}, |
| {TestEnum::kTest2, TestEnum::kTest4}, |
| }; |
| size_t i = 0; |
| for (const TestEnumSet& set : kSets) { |
| SCOPED_TRACE(i++); |
| const uint64_t val = set.ToEnumBitmask(); |
| |
| // Produce a bitstring for a single enum value. When `e` is in range |
| // relative to TestEnumSet, this function behaves identically to |
| // `single_val_bitstring`. When `e` is not in range, this function attempts |
| // to compute a value, while `single_val_bitstring` intentionally crashes. |
| auto single_val_bitstring = [](TestEnum e) -> uint64_t { |
| uint64_t shift_amount = static_cast<uint64_t>(e); |
| // Shifting left more than the number of bits in the lhs would be UB. |
| CHECK_LT(shift_amount, sizeof(uint64_t) * 8); |
| return 1ULL << shift_amount; |
| }; |
| |
| const uint64_t kJunkVals[] = { |
| // Add junk bits above kTestMax. |
| val | single_val_bitstring(TestEnum::kTest6OutOfBounds), |
| val | single_val_bitstring(TestEnum::kTest7OutOfBounds), |
| val | single_val_bitstring(TestEnum::kTest6OutOfBounds) | |
| single_val_bitstring(TestEnum::kTest7OutOfBounds), |
| // Add junk bits below kTestMin. |
| val | single_val_bitstring(TestEnum::kTestBelowMin), |
| }; |
| for (uint64_t junk_val : kJunkVals) { |
| SCOPED_TRACE(junk_val); |
| ASSERT_NE(val, junk_val); |
| |
| const TestEnumSet set_from_junk = TestEnumSet::FromEnumBitmask(junk_val); |
| EXPECT_EQ(set_from_junk, set); |
| EXPECT_EQ(set_from_junk.ToEnumBitmask(), set.ToEnumBitmask()); |
| |
| // Iterating both sets should produce the same sequence. |
| auto it1 = set.begin(); |
| auto it2 = set_from_junk.begin(); |
| while (it1 != set.end() && it2 != set_from_junk.end()) { |
| EXPECT_EQ(*it1, *it2); |
| ++it1; |
| ++it2; |
| } |
| EXPECT_TRUE(it1 == set.end()); |
| EXPECT_TRUE(it2 == set_from_junk.end()); |
| } |
| } |
| } |
| |
| TEST_F(EnumSetTest, OneEnumValue) { |
| enum class TestEnumOne { |
| kTest1 = 1, |
| kTestMin = kTest1, |
| kTestMax = kTest1, |
| }; |
| using TestEnumOneSet = |
| EnumSet<TestEnumOne, TestEnumOne::kTestMin, TestEnumOne::kTestMax>; |
| EXPECT_EQ(TestEnumOne::kTestMin, TestEnumOneSet::kMinValue); |
| EXPECT_EQ(TestEnumOne::kTestMax, TestEnumOneSet::kMaxValue); |
| EXPECT_EQ(1u, TestEnumOneSet::kValueCount); |
| } |
| |
| TEST_F(EnumSetTest, SparseEnum) { |
| enum class TestEnumSparse { |
| kTest1 = 1, |
| kTestMin = 1, |
| kTest50 = 50, |
| kTest100 = 100, |
| kTestMax = kTest100, |
| }; |
| using TestEnumSparseSet = EnumSet<TestEnumSparse, TestEnumSparse::kTestMin, |
| TestEnumSparse::kTestMax>; |
| TestEnumSparseSet sparse; |
| sparse.Put(TestEnumSparse::kTestMin); |
| sparse.Put(TestEnumSparse::kTestMax); |
| EXPECT_EQ(sparse.size(), 2u); |
| |
| EXPECT_EQ(TestEnumSparseSet::All().size(), 100u); |
| } |
| |
| TEST_F(EnumSetTest, GetNth64bitWordBitmaskFromEnum) { |
| enum class TestEnumEdgeCase { |
| kTest1 = 1, |
| kTestMin = kTest1, |
| kTest63 = 63, |
| kTest64 = 64, |
| kTest100 = 100, |
| kTestMax = kTest100, |
| }; |
| using TestEnumEdgeCaseSet = |
| EnumSet<TestEnumEdgeCase, TestEnumEdgeCase::kTestMin, |
| TestEnumEdgeCase::kTestMax>; |
| TestEnumEdgeCaseSet sparse; |
| sparse.Put(TestEnumEdgeCase::kTest1); |
| sparse.Put(TestEnumEdgeCase::kTest63); |
| sparse.Put(TestEnumEdgeCase::kTest64); |
| sparse.Put(TestEnumEdgeCase::kTest100); |
| std::optional<uint64_t> bit_mask_0 = sparse.GetNth64bitWordBitmask(0); |
| ASSERT_TRUE(bit_mask_0.has_value()); |
| ASSERT_EQ(bit_mask_0.value(), |
| 1ull << static_cast<uint32_t>(TestEnumEdgeCase::kTest1) | |
| 1ull << static_cast<uint32_t>(TestEnumEdgeCase::kTest63)); |
| std::optional<uint64_t> bit_mask_1 = sparse.GetNth64bitWordBitmask(1); |
| ASSERT_TRUE(bit_mask_1.has_value()); |
| ASSERT_EQ( |
| bit_mask_1.value(), |
| 1ull << (static_cast<uint32_t>(TestEnumEdgeCase::kTest64) - 64u) | |
| 1ull << (static_cast<uint32_t>(TestEnumEdgeCase::kTest100) - 64u)); |
| std::optional<uint64_t> bit_mask_2 = sparse.GetNth64bitWordBitmask(2); |
| ASSERT_FALSE(bit_mask_2.has_value()); |
| } |
| |
| TEST_F(EnumSetTest, SparseEnumSmall) { |
| enum class TestEnumSparse { |
| kTest1 = 1, |
| kTestMin = 1, |
| kTest50 = 50, |
| kTest60 = 60, |
| kTestMax = kTest60, |
| }; |
| using TestEnumSparseSet = EnumSet<TestEnumSparse, TestEnumSparse::kTestMin, |
| TestEnumSparse::kTestMax>; |
| TestEnumSparseSet sparse; |
| sparse.Put(TestEnumSparse::kTestMin); |
| sparse.Put(TestEnumSparse::kTestMax); |
| EXPECT_EQ(sparse.size(), 2u); |
| |
| // This may seem a little surprising! There are only 3 distinct values in |
| // TestEnumSparse, so why does TestEnumSparseSet think it has 60 of them? This |
| // is an artifact of EnumSet's design, as it has no way of knowing which |
| // values between the min and max are actually named in the enum's definition. |
| EXPECT_EQ(TestEnumSparseSet::All().size(), 60u); |
| } |
| |
| TEST_F(EnumSetDeathTest, CrashesOnOutOfRange) { |
| EXPECT_CHECK_DEATH(TestEnumSet({TestEnum::kTestBelowMin})); |
| EXPECT_CHECK_DEATH(TestEnumSet({TestEnum::kTest6OutOfBounds})); |
| EXPECT_CHECK_DEATH(TestEnumSet({TestEnum::kTest7OutOfBounds})); |
| } |
| |
| TEST_F(EnumSetDeathTest, EnumWithNegatives) { |
| enum class TestEnumNeg { |
| kTestBelowMin = -3, |
| kTestA = -2, |
| kTestMin = kTestA, |
| kTestB = -1, |
| kTestC = 0, |
| kTestD = 1, |
| kTestE = 2, |
| kTestMax = kTestE, |
| kTestF = 3, |
| }; |
| // This EnumSet starts negative and ends positive. |
| using TestEnumWithNegSet = |
| EnumSet<TestEnumNeg, TestEnumNeg::kTestMin, TestEnumNeg::kTestMax>; |
| |
| // Should crash because kTestBelowMin is not in range. |
| EXPECT_CHECK_DEATH(TestEnumWithNegSet({TestEnumNeg::kTestBelowMin})); |
| // kTestD is in range, but note that kTestMin is negative. This should work. |
| EXPECT_TRUE( |
| TestEnumWithNegSet({TestEnumNeg::kTestD}).Has(TestEnumNeg::kTestD)); |
| // Even though kTestA is negative, it is in range, so this should work. |
| EXPECT_TRUE( |
| TestEnumWithNegSet({TestEnumNeg::kTestA}).Has(TestEnumNeg::kTestA)); |
| } |
| |
| TEST_F(EnumSetDeathTest, EnumWithOnlyNegatives) { |
| enum class TestEnumNeg { |
| kTestBelowMin = -10, |
| kTestA = -9, |
| kTestMin = kTestA, |
| kTestB = -8, |
| kTestC = -7, |
| kTestD = -6, |
| kTestMax = kTestD, |
| kTestF = -5, |
| }; |
| // This EnumSet starts negative and ends negative. |
| using TestEnumWithNegSet = |
| EnumSet<TestEnumNeg, TestEnumNeg::kTestMin, TestEnumNeg::kTestMax>; |
| |
| // Should crash because kTestBelowMin is not in range. |
| EXPECT_CHECK_DEATH(TestEnumWithNegSet({TestEnumNeg::kTestBelowMin})); |
| // kTestA, kTestD are in range, but note that kTestMin and values are |
| // negative. This should work. |
| EXPECT_TRUE( |
| TestEnumWithNegSet({TestEnumNeg::kTestA}).Has(TestEnumNeg::kTestA)); |
| EXPECT_TRUE( |
| TestEnumWithNegSet({TestEnumNeg::kTestD}).Has(TestEnumNeg::kTestD)); |
| } |
| |
| TEST_F(EnumSetDeathTest, VariadicConstructorCrashesOnOutOfRange) { |
| // Constructor should crash given out-of-range values. |
| EXPECT_CHECK_DEATH(TestEnumSet({TestEnum::kTestBelowMin}).empty()); |
| EXPECT_CHECK_DEATH(TestEnumSet({TestEnum::kTestBelowMinNegative}).empty()); |
| EXPECT_CHECK_DEATH(TestEnumSet({TestEnum::kTest6OutOfBounds}).empty()); |
| } |
| |
| TEST_F(EnumSetDeathTest, FromRangeCrashesOnBadInputs) { |
| // FromRange crashes when the bounds are in range, but out of order. |
| EXPECT_CHECK_DEATH( |
| TestEnumSet().FromRange(TestEnum::kTest3, TestEnum::kTest1)); |
| |
| // FromRange crashes when the start value is out of range. |
| EXPECT_CHECK_DEATH( |
| TestEnumSet().FromRange(TestEnum::kTestBelowMin, TestEnum::kTest1)); |
| EXPECT_CHECK_DEATH(TestEnumSet().FromRange(TestEnum::kTestBelowMinNegative, |
| TestEnum::kTest1)); |
| EXPECT_CHECK_DEATH( |
| TestEnumSet().FromRange(TestEnum::kTest6OutOfBounds, TestEnum::kTest1)); |
| |
| // FromRange crashes when the end value is out of range. |
| EXPECT_CHECK_DEATH( |
| TestEnumSet().FromRange(TestEnum::kTest3, TestEnum::kTestBelowMin)); |
| EXPECT_CHECK_DEATH(TestEnumSet().FromRange(TestEnum::kTest3, |
| TestEnum::kTestBelowMinNegative)); |
| EXPECT_CHECK_DEATH( |
| TestEnumSet().FromRange(TestEnum::kTest3, TestEnum::kTest6OutOfBounds)); |
| |
| // Crashes when start and end are both out of range. |
| EXPECT_CHECK_DEATH(TestEnumSet().FromRange(TestEnum::kTest6OutOfBounds, |
| TestEnum::kTest7OutOfBounds)); |
| EXPECT_CHECK_DEATH(TestEnumSet().FromRange(TestEnum::kTest6OutOfBounds, |
| TestEnum::kTest7OutOfBounds)); |
| } |
| |
| TEST_F(EnumSetDeathTest, PutCrashesOnOutOfRange) { |
| EXPECT_CHECK_DEATH(TestEnumSet().Put(TestEnum::kTestBelowMin)); |
| EXPECT_CHECK_DEATH(TestEnumSet().Put(TestEnum::kTestBelowMinNegative)); |
| EXPECT_CHECK_DEATH(TestEnumSet().Put(TestEnum::kTest6OutOfBounds)); |
| EXPECT_CHECK_DEATH(TestEnumSet().Put(TestEnum::kTest7OutOfBounds)); |
| } |
| |
| TEST_F(EnumSetDeathTest, PutRangeCrashesOnBadInputs) { |
| // Crashes when one input is out of range. |
| EXPECT_CHECK_DEATH(TestEnumSet().PutRange(TestEnum::kTestBelowMinNegative, |
| TestEnum::kTestBelowMin)); |
| EXPECT_CHECK_DEATH( |
| TestEnumSet().PutRange(TestEnum::kTest3, TestEnum::kTest7OutOfBounds)); |
| |
| // Crashes when both inputs are out of range. |
| EXPECT_CHECK_DEATH(TestEnumSet().PutRange(TestEnum::kTest6OutOfBounds, |
| TestEnum::kTest7OutOfBounds)); |
| |
| // Crashes when inputs are out of order. |
| EXPECT_CHECK_DEATH( |
| TestEnumSet().PutRange(TestEnum::kTest2, TestEnum::kTest1)); |
| } |
| |
| TEST_F(EnumSetTest, ToStringEmpty) { |
| const TestEnumSet enums; |
| EXPECT_THAT(enums.ToString(), testing::Eq("00000")); |
| } |
| |
| TEST_F(EnumSetTest, ToString) { |
| const TestEnumSet enums = {TestEnum::kTest4}; |
| EXPECT_THAT(enums.ToString(), testing::Eq("01000")); |
| } |
| |
| TEST_F(EnumSetTest, ToVectorEmpty) { |
| const TestEnumSet enums; |
| EXPECT_TRUE(ToVector(enums).empty()); |
| } |
| |
| TEST_F(EnumSetTest, ToVector) { |
| const TestEnumSet enums = {TestEnum::kTest2, TestEnum::kTest4}; |
| EXPECT_THAT(ToVector(enums), |
| testing::ElementsAre(TestEnum::kTest2, TestEnum::kTest4)); |
| } |
| |
| } // namespace |
| } // namespace base |