// Copyright 2020 The Dawn Authors
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

#include "dawn/native/EnumMaskIterator.h"

#include "gtest/gtest.h"

namespace dawn::native {

enum class TestAspect : uint8_t {
    Color = 1,
    Depth = 2,
    Stencil = 4,
};

template <>
struct EnumBitmaskSize<TestAspect> {
    static constexpr unsigned value = 3;
};

}  // namespace dawn::native

namespace dawn {

template <>
struct IsDawnBitmask<dawn::native::TestAspect> {
    static constexpr bool enable = true;
};

}  // namespace dawn

namespace dawn::native {

static_assert(EnumBitmaskSize<TestAspect>::value == 3);

TEST(EnumMaskIteratorTests, None) {
    for (TestAspect aspect : IterateEnumMask(static_cast<TestAspect>(0))) {
        FAIL();
        DAWN_UNUSED(aspect);
    }
}

TEST(EnumMaskIteratorTests, All) {
    TestAspect expected[] = {TestAspect::Color, TestAspect::Depth, TestAspect::Stencil};
    uint32_t i = 0;
    TestAspect aspects = TestAspect::Color | TestAspect::Depth | TestAspect::Stencil;
    for (TestAspect aspect : IterateEnumMask(aspects)) {
        EXPECT_EQ(aspect, expected[i++]);
    }
}

TEST(EnumMaskIteratorTests, Partial) {
    TestAspect expected[] = {TestAspect::Color, TestAspect::Stencil};
    uint32_t i = 0;
    TestAspect aspects = TestAspect::Stencil | TestAspect::Color;
    for (TestAspect aspect : IterateEnumMask(aspects)) {
        EXPECT_EQ(aspect, expected[i++]);
    }
}

}  // namespace dawn::native