o3de/Code/Framework/AzManipulatorTestFramework/Tests/DirectCallTest.cpp

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

namespace UnitTest
{
    class CustomManipulatorManager : public AzToolsFramework::ManipulatorManager
    {
        using ManagerBase = AzToolsFramework::ManipulatorManager;

    public:
        using ManagerBase::ManagerBase;

        size_t RegisteredManipulatorCount() const
        {
            return m_manipulatorIdToPtrMap.size();
        }
    };

    class AzManipulatorTestFrameworkCustomManagerTestFixture : public LinearManipulatorTestFixture
    {
    protected:
        AzManipulatorTestFrameworkCustomManagerTestFixture()
            : LinearManipulatorTestFixture(AzToolsFramework::ManipulatorManagerId(AZ::Crc32("TestManipulatorManagerId")))
        {
        }

        void SetUpEditorFixtureImpl() override
        {
            m_manipulatorManager = AZStd::make_shared<CustomManipulatorManager>(m_manipulatorManagerId);

            LinearManipulatorTestFixture::SetUpEditorFixtureImpl();
        }

        void TearDownEditorFixtureImpl() override
        {
            LinearManipulatorTestFixture::TearDownEditorFixtureImpl();
            m_manipulatorManager.reset();
        }

        AZStd::shared_ptr<CustomManipulatorManager> m_manipulatorManager;
    };

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

    TEST_F(AzManipulatorTestFrameworkCustomManagerTestFixture, ConsumeViewportLeftMouseClick)
    {
        // consume the mouse down and up events
        m_manipulatorManager->ConsumeViewportMousePress(m_interaction);
        m_manipulatorManager->ConsumeViewportMouseRelease(m_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(AzManipulatorTestFrameworkCustomManagerTestFixture, ConsumeViewportMouseMoveHover)
    {
        // consume the mouse move event
        m_manipulatorManager->ConsumeViewportMouseMove(m_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(AzManipulatorTestFrameworkCustomManagerTestFixture, ConsumeViewportMouseMoveActive)
    {
        // consume the mouse down and mouse move events
        m_manipulatorManager->ConsumeViewportMousePress(m_interaction);
        m_manipulatorManager->ConsumeViewportMouseMove(m_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(m_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(AzManipulatorTestFrameworkCustomManagerTestFixture, 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;
            });

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

        // move the mouse along the -x axis
        m_interaction.m_mousePick.m_rayOrigin += expectedPositionAfterMovementAlongAxis;
        m_manipulatorManager->ConsumeViewportMouseMove(m_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(m_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