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.
123 lines
4.1 KiB
C++
123 lines
4.1 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 <AzTest/AzTest.h>
|
|
#include <AzCore/UnitTest/TestTypes.h>
|
|
#include <AzCore/Component/ComponentApplication.h>
|
|
#include <AzCore/RTTI/BehaviorContext.h>
|
|
|
|
#include <EditorPythonBindings/EditorPythonBindingsSymbols.h>
|
|
|
|
#include "Source/PythonAssetBuilderSystemComponent.h"
|
|
#include <PythonAssetBuilder/PythonAssetBuilderBus.h>
|
|
#include "PythonBuilderTestShared.h"
|
|
|
|
namespace UnitTest
|
|
{
|
|
class PythonBuilderProcessJobTest
|
|
: public ScopedAllocatorSetupFixture
|
|
{
|
|
protected:
|
|
AZStd::unique_ptr<AZ::ComponentApplication> m_app;
|
|
AZ::Entity* m_systemEntity = nullptr;
|
|
|
|
void SetUp() override
|
|
{
|
|
AZ::ComponentApplication::Descriptor appDesc;
|
|
m_app = AZStd::make_unique<AZ::ComponentApplication>();
|
|
m_systemEntity = m_app->Create(appDesc);
|
|
}
|
|
|
|
void TearDown() override
|
|
{
|
|
m_app.reset();
|
|
}
|
|
};
|
|
|
|
TEST_F(PythonBuilderProcessJobTest, PythonBuilder_ProcessJob_ResultSuccess)
|
|
{
|
|
using namespace PythonAssetBuilder;
|
|
using namespace AssetBuilderSDK;
|
|
|
|
const AZ::Uuid builderId = RegisterAssetBuilder(m_app.get(), m_systemEntity);
|
|
|
|
MockJobHandler mockJobHandler;
|
|
mockJobHandler.BusConnect(builderId);
|
|
|
|
AssetBuilderSDK::ProcessJobRequest request;
|
|
request.m_builderGuid = builderId;
|
|
request.m_sourceFileUUID = AZ::Uuid::CreateRandom();
|
|
|
|
AssetBuilderSDK::ProcessJobResponse response;
|
|
response.m_resultCode = AssetBuilderSDK::ProcessJobResult_NetworkIssue;
|
|
|
|
PythonBuilderNotificationBus::EventResult(
|
|
response,
|
|
builderId,
|
|
&PythonBuilderNotificationBus::Events::OnProcessJobRequest,
|
|
request);
|
|
|
|
EXPECT_EQ(AssetBuilderSDK::ProcessJobResult_Success, response.m_resultCode);
|
|
EXPECT_EQ(0, mockJobHandler.m_onCancelCount);
|
|
}
|
|
|
|
TEST_F(PythonBuilderProcessJobTest, PythonBuilder_ProcessJob_ResultFailed)
|
|
{
|
|
using namespace PythonAssetBuilder;
|
|
using namespace AssetBuilderSDK;
|
|
|
|
const AZ::Uuid builderId = RegisterAssetBuilder(m_app.get(), m_systemEntity);
|
|
|
|
MockJobHandler mockJobHandler;
|
|
mockJobHandler.BusConnect(builderId);
|
|
|
|
AssetBuilderSDK::ProcessJobRequest request;
|
|
request.m_builderGuid = builderId;
|
|
request.m_sourceFileUUID = AZ::Uuid::CreateNull();
|
|
|
|
AssetBuilderSDK::ProcessJobResponse response;
|
|
response.m_resultCode = AssetBuilderSDK::ProcessJobResult_Success;
|
|
|
|
PythonBuilderNotificationBus::EventResult(
|
|
response,
|
|
builderId,
|
|
&PythonBuilderNotificationBus::Events::OnProcessJobRequest,
|
|
request);
|
|
|
|
EXPECT_EQ(AssetBuilderSDK::ProcessJobResult_Failed, response.m_resultCode);
|
|
EXPECT_EQ(0, mockJobHandler.m_onCancelCount);
|
|
}
|
|
|
|
TEST_F(PythonBuilderProcessJobTest, PythonBuilder_ProcessJob_OnCancel)
|
|
{
|
|
using namespace PythonAssetBuilder;
|
|
using namespace AssetBuilderSDK;
|
|
|
|
const AZ::Uuid builderId = RegisterAssetBuilder(m_app.get(), m_systemEntity);
|
|
|
|
MockJobHandler mockJobHandler;
|
|
mockJobHandler.BusConnect(builderId);
|
|
|
|
PythonBuilderNotificationBus::Event(builderId, &PythonBuilderNotificationBus::Events::OnCancel);
|
|
EXPECT_EQ(1, mockJobHandler.m_onCancelCount);
|
|
}
|
|
|
|
TEST_F(PythonBuilderProcessJobTest, PythonBuilderRequestBus_Behavior_Exists)
|
|
{
|
|
using namespace PythonAssetBuilder;
|
|
using namespace AssetBuilderSDK;
|
|
|
|
RegisterAssetBuilder(m_app.get(), m_systemEntity);
|
|
|
|
auto entry = m_app->GetBehaviorContext()->m_ebuses.find("PythonBuilderRequestBus");
|
|
ASSERT_NE(m_app->GetBehaviorContext()->m_ebuses.end(), entry);
|
|
EXPECT_NE(entry->second->m_events.end(), entry->second->m_events.find("WriteSliceFile"));
|
|
EXPECT_NE(entry->second->m_events.end(), entry->second->m_events.find("CreateEditorEntity"));
|
|
}
|
|
}
|