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.
o3de/Code/Framework/Tests/EntityOwnershipService/EntityOwnershipServiceTestF...

59 lines
2.3 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
*
*/
#pragma once
#include <AzCore/Asset/AssetManager.h>
#include <AzCore/Component/ComponentApplication.h>
#include <AzCore/Memory/PoolAllocator.h>
#include <AzCore/Slice/SliceAssetHandler.h>
#include <AzCore/Slice/SliceComponent.h>
#include <AzCore/UnitTest/TestTypes.h>
#include <AzFramework/Application/Application.h>
#include <AzFramework/Entity/GameEntityContextComponent.h>
#include <AzFramework/Entity/SliceEntityOwnershipService.h>
#include <AzToolsFramework/Application/ToolsApplication.h>
#include <AzToolsFramework/Entity/SliceEditorEntityOwnershipService.h>
namespace UnitTest
{
using EntityList = AZStd::vector<AZ::Entity*>;
class EntityOwnershipServiceTestFixture
: public AllocatorsTestFixture
{
protected:
AzFramework::RootSliceAsset GetRootSliceAsset();
void SetUpEntityOwnershipServiceTest();
void TearDownEntityOwnershipServiceTest();
AzFramework::SliceInstantiationTicket AddSlice(const EntityList& entityList, const bool isAsynchronous = false);
void AddEditorSlice(
AZ::Data::Asset<AZ::SliceAsset>& sliceAsset, const AZ::Transform& worldTransform, const EntityList& entityList);
AzFramework::SliceInstantiationTicket AddSlice(const EntityList& entityList, const bool isAsynchronous,
AZ::Data::Asset<AZ::SliceAsset>& sliceAsset);
void HandleEntitiesAdded(const AzFramework::EntityList& entityList);
void HandleEntitiesRemoved(const AzFramework::EntityIdList& entityIds);
bool ValidateEntities(const AzFramework::EntityList&);
void AddSliceComponentToAsset(AZ::Data::Asset<AZ::SliceAsset>& sliceAsset, const EntityList& entityList);
AZStd::unique_ptr<AzFramework::Application> m_app;
bool m_entitiesAddedCallbackTriggered = false;
bool m_entityRemovedCallbackTriggered = false;
bool m_validateEntitiesCallbackTriggered = false;
bool m_areEntitiesValidForContext = true;
class EntityOwnershipServiceApplication : public AzToolsFramework::ToolsApplication
{
public:
AZ::ComponentTypeList GetRequiredSystemComponents() const override;
};
};
}