mfbt/tests/TestEnumSet.cpp

Wed, 31 Dec 2014 06:55:50 +0100

author
Michael Schloh von Bennewitz <michael@schloh.com>
date
Wed, 31 Dec 2014 06:55:50 +0100
changeset 2
7e26c7da4463
permissions
-rw-r--r--

Added tag UPSTREAM_283F7C6 for changeset ca08bd8f51b2

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 }

mercurial