1.1 --- /dev/null Thu Jan 01 00:00:00 1970 +0000
1.2 +++ b/mfbt/tests/TestEnumSet.cpp Wed Dec 31 06:09:35 2014 +0100
1.3 @@ -0,0 +1,232 @@
1.4 +/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
1.5 +/* This Source Code Form is subject to the terms of the Mozilla Public
1.6 + * License, v. 2.0. If a copy of the MPL was not distributed with this file,
1.7 + * You can obtain one at http://mozilla.org/MPL/2.0/. */
1.8 +
1.9 +#include "mozilla/EnumSet.h"
1.10 +
1.11 +using namespace mozilla;
1.12 +
1.13 +enum SeaBird {
1.14 + PENGUIN,
1.15 + ALBATROSS,
1.16 + FULMAR,
1.17 + PRION,
1.18 + SHEARWATER,
1.19 + GADFLY_PETREL,
1.20 + TRUE_PETREL,
1.21 + DIVING_PETREL,
1.22 + STORM_PETREL,
1.23 + PELICAN,
1.24 + GANNET,
1.25 + BOOBY,
1.26 + CORMORANT,
1.27 + FRIGATEBIRD,
1.28 + TROPICBIRD,
1.29 + SKUA,
1.30 + GULL,
1.31 + TERN,
1.32 + SKIMMER,
1.33 + AUK
1.34 +};
1.35 +
1.36 +class EnumSetSuite {
1.37 + public:
1.38 + EnumSetSuite()
1.39 + : mAlcidae(),
1.40 + mDiomedeidae(ALBATROSS),
1.41 + mPetrelProcellariidae(GADFLY_PETREL, TRUE_PETREL),
1.42 + mNonPetrelProcellariidae(FULMAR, PRION, SHEARWATER),
1.43 + mPetrels(GADFLY_PETREL, TRUE_PETREL, DIVING_PETREL, STORM_PETREL)
1.44 + { }
1.45 +
1.46 + void runTests() {
1.47 + testSize();
1.48 + testContains();
1.49 + testAddTo();
1.50 + testAdd();
1.51 + testAddAll();
1.52 + testUnion();
1.53 + testRemoveFrom();
1.54 + testRemove();
1.55 + testRemoveAllFrom();
1.56 + testRemoveAll();
1.57 + testIntersect();
1.58 + testInsersection();
1.59 + testEquality();
1.60 + testDuplicates();
1.61 + }
1.62 +
1.63 + private:
1.64 + void testSize() {
1.65 + MOZ_RELEASE_ASSERT(mAlcidae.size() == 0);
1.66 + MOZ_RELEASE_ASSERT(mDiomedeidae.size() == 1);
1.67 + MOZ_RELEASE_ASSERT(mPetrelProcellariidae.size() == 2);
1.68 + MOZ_RELEASE_ASSERT(mNonPetrelProcellariidae.size() == 3);
1.69 + MOZ_RELEASE_ASSERT(mPetrels.size() == 4);
1.70 + }
1.71 +
1.72 + void testContains() {
1.73 + MOZ_RELEASE_ASSERT(!mPetrels.contains(PENGUIN));
1.74 + MOZ_RELEASE_ASSERT(!mPetrels.contains(ALBATROSS));
1.75 + MOZ_RELEASE_ASSERT(!mPetrels.contains(FULMAR));
1.76 + MOZ_RELEASE_ASSERT(!mPetrels.contains(PRION));
1.77 + MOZ_RELEASE_ASSERT(!mPetrels.contains(SHEARWATER));
1.78 + MOZ_RELEASE_ASSERT(mPetrels.contains(GADFLY_PETREL));
1.79 + MOZ_RELEASE_ASSERT(mPetrels.contains(TRUE_PETREL));
1.80 + MOZ_RELEASE_ASSERT(mPetrels.contains(DIVING_PETREL));
1.81 + MOZ_RELEASE_ASSERT(mPetrels.contains(STORM_PETREL));
1.82 + MOZ_RELEASE_ASSERT(!mPetrels.contains(PELICAN));
1.83 + MOZ_RELEASE_ASSERT(!mPetrels.contains(GANNET));
1.84 + MOZ_RELEASE_ASSERT(!mPetrels.contains(BOOBY));
1.85 + MOZ_RELEASE_ASSERT(!mPetrels.contains(CORMORANT));
1.86 + MOZ_RELEASE_ASSERT(!mPetrels.contains(FRIGATEBIRD));
1.87 + MOZ_RELEASE_ASSERT(!mPetrels.contains(TROPICBIRD));
1.88 + MOZ_RELEASE_ASSERT(!mPetrels.contains(SKUA));
1.89 + MOZ_RELEASE_ASSERT(!mPetrels.contains(GULL));
1.90 + MOZ_RELEASE_ASSERT(!mPetrels.contains(TERN));
1.91 + MOZ_RELEASE_ASSERT(!mPetrels.contains(SKIMMER));
1.92 + MOZ_RELEASE_ASSERT(!mPetrels.contains(AUK));
1.93 + }
1.94 +
1.95 + void testCopy() {
1.96 + EnumSet<SeaBird> likes = mPetrels;
1.97 + likes -= TRUE_PETREL;
1.98 + MOZ_RELEASE_ASSERT(mPetrels.size() == 4);
1.99 + MOZ_RELEASE_ASSERT(mPetrels.contains(TRUE_PETREL));
1.100 +
1.101 + MOZ_RELEASE_ASSERT(likes.size() == 3);
1.102 + MOZ_RELEASE_ASSERT(likes.contains(GADFLY_PETREL));
1.103 + MOZ_RELEASE_ASSERT(likes.contains(DIVING_PETREL));
1.104 + MOZ_RELEASE_ASSERT(likes.contains(STORM_PETREL));
1.105 + }
1.106 +
1.107 + void testAddTo() {
1.108 + EnumSet<SeaBird> seen = mPetrels;
1.109 + seen += CORMORANT;
1.110 + seen += TRUE_PETREL;
1.111 + MOZ_RELEASE_ASSERT(mPetrels.size() == 4);
1.112 + MOZ_RELEASE_ASSERT(!mPetrels.contains(CORMORANT));
1.113 + MOZ_RELEASE_ASSERT(seen.size() == 5);
1.114 + MOZ_RELEASE_ASSERT(seen.contains(GADFLY_PETREL));
1.115 + MOZ_RELEASE_ASSERT(seen.contains(TRUE_PETREL));
1.116 + MOZ_RELEASE_ASSERT(seen.contains(DIVING_PETREL));
1.117 + MOZ_RELEASE_ASSERT(seen.contains(STORM_PETREL));
1.118 + MOZ_RELEASE_ASSERT(seen.contains(CORMORANT));
1.119 + }
1.120 +
1.121 + void testAdd() {
1.122 + EnumSet<SeaBird> seen = mPetrels + CORMORANT +
1.123 + STORM_PETREL;
1.124 + MOZ_RELEASE_ASSERT(mPetrels.size() == 4);
1.125 + MOZ_RELEASE_ASSERT(!mPetrels.contains(CORMORANT));
1.126 + MOZ_RELEASE_ASSERT(seen.size() == 5);
1.127 + MOZ_RELEASE_ASSERT(seen.contains(GADFLY_PETREL));
1.128 + MOZ_RELEASE_ASSERT(seen.contains(TRUE_PETREL));
1.129 + MOZ_RELEASE_ASSERT(seen.contains(DIVING_PETREL));
1.130 + MOZ_RELEASE_ASSERT(seen.contains(STORM_PETREL));
1.131 + MOZ_RELEASE_ASSERT(seen.contains(CORMORANT));
1.132 + }
1.133 +
1.134 + void testAddAll() {
1.135 + EnumSet<SeaBird> procellariidae;
1.136 + procellariidae += mPetrelProcellariidae;
1.137 + procellariidae += mNonPetrelProcellariidae;
1.138 + MOZ_RELEASE_ASSERT(procellariidae.size() == 5);
1.139 +
1.140 + // Both procellariidae and mPetrels include GADFLY_PERTEL and TRUE_PETREL
1.141 + EnumSet<SeaBird> procellariiformes;
1.142 + procellariiformes += mDiomedeidae;
1.143 + procellariiformes += procellariidae;
1.144 + procellariiformes += mPetrels;
1.145 + MOZ_RELEASE_ASSERT(procellariiformes.size() == 8);
1.146 + }
1.147 +
1.148 + void testUnion() {
1.149 + EnumSet<SeaBird> procellariidae = mPetrelProcellariidae +
1.150 + mNonPetrelProcellariidae;
1.151 + MOZ_RELEASE_ASSERT(procellariidae.size() == 5);
1.152 +
1.153 + // Both procellariidae and mPetrels include GADFLY_PETREL and TRUE_PETREL
1.154 + EnumSet<SeaBird> procellariiformes = mDiomedeidae + procellariidae +
1.155 + mPetrels;
1.156 + MOZ_RELEASE_ASSERT(procellariiformes.size() == 8);
1.157 + }
1.158 +
1.159 + void testRemoveFrom() {
1.160 + EnumSet<SeaBird> likes = mPetrels;
1.161 + likes -= TRUE_PETREL;
1.162 + likes -= DIVING_PETREL;
1.163 + MOZ_RELEASE_ASSERT(likes.size() == 2);
1.164 + MOZ_RELEASE_ASSERT(likes.contains(GADFLY_PETREL));
1.165 + MOZ_RELEASE_ASSERT(likes.contains(STORM_PETREL));
1.166 + }
1.167 +
1.168 + void testRemove() {
1.169 + EnumSet<SeaBird> likes = mPetrels - TRUE_PETREL -
1.170 + DIVING_PETREL;
1.171 + MOZ_RELEASE_ASSERT(likes.size() == 2);
1.172 + MOZ_RELEASE_ASSERT(likes.contains(GADFLY_PETREL));
1.173 + MOZ_RELEASE_ASSERT(likes.contains(STORM_PETREL));
1.174 + }
1.175 +
1.176 + void testRemoveAllFrom() {
1.177 + EnumSet<SeaBird> likes = mPetrels;
1.178 + likes -= mPetrelProcellariidae;
1.179 + MOZ_RELEASE_ASSERT(likes.size() == 2);
1.180 + MOZ_RELEASE_ASSERT(likes.contains(DIVING_PETREL));
1.181 + MOZ_RELEASE_ASSERT(likes.contains(STORM_PETREL));
1.182 + }
1.183 +
1.184 + void testRemoveAll() {
1.185 + EnumSet<SeaBird> likes = mPetrels - mPetrelProcellariidae;
1.186 + MOZ_RELEASE_ASSERT(likes.size() == 2);
1.187 + MOZ_RELEASE_ASSERT(likes.contains(DIVING_PETREL));
1.188 + MOZ_RELEASE_ASSERT(likes.contains(STORM_PETREL));
1.189 + }
1.190 +
1.191 + void testIntersect() {
1.192 + EnumSet<SeaBird> likes = mPetrels;
1.193 + likes &= mPetrelProcellariidae;
1.194 + MOZ_RELEASE_ASSERT(likes.size() == 2);
1.195 + MOZ_RELEASE_ASSERT(likes.contains(GADFLY_PETREL));
1.196 + MOZ_RELEASE_ASSERT(likes.contains(TRUE_PETREL));
1.197 + }
1.198 +
1.199 + void testInsersection() {
1.200 + EnumSet<SeaBird> likes = mPetrels & mPetrelProcellariidae;
1.201 + MOZ_RELEASE_ASSERT(likes.size() == 2);
1.202 + MOZ_RELEASE_ASSERT(likes.contains(GADFLY_PETREL));
1.203 + MOZ_RELEASE_ASSERT(likes.contains(TRUE_PETREL));
1.204 + }
1.205 +
1.206 + void testEquality() {
1.207 + EnumSet<SeaBird> likes = mPetrels & mPetrelProcellariidae;
1.208 + MOZ_RELEASE_ASSERT(likes == EnumSet<SeaBird>(GADFLY_PETREL,
1.209 + TRUE_PETREL));
1.210 + }
1.211 +
1.212 + void testDuplicates() {
1.213 + EnumSet<SeaBird> likes = mPetrels;
1.214 + likes += GADFLY_PETREL;
1.215 + likes += TRUE_PETREL;
1.216 + likes += DIVING_PETREL;
1.217 + likes += STORM_PETREL;
1.218 + MOZ_RELEASE_ASSERT(likes.size() == 4);
1.219 + MOZ_RELEASE_ASSERT(likes == mPetrels);
1.220 + }
1.221 +
1.222 +
1.223 + EnumSet<SeaBird> mAlcidae;
1.224 + EnumSet<SeaBird> mDiomedeidae;
1.225 + EnumSet<SeaBird> mPetrelProcellariidae;
1.226 + EnumSet<SeaBird> mNonPetrelProcellariidae;
1.227 + EnumSet<SeaBird> mPetrels;
1.228 +};
1.229 +
1.230 +int main()
1.231 +{
1.232 + EnumSetSuite suite;
1.233 + suite.runTests();
1.234 + return 0;
1.235 +}
