// Copyright 2018 The Fuchsia Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include <unittest/unittest.h>

#include "test_library.h"

namespace {

bool GoodEnumTestSimple() {
    BEGIN_TEST;

    TestLibrary library(R"FIDL(
library example;

enum Fruit : uint64 {
    ORANGE = 1;
    APPLE = 2;
    BANANA = 3;
};
)FIDL");
    ASSERT_TRUE(library.Compile());

    END_TEST;
}

bool BadEnumTestWithNonUniqueValues() {
    BEGIN_TEST;

    TestLibrary library(R"FIDL(
library example;

enum Fruit : uint64 {
    ORANGE = 1;
    APPLE = 1;
};
)FIDL");
    ASSERT_FALSE(library.Compile());
    auto errors = library.errors();
    ASSERT_EQ(errors.size(), 1);
    ASSERT_STR_STR(errors[0].c_str(), "value of member APPLE conflicts with previously declared "
                                      "member ORANGE in the enum Fruit");

    END_TEST;
}

bool BadEnumTestWithNonUniqueValuesOutOfLine() {
    BEGIN_TEST;

    TestLibrary library(R"FIDL(
library example;

enum Fruit {
    ORANGE = FOUR;
    APPLE = TWO_SQUARED;
};

const uint32 FOUR = 4;
const uint32 TWO_SQUARED = 4;
)FIDL");
    ASSERT_FALSE(library.Compile());
    auto errors = library.errors();
    ASSERT_EQ(errors.size(), 1);
    ASSERT_STR_STR(errors[0].c_str(), "value of member APPLE conflicts with previously declared "
                                      "member ORANGE in the enum Fruit");

    END_TEST;
}

bool BadEnumTestUnsignedWithNegativeMember() {
    BEGIN_TEST;

    TestLibrary library(R"FIDL(
library example;

enum Fruit : uint64 {
    ORANGE = 1;
    APPLE = -2;
};
)FIDL");
    ASSERT_FALSE(library.Compile());
    auto errors = library.errors();
    ASSERT_GE(errors.size(), 1);
    ASSERT_STR_STR(errors[0].c_str(), "-2 cannot be interpreted as type uint64");

    END_TEST;
}

bool BadEnumTestInferredUnsignedWithNegativeMember() {
    BEGIN_TEST;

    TestLibrary library(R"FIDL(
library example;

enum Fruit {
    ORANGE = 1;
    APPLE = -2;
};
)FIDL");
    ASSERT_FALSE(library.Compile());
    auto errors = library.errors();
    ASSERT_GE(errors.size(), 1);
    ASSERT_STR_STR(errors[0].c_str(), "-2 cannot be interpreted as type uint32");

    END_TEST;
}

bool BadEnumTestMemberOverflow() {
    BEGIN_TEST;

    TestLibrary library(R"FIDL(
library example;

enum Fruit : uint8 {
    ORANGE = 1;
    APPLE = 256;
};
)FIDL");
    ASSERT_FALSE(library.Compile());
    auto errors = library.errors();
    ASSERT_GE(errors.size(), 1);
    ASSERT_STR_STR(errors[0].c_str(), "256 cannot be interpreted as type uint8");

    END_TEST;
}

bool BadEnumTestDuplicateMember() {
    BEGIN_TEST;

    TestLibrary library(R"FIDL(
library example;

enum Fruit : uint64 {
    ORANGE = 1;
    APPLE = 2;
    ORANGE = 3;
};
)FIDL");
    ASSERT_FALSE(library.Compile());
    auto errors = library.errors();
    ASSERT_GE(errors.size(), 1);
    ASSERT_STR_STR(errors[0].c_str(), "name of member ORANGE conflicts with previously declared "
                                      "member in the enum Fruit");

    END_TEST;
}

} // namespace

BEGIN_TEST_CASE(enums_tests);

RUN_TEST(GoodEnumTestSimple);
RUN_TEST(BadEnumTestWithNonUniqueValues);
RUN_TEST(BadEnumTestWithNonUniqueValuesOutOfLine);
RUN_TEST(BadEnumTestUnsignedWithNegativeMember);
RUN_TEST(BadEnumTestInferredUnsignedWithNegativeMember);
RUN_TEST(BadEnumTestMemberOverflow);
RUN_TEST(BadEnumTestDuplicateMember);

END_TEST_CASE(enums_tests);