You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
87 lines
4.2 KiB
C++
87 lines
4.2 KiB
C++
/*
|
|
* Copyright (c) Contributors to the Open 3D Engine Project.
|
|
* For complete copyright and license terms please see the LICENSE at the root of this distribution.
|
|
*
|
|
* SPDX-License-Identifier: Apache-2.0 OR MIT
|
|
*
|
|
*/
|
|
|
|
#include <gtest/gtest.h>
|
|
|
|
#include <Tests/UI/UIFixture.h>
|
|
#include <EMotionFX/CommandSystem/Source/CommandManager.h>
|
|
#include <EMotionFX/Source/AnimGraphManager.h>
|
|
#include <EMotionFX/Source/AnimGraphMotionNode.h>
|
|
#include <EMotionFX/Source/AnimGraphStateMachine.h>
|
|
#include <EMotionStudio/Plugins/StandardPlugins/Source/AnimGraph/AnimGraphPlugin.h>
|
|
#include <EMotionStudio/Plugins/StandardPlugins/Source/AnimGraph/BlendGraphViewWidget.h>
|
|
#include <QApplication>
|
|
#include <QtTest>
|
|
#include "qtestsystem.h"
|
|
|
|
namespace EMotionFX
|
|
{
|
|
class RemoveTransitionFixture
|
|
: public UIFixture
|
|
{
|
|
public:
|
|
void SetUp() override
|
|
{
|
|
UIFixture::SetUp();
|
|
AZStd::string commandResult;
|
|
|
|
MCore::CommandGroup group;
|
|
|
|
// Create empty anim graph, add a motion node and a blend tree.
|
|
group.AddCommandString(AZStd::string::format("CreateAnimGraph -animGraphID %d", m_animGraphId));
|
|
group.AddCommandString(AZStd::string::format("AnimGraphCreateNode -animGraphID %d -type %s -parentName Root -xPos 100 -yPos 100 -name testMotion",
|
|
m_animGraphId, azrtti_typeid<AnimGraphMotionNode>().ToString<AZStd::string>().c_str()));
|
|
group.AddCommandString(AZStd::string::format("AnimGraphCreateNode -animGraphID %d -type %s -parentName Root -xPos 200 -yPos 100 -name testBlendTree",
|
|
m_animGraphId, azrtti_typeid<BlendTree>().ToString<AZStd::string>().c_str()));
|
|
group.AddCommandString(AZStd::string::format("AnimGraphCreateConnection -animGraphID %d -transitionType %s -sourceNode testMotion -targetNode testBlendTree",
|
|
m_animGraphId, azrtti_typeid<AnimGraphStateTransition>().ToString<AZStd::string>().c_str()));
|
|
|
|
EXPECT_TRUE(CommandSystem::GetCommandManager()->ExecuteCommandGroup(group, commandResult)) << commandResult.c_str();
|
|
m_animGraph = GetAnimGraphManager().FindAnimGraphByID(m_animGraphId);
|
|
EXPECT_NE(m_animGraph, nullptr) << "Cannot find newly created anim graph.";
|
|
}
|
|
|
|
void TearDown() override
|
|
{
|
|
QApplication::processEvents(QEventLoop::ExcludeUserInputEvents);
|
|
delete m_animGraph;
|
|
UIFixture::TearDown();
|
|
}
|
|
|
|
public:
|
|
const AZ::u32 m_animGraphId = 64;
|
|
AnimGraph* m_animGraph = nullptr;
|
|
};
|
|
|
|
TEST_F(RemoveTransitionFixture, RemoveTransition)
|
|
{
|
|
RecordProperty("test_case_id", "C15031141");
|
|
|
|
auto animGraphPlugin = static_cast<EMStudio::AnimGraphPlugin*>(EMStudio::GetPluginManager()->FindActivePlugin(EMStudio::AnimGraphPlugin::CLASS_ID));
|
|
ASSERT_TRUE(animGraphPlugin) << "Anim graph plugin not found.";
|
|
|
|
EMStudio::AnimGraphModel& animGraphModel = animGraphPlugin->GetAnimGraphModel();
|
|
|
|
// Find the transition between the motion node and the blend tree.
|
|
AnimGraphStateTransition* transition = m_animGraph->GetRootStateMachine()->GetTransition(0);
|
|
ASSERT_TRUE(transition) << "Anim graph transition not found.";
|
|
|
|
// Select the transition in the anim graph model.
|
|
const QModelIndex& modelIndex = animGraphModel.FindFirstModelIndex(transition);
|
|
const EMStudio::AnimGraphModel::ModelItemType itemType = modelIndex.data(EMStudio::AnimGraphModel::ROLE_MODEL_ITEM_TYPE).value<EMStudio::AnimGraphModel::ModelItemType>();
|
|
ASSERT_TRUE(modelIndex.isValid()) << "Anim graph transition has an invalid model index.";
|
|
animGraphModel.GetSelectionModel().select(QItemSelection(modelIndex, modelIndex), QItemSelectionModel::Current | QItemSelectionModel::ClearAndSelect | QItemSelectionModel::Rows);
|
|
|
|
EMStudio::BlendGraphViewWidget* blendGraphViewWidget = animGraphPlugin->GetViewWidget();
|
|
blendGraphViewWidget->GetAction(EMStudio::BlendGraphViewWidget::EDIT_DELETE)->trigger();
|
|
|
|
// Check if the transition get deleted.
|
|
ASSERT_EQ(0, m_animGraph->GetRootStateMachine()->GetNumTransitions()) << " Anim Graph transition should be removed";
|
|
}
|
|
} // namespace EMotionFX
|