Fri, 16 Jan 2015 18:13:44 +0100
Integrate suggestion from review to improve consistency with existing code.
michael@0 | 1 | /* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */ |
michael@0 | 2 | /* This Source Code Form is subject to the terms of the Mozilla Public |
michael@0 | 3 | * License, v. 2.0. If a copy of the MPL was not distributed with this file, |
michael@0 | 4 | * You can obtain one at http://mozilla.org/MPL/2.0/. */ |
michael@0 | 5 | |
michael@0 | 6 | #include "mozilla/EnumSet.h" |
michael@0 | 7 | |
michael@0 | 8 | using namespace mozilla; |
michael@0 | 9 | |
michael@0 | 10 | enum SeaBird { |
michael@0 | 11 | PENGUIN, |
michael@0 | 12 | ALBATROSS, |
michael@0 | 13 | FULMAR, |
michael@0 | 14 | PRION, |
michael@0 | 15 | SHEARWATER, |
michael@0 | 16 | GADFLY_PETREL, |
michael@0 | 17 | TRUE_PETREL, |
michael@0 | 18 | DIVING_PETREL, |
michael@0 | 19 | STORM_PETREL, |
michael@0 | 20 | PELICAN, |
michael@0 | 21 | GANNET, |
michael@0 | 22 | BOOBY, |
michael@0 | 23 | CORMORANT, |
michael@0 | 24 | FRIGATEBIRD, |
michael@0 | 25 | TROPICBIRD, |
michael@0 | 26 | SKUA, |
michael@0 | 27 | GULL, |
michael@0 | 28 | TERN, |
michael@0 | 29 | SKIMMER, |
michael@0 | 30 | AUK |
michael@0 | 31 | }; |
michael@0 | 32 | |
michael@0 | 33 | class EnumSetSuite { |
michael@0 | 34 | public: |
michael@0 | 35 | EnumSetSuite() |
michael@0 | 36 | : mAlcidae(), |
michael@0 | 37 | mDiomedeidae(ALBATROSS), |
michael@0 | 38 | mPetrelProcellariidae(GADFLY_PETREL, TRUE_PETREL), |
michael@0 | 39 | mNonPetrelProcellariidae(FULMAR, PRION, SHEARWATER), |
michael@0 | 40 | mPetrels(GADFLY_PETREL, TRUE_PETREL, DIVING_PETREL, STORM_PETREL) |
michael@0 | 41 | { } |
michael@0 | 42 | |
michael@0 | 43 | void runTests() { |
michael@0 | 44 | testSize(); |
michael@0 | 45 | testContains(); |
michael@0 | 46 | testAddTo(); |
michael@0 | 47 | testAdd(); |
michael@0 | 48 | testAddAll(); |
michael@0 | 49 | testUnion(); |
michael@0 | 50 | testRemoveFrom(); |
michael@0 | 51 | testRemove(); |
michael@0 | 52 | testRemoveAllFrom(); |
michael@0 | 53 | testRemoveAll(); |
michael@0 | 54 | testIntersect(); |
michael@0 | 55 | testInsersection(); |
michael@0 | 56 | testEquality(); |
michael@0 | 57 | testDuplicates(); |
michael@0 | 58 | } |
michael@0 | 59 | |
michael@0 | 60 | private: |
michael@0 | 61 | void testSize() { |
michael@0 | 62 | MOZ_RELEASE_ASSERT(mAlcidae.size() == 0); |
michael@0 | 63 | MOZ_RELEASE_ASSERT(mDiomedeidae.size() == 1); |
michael@0 | 64 | MOZ_RELEASE_ASSERT(mPetrelProcellariidae.size() == 2); |
michael@0 | 65 | MOZ_RELEASE_ASSERT(mNonPetrelProcellariidae.size() == 3); |
michael@0 | 66 | MOZ_RELEASE_ASSERT(mPetrels.size() == 4); |
michael@0 | 67 | } |
michael@0 | 68 | |
michael@0 | 69 | void testContains() { |
michael@0 | 70 | MOZ_RELEASE_ASSERT(!mPetrels.contains(PENGUIN)); |
michael@0 | 71 | MOZ_RELEASE_ASSERT(!mPetrels.contains(ALBATROSS)); |
michael@0 | 72 | MOZ_RELEASE_ASSERT(!mPetrels.contains(FULMAR)); |
michael@0 | 73 | MOZ_RELEASE_ASSERT(!mPetrels.contains(PRION)); |
michael@0 | 74 | MOZ_RELEASE_ASSERT(!mPetrels.contains(SHEARWATER)); |
michael@0 | 75 | MOZ_RELEASE_ASSERT(mPetrels.contains(GADFLY_PETREL)); |
michael@0 | 76 | MOZ_RELEASE_ASSERT(mPetrels.contains(TRUE_PETREL)); |
michael@0 | 77 | MOZ_RELEASE_ASSERT(mPetrels.contains(DIVING_PETREL)); |
michael@0 | 78 | MOZ_RELEASE_ASSERT(mPetrels.contains(STORM_PETREL)); |
michael@0 | 79 | MOZ_RELEASE_ASSERT(!mPetrels.contains(PELICAN)); |
michael@0 | 80 | MOZ_RELEASE_ASSERT(!mPetrels.contains(GANNET)); |
michael@0 | 81 | MOZ_RELEASE_ASSERT(!mPetrels.contains(BOOBY)); |
michael@0 | 82 | MOZ_RELEASE_ASSERT(!mPetrels.contains(CORMORANT)); |
michael@0 | 83 | MOZ_RELEASE_ASSERT(!mPetrels.contains(FRIGATEBIRD)); |
michael@0 | 84 | MOZ_RELEASE_ASSERT(!mPetrels.contains(TROPICBIRD)); |
michael@0 | 85 | MOZ_RELEASE_ASSERT(!mPetrels.contains(SKUA)); |
michael@0 | 86 | MOZ_RELEASE_ASSERT(!mPetrels.contains(GULL)); |
michael@0 | 87 | MOZ_RELEASE_ASSERT(!mPetrels.contains(TERN)); |
michael@0 | 88 | MOZ_RELEASE_ASSERT(!mPetrels.contains(SKIMMER)); |
michael@0 | 89 | MOZ_RELEASE_ASSERT(!mPetrels.contains(AUK)); |
michael@0 | 90 | } |
michael@0 | 91 | |
michael@0 | 92 | void testCopy() { |
michael@0 | 93 | EnumSet<SeaBird> likes = mPetrels; |
michael@0 | 94 | likes -= TRUE_PETREL; |
michael@0 | 95 | MOZ_RELEASE_ASSERT(mPetrels.size() == 4); |
michael@0 | 96 | MOZ_RELEASE_ASSERT(mPetrels.contains(TRUE_PETREL)); |
michael@0 | 97 | |
michael@0 | 98 | MOZ_RELEASE_ASSERT(likes.size() == 3); |
michael@0 | 99 | MOZ_RELEASE_ASSERT(likes.contains(GADFLY_PETREL)); |
michael@0 | 100 | MOZ_RELEASE_ASSERT(likes.contains(DIVING_PETREL)); |
michael@0 | 101 | MOZ_RELEASE_ASSERT(likes.contains(STORM_PETREL)); |
michael@0 | 102 | } |
michael@0 | 103 | |
michael@0 | 104 | void testAddTo() { |
michael@0 | 105 | EnumSet<SeaBird> seen = mPetrels; |
michael@0 | 106 | seen += CORMORANT; |
michael@0 | 107 | seen += TRUE_PETREL; |
michael@0 | 108 | MOZ_RELEASE_ASSERT(mPetrels.size() == 4); |
michael@0 | 109 | MOZ_RELEASE_ASSERT(!mPetrels.contains(CORMORANT)); |
michael@0 | 110 | MOZ_RELEASE_ASSERT(seen.size() == 5); |
michael@0 | 111 | MOZ_RELEASE_ASSERT(seen.contains(GADFLY_PETREL)); |
michael@0 | 112 | MOZ_RELEASE_ASSERT(seen.contains(TRUE_PETREL)); |
michael@0 | 113 | MOZ_RELEASE_ASSERT(seen.contains(DIVING_PETREL)); |
michael@0 | 114 | MOZ_RELEASE_ASSERT(seen.contains(STORM_PETREL)); |
michael@0 | 115 | MOZ_RELEASE_ASSERT(seen.contains(CORMORANT)); |
michael@0 | 116 | } |
michael@0 | 117 | |
michael@0 | 118 | void testAdd() { |
michael@0 | 119 | EnumSet<SeaBird> seen = mPetrels + CORMORANT + |
michael@0 | 120 | STORM_PETREL; |
michael@0 | 121 | MOZ_RELEASE_ASSERT(mPetrels.size() == 4); |
michael@0 | 122 | MOZ_RELEASE_ASSERT(!mPetrels.contains(CORMORANT)); |
michael@0 | 123 | MOZ_RELEASE_ASSERT(seen.size() == 5); |
michael@0 | 124 | MOZ_RELEASE_ASSERT(seen.contains(GADFLY_PETREL)); |
michael@0 | 125 | MOZ_RELEASE_ASSERT(seen.contains(TRUE_PETREL)); |
michael@0 | 126 | MOZ_RELEASE_ASSERT(seen.contains(DIVING_PETREL)); |
michael@0 | 127 | MOZ_RELEASE_ASSERT(seen.contains(STORM_PETREL)); |
michael@0 | 128 | MOZ_RELEASE_ASSERT(seen.contains(CORMORANT)); |
michael@0 | 129 | } |
michael@0 | 130 | |
michael@0 | 131 | void testAddAll() { |
michael@0 | 132 | EnumSet<SeaBird> procellariidae; |
michael@0 | 133 | procellariidae += mPetrelProcellariidae; |
michael@0 | 134 | procellariidae += mNonPetrelProcellariidae; |
michael@0 | 135 | MOZ_RELEASE_ASSERT(procellariidae.size() == 5); |
michael@0 | 136 | |
michael@0 | 137 | // Both procellariidae and mPetrels include GADFLY_PERTEL and TRUE_PETREL |
michael@0 | 138 | EnumSet<SeaBird> procellariiformes; |
michael@0 | 139 | procellariiformes += mDiomedeidae; |
michael@0 | 140 | procellariiformes += procellariidae; |
michael@0 | 141 | procellariiformes += mPetrels; |
michael@0 | 142 | MOZ_RELEASE_ASSERT(procellariiformes.size() == 8); |
michael@0 | 143 | } |
michael@0 | 144 | |
michael@0 | 145 | void testUnion() { |
michael@0 | 146 | EnumSet<SeaBird> procellariidae = mPetrelProcellariidae + |
michael@0 | 147 | mNonPetrelProcellariidae; |
michael@0 | 148 | MOZ_RELEASE_ASSERT(procellariidae.size() == 5); |
michael@0 | 149 | |
michael@0 | 150 | // Both procellariidae and mPetrels include GADFLY_PETREL and TRUE_PETREL |
michael@0 | 151 | EnumSet<SeaBird> procellariiformes = mDiomedeidae + procellariidae + |
michael@0 | 152 | mPetrels; |
michael@0 | 153 | MOZ_RELEASE_ASSERT(procellariiformes.size() == 8); |
michael@0 | 154 | } |
michael@0 | 155 | |
michael@0 | 156 | void testRemoveFrom() { |
michael@0 | 157 | EnumSet<SeaBird> likes = mPetrels; |
michael@0 | 158 | likes -= TRUE_PETREL; |
michael@0 | 159 | likes -= DIVING_PETREL; |
michael@0 | 160 | MOZ_RELEASE_ASSERT(likes.size() == 2); |
michael@0 | 161 | MOZ_RELEASE_ASSERT(likes.contains(GADFLY_PETREL)); |
michael@0 | 162 | MOZ_RELEASE_ASSERT(likes.contains(STORM_PETREL)); |
michael@0 | 163 | } |
michael@0 | 164 | |
michael@0 | 165 | void testRemove() { |
michael@0 | 166 | EnumSet<SeaBird> likes = mPetrels - TRUE_PETREL - |
michael@0 | 167 | DIVING_PETREL; |
michael@0 | 168 | MOZ_RELEASE_ASSERT(likes.size() == 2); |
michael@0 | 169 | MOZ_RELEASE_ASSERT(likes.contains(GADFLY_PETREL)); |
michael@0 | 170 | MOZ_RELEASE_ASSERT(likes.contains(STORM_PETREL)); |
michael@0 | 171 | } |
michael@0 | 172 | |
michael@0 | 173 | void testRemoveAllFrom() { |
michael@0 | 174 | EnumSet<SeaBird> likes = mPetrels; |
michael@0 | 175 | likes -= mPetrelProcellariidae; |
michael@0 | 176 | MOZ_RELEASE_ASSERT(likes.size() == 2); |
michael@0 | 177 | MOZ_RELEASE_ASSERT(likes.contains(DIVING_PETREL)); |
michael@0 | 178 | MOZ_RELEASE_ASSERT(likes.contains(STORM_PETREL)); |
michael@0 | 179 | } |
michael@0 | 180 | |
michael@0 | 181 | void testRemoveAll() { |
michael@0 | 182 | EnumSet<SeaBird> likes = mPetrels - mPetrelProcellariidae; |
michael@0 | 183 | MOZ_RELEASE_ASSERT(likes.size() == 2); |
michael@0 | 184 | MOZ_RELEASE_ASSERT(likes.contains(DIVING_PETREL)); |
michael@0 | 185 | MOZ_RELEASE_ASSERT(likes.contains(STORM_PETREL)); |
michael@0 | 186 | } |
michael@0 | 187 | |
michael@0 | 188 | void testIntersect() { |
michael@0 | 189 | EnumSet<SeaBird> likes = mPetrels; |
michael@0 | 190 | likes &= mPetrelProcellariidae; |
michael@0 | 191 | MOZ_RELEASE_ASSERT(likes.size() == 2); |
michael@0 | 192 | MOZ_RELEASE_ASSERT(likes.contains(GADFLY_PETREL)); |
michael@0 | 193 | MOZ_RELEASE_ASSERT(likes.contains(TRUE_PETREL)); |
michael@0 | 194 | } |
michael@0 | 195 | |
michael@0 | 196 | void testInsersection() { |
michael@0 | 197 | EnumSet<SeaBird> likes = mPetrels & mPetrelProcellariidae; |
michael@0 | 198 | MOZ_RELEASE_ASSERT(likes.size() == 2); |
michael@0 | 199 | MOZ_RELEASE_ASSERT(likes.contains(GADFLY_PETREL)); |
michael@0 | 200 | MOZ_RELEASE_ASSERT(likes.contains(TRUE_PETREL)); |
michael@0 | 201 | } |
michael@0 | 202 | |
michael@0 | 203 | void testEquality() { |
michael@0 | 204 | EnumSet<SeaBird> likes = mPetrels & mPetrelProcellariidae; |
michael@0 | 205 | MOZ_RELEASE_ASSERT(likes == EnumSet<SeaBird>(GADFLY_PETREL, |
michael@0 | 206 | TRUE_PETREL)); |
michael@0 | 207 | } |
michael@0 | 208 | |
michael@0 | 209 | void testDuplicates() { |
michael@0 | 210 | EnumSet<SeaBird> likes = mPetrels; |
michael@0 | 211 | likes += GADFLY_PETREL; |
michael@0 | 212 | likes += TRUE_PETREL; |
michael@0 | 213 | likes += DIVING_PETREL; |
michael@0 | 214 | likes += STORM_PETREL; |
michael@0 | 215 | MOZ_RELEASE_ASSERT(likes.size() == 4); |
michael@0 | 216 | MOZ_RELEASE_ASSERT(likes == mPetrels); |
michael@0 | 217 | } |
michael@0 | 218 | |
michael@0 | 219 | |
michael@0 | 220 | EnumSet<SeaBird> mAlcidae; |
michael@0 | 221 | EnumSet<SeaBird> mDiomedeidae; |
michael@0 | 222 | EnumSet<SeaBird> mPetrelProcellariidae; |
michael@0 | 223 | EnumSet<SeaBird> mNonPetrelProcellariidae; |
michael@0 | 224 | EnumSet<SeaBird> mPetrels; |
michael@0 | 225 | }; |
michael@0 | 226 | |
michael@0 | 227 | int main() |
michael@0 | 228 | { |
michael@0 | 229 | EnumSetSuite suite; |
michael@0 | 230 | suite.runTests(); |
michael@0 | 231 | return 0; |
michael@0 | 232 | } |