/*
 * 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 "AzManipulatorTestFrameworkDirectCallTest.h"

namespace UnitTest
{
    TEST_F(AzManipulatorTestFrameworkCustomManagerTest, RegisterManipulatorWithCustomManager)
    {
        // Expect the manager to contain one manipulator
        EXPECT_EQ(m_manipulatorManager->RegisteredManipulatorCount(), 1);
    }

    TEST_F(AzManipulatorTestFrameworkCustomManagerTest, ConsumeViewportLeftMouseClick)
    {
        // given a left mouse down ray in world space
        const auto interaction = CreateMouseInteraction(
            m_mouseStartingPositionRay,
            AzToolsFramework::ViewportInteraction::MouseButton::Left);

        // consume the mouse down and up events
        m_manipulatorManager->ConsumeViewportMousePress(interaction);
        m_manipulatorManager->ConsumeViewportMouseRelease(interaction);

        // expect the left mouse down and mouse up sanity flags to be set
        EXPECT_TRUE(m_receivedLeftMouseDown);
        EXPECT_TRUE(m_receivedLeftMouseUp);

        // do not expect the mouse move sanity flag to be set
        EXPECT_FALSE(m_receivedMouseMove);
    }

    TEST_F(AzManipulatorTestFrameworkCustomManagerTest, ConsumeViewportMouseMoveHover)
    {
        // given a left mouse down ray in world space
        const auto interaction = CreateMouseInteraction(
            m_mouseStartingPositionRay,
            AzToolsFramework::ViewportInteraction::MouseButton::Left);

        // consume the mouse move event
        m_manipulatorManager->ConsumeViewportMouseMove(interaction);

        // do not expect the manipulator to be performing an action
        EXPECT_FALSE(m_linearManipulator->PerformingAction());
        EXPECT_FALSE(m_manipulatorManager->Interacting());

        // do not expect the left mouse down/up and mouse move sanity flags to be set
        EXPECT_FALSE(m_receivedLeftMouseDown);
        EXPECT_FALSE(m_receivedMouseMove);
        EXPECT_FALSE(m_receivedLeftMouseUp);
    }

    TEST_F(AzManipulatorTestFrameworkCustomManagerTest, ConsumeViewportMouseMoveActive)
    {
        // given a left mouse down ray in world space
        const auto interaction = CreateMouseInteraction(
            m_mouseStartingPositionRay,
            AzToolsFramework::ViewportInteraction::MouseButton::Left);

        // consume the mouse down and mouse move events
        m_manipulatorManager->ConsumeViewportMousePress(interaction);
        m_manipulatorManager->ConsumeViewportMouseMove(interaction);

        // do not expect the mouse to be hovering over the manipulator
        EXPECT_FALSE(m_linearManipulator->MouseOver());

        // expect the manipulator to be performing an action
        EXPECT_TRUE(m_linearManipulator->PerformingAction());
        EXPECT_TRUE(m_manipulatorManager->Interacting());

        // consume the mouse up event
        m_manipulatorManager->ConsumeViewportMouseRelease(interaction);

        // expect the left mouse down/up and mouse move sanity flags to be set
        EXPECT_TRUE(m_receivedLeftMouseDown);
        EXPECT_TRUE(m_receivedMouseMove);
        EXPECT_TRUE(m_receivedLeftMouseUp);
    }

    TEST_F(AzManipulatorTestFrameworkCustomManagerTest, MoveManipulatorAlongAxis)
    {
        AZ::Vector3 movementAlongAxis = AZ::Vector3::CreateZero();
        const AZ::Vector3 expectedPositionAfterMovementAlongAxis = AZ::Vector3(-5.0f, 0.0f, 0.0f);

        m_linearManipulator->InstallMouseMoveCallback(
            [&movementAlongAxis](const AzToolsFramework::LinearManipulator::Action& action)
        {
            movementAlongAxis = action.m_current.m_localPositionOffset;
        });

        // given a left mouse down ray in world space
        auto interaction = CreateMouseInteraction(
            m_mouseStartingPositionRay,
            AzToolsFramework::ViewportInteraction::MouseButton::Left);

        // consume the mouse down event
        m_manipulatorManager->ConsumeViewportMousePress(interaction);

        // move the mouse along the -x axis
        interaction.m_mousePick.m_rayOrigin += expectedPositionAfterMovementAlongAxis;
        m_manipulatorManager->ConsumeViewportMouseMove(interaction);

        // expect the manipulator to be performing an action
        EXPECT_TRUE(m_linearManipulator->PerformingAction());
        EXPECT_TRUE(m_manipulatorManager->Interacting());

        // consume the mouse up event
        m_manipulatorManager->ConsumeViewportMouseRelease(interaction);

        // expect the left mouse down/up sanity flags to be set
        EXPECT_TRUE(m_receivedLeftMouseDown);
        EXPECT_TRUE(m_receivedLeftMouseUp);

        // expect the manipulator movement along the axis to match the mouse movement along the axis
        EXPECT_EQ(movementAlongAxis, expectedPositionAfterMovementAlongAxis);
    }
} // namespace UnitTest