/*
 * 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>

namespace UnitTest
{
    using EntityList = AZStd::vector<AZ::Entity*>;

    class EntityOwnershipServiceTestFixture
        : public LeakDetectionFixture
    {
    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;
        };
    };
}