summaryrefslogtreecommitdiffstats
path: root/mfbt/tests/TestEnumeratedArray.cpp
blob: d5ec9ebbb313a3be4a5d4d5a67a8a3b924650257 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/* vim: set ts=8 sts=2 et sw=2 tw=80: */
/* This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this file,
 * You can obtain one at http://mozilla.org/MPL/2.0/. */

#include "mozilla/ArrayUtils.h"
#include "mozilla/EnumeratedArray.h"
#include "mozilla/EnumTypeTraits.h"

using mozilla::EnumeratedArray;

enum class AnimalSpecies { Cow, Sheep, Pig };

template <>
struct mozilla::MaxContiguousEnumValue<AnimalSpecies> {
  static constexpr AnimalSpecies value = AnimalSpecies::Pig;
};

using TestArray = EnumeratedArray<AnimalSpecies, int>;

void TestInitialValueByConstructor() {
  // Style 1
  TestArray headCount(1, 2, 3);
  MOZ_RELEASE_ASSERT(mozilla::ArrayLength(headCount) == 3);
  MOZ_RELEASE_ASSERT(headCount[AnimalSpecies::Cow] == 1);
  MOZ_RELEASE_ASSERT(headCount[AnimalSpecies::Sheep] == 2);
  MOZ_RELEASE_ASSERT(headCount[AnimalSpecies::Pig] == 3);
  // Style 2
  TestArray headCount2{5, 6, 7};
  MOZ_RELEASE_ASSERT(headCount2[AnimalSpecies::Cow] == 5);
  MOZ_RELEASE_ASSERT(headCount2[AnimalSpecies::Sheep] == 6);
  MOZ_RELEASE_ASSERT(headCount2[AnimalSpecies::Pig] == 7);
  // Style 3
  TestArray headCount3({8, 9, 10});
  MOZ_RELEASE_ASSERT(headCount3[AnimalSpecies::Cow] == 8);
  MOZ_RELEASE_ASSERT(headCount3[AnimalSpecies::Sheep] == 9);
  MOZ_RELEASE_ASSERT(headCount3[AnimalSpecies::Pig] == 10);
}

void TestAssignment() {
  TestArray headCount{8, 9, 10};
  TestArray headCount2;
  headCount2 = headCount;
  MOZ_RELEASE_ASSERT(headCount2[AnimalSpecies::Cow] == 8);
  MOZ_RELEASE_ASSERT(headCount2[AnimalSpecies::Sheep] == 9);
  MOZ_RELEASE_ASSERT(headCount2[AnimalSpecies::Pig] == 10);
}

int main() {
  TestInitialValueByConstructor();
  TestAssignment();
  return 0;
}