o3de/Gems/ScriptCanvasDeveloper/Code/Editor/Source/EditorAutomationTests/GraphCreationTests.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 "precompiled.h"

#include <qpushbutton.h>

#include <GraphCanvas/Components/GridBus.h>
#include <GraphCanvas/Components/Nodes/Group/NodeGroupBus.h>

#include <Editor/Source/EditorAutomationTests/GraphCreationTests.h>

#include <Editor/GraphCanvas/GraphCanvasEditorNotificationBusId.h>

#include <ScriptCanvas/GraphCanvas/MappingBus.h>
#include <ScriptCanvas/Core/Node.h>
#include <ScriptCanvas/Core/NodeBus.h>
#include <ScriptCanvas/Bus/EditorScriptCanvasBus.h>

#include <ScriptCanvasDeveloperEditor/EditorAutomation/EditorAutomationStates/GraphStates.h>

namespace ScriptCanvasDeveloper
{
    ///////////////////////////
    // CreateGraphHotKeyState
    ///////////////////////////
#if !defined(AZ_COMPILER_MSVC)
#ifndef VK_CONTROL
#define VK_CONTROL 0x11
#endif 
#endif

    CreateGraphTest::CreateGraphHotKeyState::CreateGraphHotKeyState()
        : m_pressControl(VK_CONTROL)
        , m_releaseControl(VK_CONTROL)
        , m_typeN(QChar('n'))
        , m_longProcessEvents(AZStd::chrono::milliseconds(1000))
    {
    }

    void CreateGraphTest::CreateGraphHotKeyState::OnActiveGraphChanged(const AZ::EntityId& graphCanvasId)
    {
        m_hotKeyGraphId = graphCanvasId;
    }

    void CreateGraphTest::CreateGraphHotKeyState::OnSetupStateActions(EditorAutomationActionRunner& actionRunner)
    {
        GraphCanvas::AssetEditorNotificationBus::Handler::BusConnect(ScriptCanvasEditor::AssetEditorId);

        actionRunner.AddAction(&m_pressControl);
        actionRunner.AddAction(&m_shortProcessEvents);
        actionRunner.AddAction(&m_typeN);
        actionRunner.AddAction(&m_longProcessEvents);
        actionRunner.AddAction(&m_releaseControl);
        actionRunner.AddAction(&m_shortProcessEvents);

        m_hotKeyGraphId.SetInvalid();
    }

    void CreateGraphTest::CreateGraphHotKeyState::OnStateActionsComplete()
    {
        if (!m_hotKeyGraphId.IsValid())
        {
            ReportError("Failed to create graph using hot key");
        }
        else
        {
            GraphCanvas::GraphId activeGraphCanvasId;
            ScriptCanvasEditor::GeneralRequestBus::BroadcastResult(activeGraphCanvasId, &ScriptCanvasEditor::GeneralRequests::GetActiveGraphCanvasGraphId);

            if (activeGraphCanvasId != m_hotKeyGraphId)
            {
                ReportError("Active graph is not the newly created graph using hotkey.");
            }
            else
            {
                ScriptCanvas::ScriptCanvasId scriptCanvasId;
                ScriptCanvasEditor::GeneralRequestBus::BroadcastResult(scriptCanvasId, &ScriptCanvasEditor::GeneralRequests::GetScriptCanvasId, activeGraphCanvasId);

                bool isRuntimeGraph = false;
                ScriptCanvasEditor::EditorGraphRequestBus::EventResult(isRuntimeGraph, scriptCanvasId, &ScriptCanvasEditor::EditorGraphRequests::IsRuntimeGraph);

                if (!isRuntimeGraph)
                {
                    ReportError("Failed to create Runtime graph using button");
                }
            }
        }

        GraphCanvas::AssetEditorNotificationBus::Handler::BusDisconnect();
    }

    ////////////////////
    // CreateGraphTest
    ////////////////////

    CreateGraphTest::CreateGraphTest()
        : EditorAutomationTest("Create Graph Test")
    {
        SetHasCustomTransitions(true);

        AddState(new CreateRuntimeGraphState());
        AddState(new CreateGraphHotKeyState());
        AddState(new ForceCloseActiveGraphState());

        SetInitialStateId<CreateRuntimeGraphStateId>();
    }

    void CreateGraphTest::OnTestStarting()
    {
        m_creationState = CreateRuntimeGraphStateId::StateID();
    }

    int CreateGraphTest::EvaluateTransition(int stateId)
    {
        if (stateId == CreateRuntimeGraphState::StateID()
            || stateId == CreateGraphHotKeyState::StateID())
        {
            return ForceCloseActiveGraphState::StateID();
        }
        else if (stateId == ForceCloseActiveGraphState::StateID())
        {
            if (m_creationState == CreateRuntimeGraphStateId::StateID())
            {
                m_creationState = CreateGraphHotKeyState::StateID();
                return m_creationState;
            }
        }

        return EditorAutomationState::EXIT_STATE_ID;
    }

    ///////////////////////
    // CreateFunctionTest
    ///////////////////////

    CreateFunctionTest::CreateFunctionTest()
        : EditorAutomationTest("Create Function Test")
    {
        AddState(new CreateFunctionGraphState());
        AddState(new ForceCloseActiveGraphState());
    }
}