/* -*- 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;
}