/*
 * All or portions of this file Copyright (c) Amazon.com, Inc. or its affiliates or
 * its licensors.
 *
 * For complete copyright and license terms please see the LICENSE at the root of this
 * distribution (the "License"). All use of this software is governed by the License,
 * or, if provided, by the license below or the license accompanying this file. Do not
 * remove or modify any license notices. This file is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *
 */

#include <AzCore/Math/Transform.h>
#include <AzCore/UnitTest/TestTypes.h>
#include <AzFramework/Viewport/CameraState.h>
#include <AzFramework/Viewport/ViewportScreen.h>
#include <AzTest/AzTest.h>
#include <AzToolsFramework/Viewport/ViewportMessages.h>
#include <AzToolsFramework/ViewportUi/ViewportUiDisplay.h>

namespace UnitTest
{
    using ButtonGroup = AzToolsFramework::ViewportUi::Internal::ButtonGroup;
    using ButtonId = AzToolsFramework::ViewportUi::ButtonId;

    TEST(ClusterTest, AddButtonAddsButtonToClusterAndReturnsId)
    {
        auto buttonGroup = AZStd::make_unique<ButtonGroup>();
        auto buttonId = buttonGroup->AddButton("");

        auto button = buttonGroup->GetButton(buttonId);
        EXPECT_TRUE(button != nullptr);
    }

    TEST(ClusterTest, SetHighlightedButtonChangesButtonStateToSelected)
    {
        auto buttonGroup = AZStd::make_unique<ButtonGroup>();
        auto buttonId = buttonGroup->AddButton("");

        // check button is not highlighted by default
        auto button = buttonGroup->GetButton(buttonId);
        EXPECT_FALSE(button->m_state == AzToolsFramework::ViewportUi::Internal::Button::State::Selected);

        buttonGroup->SetHighlightedButton(buttonId);
        EXPECT_TRUE(button->m_state == AzToolsFramework::ViewportUi::Internal::Button::State::Selected);
    }

    TEST(ClusterTest, ConnectEventHandlerConnectsHandlerToButtonTriggeredEvent)
    {
        auto buttonGroup = AZStd::make_unique<ButtonGroup>();
        auto buttonId = buttonGroup->AddButton("");

        // create a handler which will be triggered by the cluster
        bool handlerTriggered = false;
        auto testButtonId = ButtonId(buttonId);
        AZ::Event<ButtonId>::Handler handler(
            [&handlerTriggered, testButtonId](ButtonId buttonId)
            {
                if (buttonId == testButtonId)
                {
                    handlerTriggered = true;
                }
            });

        buttonGroup->ConnectEventHandler(handler);
        buttonGroup->PressButton(buttonId);

        EXPECT_TRUE(handlerTriggered);
    }
} // namespace UnitTest