Move files to AzFramework/Test and AzToolsFramework/Test accordingly

Signed-off-by: Esteban Papp <81431996+amznestebanpapp@users.noreply.github.com>
2021-07-09 08:07:47 -07:00
parent f278bbcd65
commit afd97abdcf
62 changed files with 0 additions and 0 deletions
@@ -0,0 +1,58 @@
+/*
+ * 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;
+        };
+    };
+}
+
+