o3de/Code/Framework/AzToolsFramework/Tests/Prefab/PrefabDetachPrefabTests.cpp

/*
 * 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 <AzToolsFramework/Entity/EditorEntityHelpers.h>
#include <Prefab/PrefabTestFixture.h>

namespace UnitTest
{
    using PrefabDetachPrefabTests = PrefabTestFixture;

    TEST_F(PrefabDetachPrefabTests, DetachPrefabUnderLevelSucceeds)
    {
        // Level
        // | Car       (prefab)  <-- detach prefab
        //   | Tire
        //     | Belt

        const AZStd::string carPrefabName = "CarPrefab";
        const AZStd::string tireEntityName = "Tire";
        const AZStd::string beltEntityName = "Belt";

        AZ::IO::Path engineRootPath;
        m_settingsRegistryInterface->Get(engineRootPath.Native(), AZ::SettingsRegistryMergeUtils::FilePathKey_EngineRootFolder);
        AZ::IO::Path carPrefabFilepath = engineRootPath / carPrefabName;

        AZ::EntityId tireEntityId = CreateEditorEntityUnderRoot(tireEntityName);
        CreateEditorEntity(beltEntityName, tireEntityId);
        AZ::EntityId carContainerId = CreateEditorPrefab(carPrefabFilepath, { tireEntityId });

        InstanceAlias carInstanceAlias = FindNestedInstanceAliasInInstance(GetRootContainerEntityId(), carPrefabName);

        // Detach the car prefab.
        PrefabOperationResult result = m_prefabPublicInterface->DetachPrefab(carContainerId);
        ASSERT_TRUE(result.IsSuccess());

        PropagateAllTemplateChanges();

        // Validate there is no nested instance in the level prefab instance.
        ValidateNestedInstanceNotUnderInstance(GetRootContainerEntityId(), carInstanceAlias);

        InstanceOptionalReference levelInstance = m_instanceEntityMapperInterface->FindOwningInstance(GetRootContainerEntityId());
        ASSERT_TRUE(levelInstance.has_value());

        // Validate there are three entities in the level prefab instance.
        EXPECT_EQ(levelInstance->get().GetEntityAliasCount(), 3);

        // Validate that the car's parent entity is the level container entity.
        AZStd::string carEntityAliasAfterDetach = FindEntityAliasInInstance(GetRootContainerEntityId(), carPrefabName);
        AZ::EntityId carEntityIdAfterDetach = levelInstance->get().GetEntityId(carEntityAliasAfterDetach);
        EXPECT_TRUE(carEntityIdAfterDetach.IsValid());

        AZ::EntityId parentEntityIdForCar;
        AZ::TransformBus::EventResult(parentEntityIdForCar, carEntityIdAfterDetach, &AZ::TransformInterface::GetParentId);
        EXPECT_EQ(levelInstance->get().GetContainerEntityId(), parentEntityIdForCar);

        // Validate that the tire's parent entity is the car.
        AZStd::string tireEntityAliasAfterDetach = FindEntityAliasInInstance(GetRootContainerEntityId(), tireEntityName);
        AZ::EntityId tireEntityIdAfterDetach = levelInstance->get().GetEntityId(tireEntityAliasAfterDetach);
        EXPECT_TRUE(tireEntityIdAfterDetach.IsValid());

        AZ::EntityId parentEntityIdForTire;
        AZ::TransformBus::EventResult(parentEntityIdForTire, tireEntityIdAfterDetach, &AZ::TransformInterface::GetParentId);
        EXPECT_EQ(carEntityIdAfterDetach, parentEntityIdForTire);

        // Validate that the belt's parent entity is the tire.
        AZStd::string beltEntityAliasAfterDetach = FindEntityAliasInInstance(GetRootContainerEntityId(), beltEntityName);
        AZ::EntityId beltEntityIdAfterDetach = levelInstance->get().GetEntityId(beltEntityAliasAfterDetach);
        EXPECT_TRUE(beltEntityIdAfterDetach.IsValid());

        AZ::EntityId parentEntityIdForBelt;
        AZ::TransformBus::EventResult(parentEntityIdForBelt, beltEntityIdAfterDetach, &AZ::TransformInterface::GetParentId);
        EXPECT_EQ(tireEntityIdAfterDetach, parentEntityIdForBelt);
    }

    TEST_F(PrefabDetachPrefabTests, DetachPrefabUnderParentSucceeds)
    {
        // Level
        // | Garage
        //   | Car       (prefab)  <-- detach prefab
        //     | Tire

        const AZStd::string carPrefabName = "CarPrefab";
        const AZStd::string garageEntityName = "Garage";
        const AZStd::string tireEntityName = "Tire";

        AZ::IO::Path engineRootPath;
        m_settingsRegistryInterface->Get(engineRootPath.Native(), AZ::SettingsRegistryMergeUtils::FilePathKey_EngineRootFolder);
        AZ::IO::Path carPrefabFilepath = engineRootPath / carPrefabName;

        AZ::EntityId garageEntityId = CreateEditorEntityUnderRoot(garageEntityName);
        AZ::EntityId tireEntityId = CreateEditorEntity(tireEntityName, garageEntityId);
        AZ::EntityId carContainerId = CreateEditorPrefab(carPrefabFilepath, { tireEntityId });

        InstanceAlias carInstanceAlias = FindNestedInstanceAliasInInstance(GetRootContainerEntityId(), carPrefabName);

        // Detach the car prefab.
        PrefabOperationResult result = m_prefabPublicInterface->DetachPrefab(carContainerId);
        ASSERT_TRUE(result.IsSuccess());

        PropagateAllTemplateChanges();

        // Validate there is no nested instance in the level prefab instance.
        ValidateNestedInstanceNotUnderInstance(GetRootContainerEntityId(), carInstanceAlias);

        InstanceOptionalReference levelInstance = m_instanceEntityMapperInterface->FindOwningInstance(GetRootContainerEntityId());
        ASSERT_TRUE(levelInstance.has_value());

        // Validate there are three entities in the level prefab instance.
        EXPECT_EQ(levelInstance->get().GetEntityAliasCount(), 3);        

        // Validate that the garage's parent entity is the level container entity.
        AZStd::string garageEntityAliasAfterDetach = FindEntityAliasInInstance(GetRootContainerEntityId(), garageEntityName);
        AZ::EntityId garageEntityIdAfterDetach = levelInstance->get().GetEntityId(garageEntityAliasAfterDetach);
        EXPECT_TRUE(garageEntityIdAfterDetach.IsValid());

        AZ::EntityId parentEntityIdForGarage;
        AZ::TransformBus::EventResult(parentEntityIdForGarage, garageEntityIdAfterDetach, &AZ::TransformInterface::GetParentId);
        EXPECT_EQ(levelInstance->get().GetContainerEntityId(), parentEntityIdForGarage);

        // Validate that the car's parent entity is the garage.
        AZStd::string carEntityAliasAfterDetach = FindEntityAliasInInstance(GetRootContainerEntityId(), carPrefabName);
        AZ::EntityId carEntityIdAfterDetach = levelInstance->get().GetEntityId(carEntityAliasAfterDetach);
        EXPECT_TRUE(carEntityIdAfterDetach.IsValid());

        AZ::EntityId parentEntityIdForCar;
        AZ::TransformBus::EventResult(parentEntityIdForCar, carEntityIdAfterDetach, &AZ::TransformInterface::GetParentId);
        EXPECT_EQ(garageEntityIdAfterDetach, parentEntityIdForCar);

        // Validate that the tire's parent entity is the car.
        AZStd::string tireEntityAliasAfterDetach = FindEntityAliasInInstance(GetRootContainerEntityId(), tireEntityName);
        AZ::EntityId tireEntityIdAfterDetach = levelInstance->get().GetEntityId(tireEntityAliasAfterDetach);
        EXPECT_TRUE(tireEntityIdAfterDetach.IsValid());

        AZ::EntityId parentEntityIdForTire;
        AZ::TransformBus::EventResult(parentEntityIdForTire, tireEntityIdAfterDetach, &AZ::TransformInterface::GetParentId);
        EXPECT_EQ(carEntityIdAfterDetach, parentEntityIdForTire);
    }

    TEST_F(PrefabDetachPrefabTests, DetachPrefabWithNestedPrefabSucceeds)
    {
        // Level
        // | Car       (prefab)  <-- detach prefab
        //   | Wheel   (prefab)
        //     | Tire

        const AZStd::string carPrefabName = "CarPrefab";
        const AZStd::string wheelPrefabName = "WheelPrefab";
        const AZStd::string tireEntityName = "Tire";

        AZ::IO::Path engineRootPath;
        m_settingsRegistryInterface->Get(engineRootPath.Native(), AZ::SettingsRegistryMergeUtils::FilePathKey_EngineRootFolder);
        AZ::IO::Path carPrefabFilepath = engineRootPath / carPrefabName;
        AZ::IO::Path wheelPrefabFilepath = engineRootPath / wheelPrefabName;

        AZ::EntityId tireEntityId = CreateEditorEntityUnderRoot(tireEntityName);
        AZ::EntityId wheelContainerId = CreateEditorPrefab(wheelPrefabFilepath, { tireEntityId });
        EntityAlias tireEntityAlias = FindEntityAliasInInstance(wheelContainerId, tireEntityName);

        AZ::EntityId carContainerId = CreateEditorPrefab(carPrefabFilepath, { wheelContainerId });
        InstanceAlias carInstanceAlias = FindNestedInstanceAliasInInstance(GetRootContainerEntityId(), carPrefabName);

        // Detach the car prefab.
        PrefabOperationResult result = m_prefabPublicInterface->DetachPrefab(carContainerId);
        ASSERT_TRUE(result.IsSuccess());

        PropagateAllTemplateChanges();

        // Validate there is no car instance in the level prefab instance.
        ValidateNestedInstanceNotUnderInstance(GetRootContainerEntityId(), carInstanceAlias);

        // Validate there is wheel instance in the level prefab instance.
        InstanceAlias wheelInstanceAlias = FindNestedInstanceAliasInInstance(GetRootContainerEntityId(), wheelPrefabName);
        ValidateNestedInstanceUnderInstance(GetRootContainerEntityId(), wheelInstanceAlias);

        InstanceOptionalReference levelInstance = m_instanceEntityMapperInterface->FindOwningInstance(GetRootContainerEntityId());
        ASSERT_TRUE(levelInstance.has_value());
        
        AZStd::vector<InstanceOptionalReference> nestedInstances;
        levelInstance->get().GetNestedInstances(
            [&nestedInstances](AZStd::unique_ptr<Instance>& nestedInstance)
            {
                nestedInstances.push_back(*(nestedInstance.get()));
            });

        EXPECT_EQ(nestedInstances.size(), 1) << "There should be only one nested instance in level after detaching.";
        EXPECT_TRUE(nestedInstances[0].has_value());

        // Validate that the car's parent entity is the level container entity.
        AZStd::string carEntityAliasAfterDetach = FindEntityAliasInInstance(GetRootContainerEntityId(), carPrefabName);
        AZ::EntityId carEntityIdAfterDetach = levelInstance->get().GetEntityId(carEntityAliasAfterDetach);
        EXPECT_TRUE(carEntityIdAfterDetach.IsValid());

        AZ::EntityId parentEntityIdForCar;
        AZ::TransformBus::EventResult(parentEntityIdForCar, carEntityIdAfterDetach, &AZ::TransformInterface::GetParentId);
        EXPECT_EQ(levelInstance->get().GetContainerEntityId(), parentEntityIdForCar);

        // Validate that the wheel's parent entity is the car.
        Instance& wheelInstanceAfterDetach = nestedInstances[0]->get();
        AZ::EntityId wheelContainerIdAfterDetach = wheelInstanceAfterDetach.GetContainerEntityId();
        EXPECT_TRUE(wheelContainerIdAfterDetach.IsValid());

        AZ::EntityId parentEntityIdForWheel;
        AZ::TransformBus::EventResult(parentEntityIdForWheel, wheelContainerIdAfterDetach, &AZ::TransformInterface::GetParentId);
        EXPECT_EQ(carEntityIdAfterDetach, parentEntityIdForWheel);

        // Validate that the tire's parent entity is the wheel.
        tireEntityId = wheelInstanceAfterDetach.GetEntityId(tireEntityAlias);
        EXPECT_TRUE(tireEntityId.IsValid());
        AZ::EntityId parentEntityIdForTire;
        AZ::TransformBus::EventResult(parentEntityIdForTire, tireEntityId, &AZ::TransformInterface::GetParentId);
        EXPECT_EQ(wheelContainerIdAfterDetach, parentEntityIdForTire);
    }

    TEST_F(PrefabDetachPrefabTests, DetachPrefabWithNestedPrefabUnderTopLevelEntitySucceeds)
    {
        // Level
        // | Car          (prefab)   <-- detach prefab
        //   | Wheels                <-- top level entity
        //     | Wheel    (prefab)
        //       | Tire

        const AZStd::string carPrefabName = "CarPrefab";
        const AZStd::string wheelPrefabName = "WheelPrefab";

        const AZStd::string wheelsEntityName = "Wheels";
        const AZStd::string tireEntityName = "Tire";

        AZ::IO::Path engineRootPath;
        m_settingsRegistryInterface->Get(engineRootPath.Native(), AZ::SettingsRegistryMergeUtils::FilePathKey_EngineRootFolder);
        AZ::IO::Path carPrefabFilepath = engineRootPath / carPrefabName;
        AZ::IO::Path wheelPrefabFilepath = engineRootPath / wheelPrefabName;

        // Create the wheels and tire entities.
        AZ::EntityId wheelsEntityId = CreateEditorEntityUnderRoot(wheelsEntityName);
        AZ::EntityId tireEntityId = CreateEditorEntity(tireEntityName, wheelsEntityId);

        // Create the wheel prefab.
        AZ::EntityId wheelContainerId = CreateEditorPrefab(wheelPrefabFilepath, { tireEntityId });

        EntityAlias tireEntityAlias = FindEntityAliasInInstance(wheelContainerId, tireEntityName);

        // Create the car prefab.
        AZ::EntityId carContainerId = CreateEditorPrefab(carPrefabFilepath, { wheelsEntityId });
        InstanceAlias carInstanceAlias = FindNestedInstanceAliasInInstance(GetRootContainerEntityId(), carPrefabName);

        // Detach the car prefab.
        PrefabOperationResult result = m_prefabPublicInterface->DetachPrefab(carContainerId);
        ASSERT_TRUE(result.IsSuccess());

        PropagateAllTemplateChanges();

        // Validate there is no car instance in the level prefab instance.
        ValidateNestedInstanceNotUnderInstance(GetRootContainerEntityId(), carInstanceAlias);

        // Validate there is wheels entity in the level prefab instance.
        EntityAlias wheelsEntityAlias = FindEntityAliasInInstance(GetRootContainerEntityId(), wheelsEntityName);
        ValidateEntityUnderInstance(GetRootContainerEntityId(), wheelsEntityAlias, wheelsEntityName);

        // Validate there is wheel instance in the level prefab instance.
        InstanceAlias wheelInstanceAlias = FindNestedInstanceAliasInInstance(GetRootContainerEntityId(), wheelPrefabName);
        ValidateNestedInstanceUnderInstance(GetRootContainerEntityId(), wheelInstanceAlias);

        InstanceOptionalReference levelInstance = m_instanceEntityMapperInterface->FindOwningInstance(GetRootContainerEntityId());
        ASSERT_TRUE(levelInstance.has_value());

        AZStd::vector<InstanceOptionalReference> nestedInstances;
        levelInstance->get().GetNestedInstances(
            [&nestedInstances](AZStd::unique_ptr<Instance>& nestedInstance)
            {
                nestedInstances.push_back(*(nestedInstance.get()));
            });

        EXPECT_EQ(nestedInstances.size(), 1) << "There should be only one nested instance in level after detaching.";
        EXPECT_TRUE(nestedInstances[0].has_value());

        // Validate that the car's parent entity is the level container entity.
        AZStd::string carEntityAliasAfterDetach = FindEntityAliasInInstance(GetRootContainerEntityId(), carPrefabName);
        AZ::EntityId carEntityIdAfterDetach = levelInstance->get().GetEntityId(carEntityAliasAfterDetach);
        EXPECT_TRUE(carEntityIdAfterDetach.IsValid());

        AZ::EntityId parentEntityIdForCar;
        AZ::TransformBus::EventResult(parentEntityIdForCar, carEntityIdAfterDetach, &AZ::TransformInterface::GetParentId);
        EXPECT_EQ(levelInstance->get().GetContainerEntityId(), parentEntityIdForCar);

        // Validate that the wheels' parent entity is the car.
        AZ::EntityId wheelsEntityIdAfterDetach = levelInstance->get().GetEntityId(wheelsEntityAlias);
        EXPECT_TRUE(wheelsEntityIdAfterDetach.IsValid());

        AZ::EntityId parentEntityIdForWheels;
        AZ::TransformBus::EventResult(parentEntityIdForWheels, wheelsEntityIdAfterDetach, &AZ::TransformInterface::GetParentId);
        EXPECT_EQ(carEntityIdAfterDetach, parentEntityIdForWheels);

        // Validate that the wheel prefab's parent entity is the wheels.
        Instance& wheelInstanceAfterDetach = nestedInstances[0]->get();
        AZ::EntityId wheelContainerIdAfterDetach = wheelInstanceAfterDetach.GetContainerEntityId();
        EXPECT_TRUE(wheelContainerIdAfterDetach.IsValid());

        AZ::EntityId parentEntityIdForWheel;
        AZ::TransformBus::EventResult(parentEntityIdForWheel, wheelContainerIdAfterDetach, &AZ::TransformInterface::GetParentId);
        EXPECT_EQ(wheelsEntityIdAfterDetach, parentEntityIdForWheel);
    }

    TEST_F(PrefabDetachPrefabTests, DetachPrefabValidatesDetachedContainerEntityOrder)
    {
        // Validate the detached container entity's sort order in its parent.
        // The detached container entity should not be moved to the beginning or end of the child entity list.
        //
        // Level
        // | Station
        // | Car       (prefab)  <-- detach prefab
        //   | Tire
        // | House

        const AZStd::string carPrefabName = "CarPrefab";

        const AZStd::string tireEntityName = "Tire";
        const AZStd::string stationEntityName = "Station";
        const AZStd::string houseEntityName = "House";

        AZ::IO::Path engineRootPath;
        m_settingsRegistryInterface->Get(engineRootPath.Native(), AZ::SettingsRegistryMergeUtils::FilePathKey_EngineRootFolder);
        AZ::IO::Path carPrefabFilepath = engineRootPath / carPrefabName;

        CreateEditorEntityUnderRoot(stationEntityName);
        AZ::EntityId tireEntityId = CreateEditorEntityUnderRoot(tireEntityName);
        AZ::EntityId carContainerId = CreateEditorPrefab(carPrefabFilepath, { tireEntityId });
        CreateEditorEntityUnderRoot(houseEntityName);

        // Validate child entity order before detaching the car prefab.
        AzToolsFramework::EntityOrderArray entityOrderArrayBeforeDetach =
            AzToolsFramework::GetEntityChildOrder(GetRootContainerEntityId());
        EXPECT_EQ(entityOrderArrayBeforeDetach.size(), 3);

        AZStd::string childEntityName;
        AZ::ComponentApplicationBus::BroadcastResult(
            childEntityName, &AZ::ComponentApplicationRequests::GetEntityName, entityOrderArrayBeforeDetach[0]);
        EXPECT_EQ(childEntityName, stationEntityName);
        AZ::ComponentApplicationBus::BroadcastResult(
            childEntityName, &AZ::ComponentApplicationRequests::GetEntityName, entityOrderArrayBeforeDetach[1]);
        EXPECT_EQ(childEntityName, carPrefabName);
        AZ::ComponentApplicationBus::BroadcastResult(
            childEntityName, &AZ::ComponentApplicationRequests::GetEntityName, entityOrderArrayBeforeDetach[2]);
        EXPECT_EQ(childEntityName, houseEntityName);

        // Detach the car prefab.
        PrefabOperationResult result = m_prefabPublicInterface->DetachPrefab(carContainerId);
        ASSERT_TRUE(result.IsSuccess());

        PropagateAllTemplateChanges();

        // Validate child entity order after detaching the car prefab.
        AzToolsFramework::EntityOrderArray entityOrderArrayAfterDetach =
            AzToolsFramework::GetEntityChildOrder(GetRootContainerEntityId());
        EXPECT_EQ(entityOrderArrayAfterDetach.size(), 3);

        childEntityName = "";
        AZ::ComponentApplicationBus::BroadcastResult(
            childEntityName, &AZ::ComponentApplicationRequests::GetEntityName, entityOrderArrayAfterDetach[0]);
        EXPECT_EQ(childEntityName, stationEntityName);
        AZ::ComponentApplicationBus::BroadcastResult(
            childEntityName, &AZ::ComponentApplicationRequests::GetEntityName, entityOrderArrayAfterDetach[1]);
        EXPECT_EQ(childEntityName, carPrefabName);
        AZ::ComponentApplicationBus::BroadcastResult(
            childEntityName, &AZ::ComponentApplicationRequests::GetEntityName, entityOrderArrayAfterDetach[2]);
        EXPECT_EQ(childEntityName, houseEntityName);
    }

    TEST_F(PrefabDetachPrefabTests, DetachPrefabValidatesDetachedChildEntityOrder)
    {
        // Validate the sort order of top-level child entities.
        //
        // Level
        // | Car       (prefab)  <-- detach prefab
        //   | Engine
        //   | Wheel   (prefab)
        //     | Tire
        //   | Battery

        const AZStd::string carPrefabName = "CarPrefab";
        const AZStd::string wheelPrefabName = "WheelPrefab";

        const AZStd::string tireEntityName = "Tire";
        const AZStd::string engineEntityName = "Engine";
        const AZStd::string batteryEntityName = "Battery";

        AZ::IO::Path engineRootPath;
        m_settingsRegistryInterface->Get(engineRootPath.Native(), AZ::SettingsRegistryMergeUtils::FilePathKey_EngineRootFolder);
        AZ::IO::Path carPrefabFilepath = engineRootPath / carPrefabName;
        AZ::IO::Path wheelPrefabFilepath = engineRootPath / wheelPrefabName;

        AZ::EntityId engineEntityId = CreateEditorEntityUnderRoot(engineEntityName);
        AZ::EntityId tireEntityId = CreateEditorEntityUnderRoot(tireEntityName);
        AZ::EntityId wheelContainerId = CreateEditorPrefab(wheelPrefabFilepath, { tireEntityId });
        AZ::EntityId batteryEntityId = CreateEditorEntityUnderRoot(batteryEntityName);
        AZ::EntityId carContainerId = CreateEditorPrefab(carPrefabFilepath, { engineEntityId, wheelContainerId, batteryEntityId });

        // Validate child entity order under car before detaching the car prefab.
        AzToolsFramework::EntityOrderArray entityOrderArrayBeforeDetach = AzToolsFramework::GetEntityChildOrder(carContainerId);
        EXPECT_EQ(entityOrderArrayBeforeDetach.size(), 3);

        AZStd::string childEntityName;
        AZ::ComponentApplicationBus::BroadcastResult(
            childEntityName, &AZ::ComponentApplicationRequests::GetEntityName, entityOrderArrayBeforeDetach[0]);
        EXPECT_EQ(childEntityName, engineEntityName);
        AZ::ComponentApplicationBus::BroadcastResult(
            childEntityName, &AZ::ComponentApplicationRequests::GetEntityName, entityOrderArrayBeforeDetach[1]);
        EXPECT_EQ(childEntityName, wheelPrefabName);
        AZ::ComponentApplicationBus::BroadcastResult(
            childEntityName, &AZ::ComponentApplicationRequests::GetEntityName, entityOrderArrayBeforeDetach[2]);
        EXPECT_EQ(childEntityName, batteryEntityName);

        // Detach the car prefab.
        PrefabOperationResult result = m_prefabPublicInterface->DetachPrefab(carContainerId);
        ASSERT_TRUE(result.IsSuccess());

        PropagateAllTemplateChanges();

        InstanceOptionalReference levelInstance = m_instanceEntityMapperInterface->FindOwningInstance(GetRootContainerEntityId());
        ASSERT_TRUE(levelInstance.has_value());

        // Validate child entity order under the car after detaching the car prefab.
        EntityAlias carEntityAliasAfterDetach = FindEntityAliasInInstance(levelInstance->get().GetContainerEntityId(), carPrefabName);
        AZ::EntityId carContainerIdAfterDetach = levelInstance->get().GetEntityId(carEntityAliasAfterDetach);
        
        AzToolsFramework::EntityOrderArray entityOrderArrayAfterDetach =
            AzToolsFramework::GetEntityChildOrder(carContainerIdAfterDetach);
        EXPECT_EQ(entityOrderArrayAfterDetach.size(), 3);

        childEntityName = "";
        AZ::ComponentApplicationBus::BroadcastResult(
            childEntityName, &AZ::ComponentApplicationRequests::GetEntityName, entityOrderArrayAfterDetach[0]);
        EXPECT_EQ(childEntityName, engineEntityName);
        AZ::ComponentApplicationBus::BroadcastResult(
            childEntityName, &AZ::ComponentApplicationRequests::GetEntityName, entityOrderArrayAfterDetach[1]);
        EXPECT_EQ(childEntityName, wheelPrefabName);
        AZ::ComponentApplicationBus::BroadcastResult(
            childEntityName, &AZ::ComponentApplicationRequests::GetEntityName, entityOrderArrayAfterDetach[2]);
        EXPECT_EQ(childEntityName, batteryEntityName);
    }

    TEST_F(PrefabDetachPrefabTests, DetachPrefabValidatesTopLevelChildEntityOrder)
    {
        // Validate the sort order of child entities and prefabs that are under the top level entity.
        // 
        // Level
        // | Car          (prefab)   <-- detach prefab
        //   | Wheels                <-- top level entity
        //     | Red_Wheel
        //     | Wheel    (prefab)
        //       | Tire
        //     | Black_Wheel

        const AZStd::string carPrefabName = "CarPrefab";
        const AZStd::string wheelPrefabName = "WheelPrefab";

        const AZStd::string wheelsEntityName = "Wheels";
        const AZStd::string redWheelEntityName = "Red_Wheel";
        const AZStd::string blackWheelEntityName = "Black_Wheel";
        const AZStd::string tireEntityName = "Tire";

        AZ::IO::Path engineRootPath;
        m_settingsRegistryInterface->Get(engineRootPath.Native(), AZ::SettingsRegistryMergeUtils::FilePathKey_EngineRootFolder);
        AZ::IO::Path carPrefabFilepath = engineRootPath / carPrefabName;
        AZ::IO::Path wheelPrefabFilepath = engineRootPath / wheelPrefabName;

        // Create the wheels, red wheel and tire entities.
        AZ::EntityId wheelsEntityId = CreateEditorEntityUnderRoot(wheelsEntityName);
        CreateEditorEntity(redWheelEntityName, wheelsEntityId);
        AZ::EntityId tireEntityId = CreateEditorEntity(tireEntityName, wheelsEntityId);

        // Create the wheel prefab.
        CreateEditorPrefab(wheelPrefabFilepath, { tireEntityId });

        // Create the black wheel entity.
        CreateEditorEntity(blackWheelEntityName, wheelsEntityId);

        // Create the car prefab.
        AZ::EntityId carContainerId = CreateEditorPrefab(carPrefabFilepath, { wheelsEntityId });

        InstanceOptionalReference levelInstance = m_instanceEntityMapperInterface->FindOwningInstance(GetRootContainerEntityId());
        ASSERT_TRUE(levelInstance.has_value());

        // Validate child entity order under wheels before detaching the car prefab.
        EntityAlias wheelsEntityAlias = FindEntityAliasInInstance(carContainerId, wheelsEntityName);
        InstanceOptionalReference carInstance = m_instanceEntityMapperInterface->FindOwningInstance(carContainerId);
        EXPECT_TRUE(carInstance.has_value());
        wheelsEntityId = carInstance->get().GetEntityId(wheelsEntityAlias);

        AzToolsFramework::EntityOrderArray entityOrderArrayBeforeDetach = AzToolsFramework::GetEntityChildOrder(wheelsEntityId);
        EXPECT_EQ(entityOrderArrayBeforeDetach.size(), 3);

        AZStd::string childEntityName;
        AZ::ComponentApplicationBus::BroadcastResult(
            childEntityName, &AZ::ComponentApplicationRequests::GetEntityName, entityOrderArrayBeforeDetach[0]);
        EXPECT_EQ(childEntityName, redWheelEntityName);
        AZ::ComponentApplicationBus::BroadcastResult(
            childEntityName, &AZ::ComponentApplicationRequests::GetEntityName, entityOrderArrayBeforeDetach[1]);
        EXPECT_EQ(childEntityName, wheelPrefabName);
        AZ::ComponentApplicationBus::BroadcastResult(
            childEntityName, &AZ::ComponentApplicationRequests::GetEntityName, entityOrderArrayBeforeDetach[2]);
        EXPECT_EQ(childEntityName, blackWheelEntityName);

        // Detach the car prefab.
        PrefabOperationResult result = m_prefabPublicInterface->DetachPrefab(carContainerId);
        ASSERT_TRUE(result.IsSuccess());

        PropagateAllTemplateChanges();

        // Validate child entity order under wheels after detaching the car prefab.
        wheelsEntityAlias = FindEntityAliasInInstance(levelInstance->get().GetContainerEntityId(), wheelsEntityName);
        wheelsEntityId = levelInstance->get().GetEntityId(wheelsEntityAlias);

        AzToolsFramework::EntityOrderArray entityOrderArrayAfterDetach = AzToolsFramework::GetEntityChildOrder(wheelsEntityId);
        EXPECT_EQ(entityOrderArrayAfterDetach.size(), 3);

        AZ::ComponentApplicationBus::BroadcastResult(
            childEntityName, &AZ::ComponentApplicationRequests::GetEntityName, entityOrderArrayAfterDetach[0]);
        EXPECT_EQ(childEntityName, redWheelEntityName);
        AZ::ComponentApplicationBus::BroadcastResult(
            childEntityName, &AZ::ComponentApplicationRequests::GetEntityName, entityOrderArrayAfterDetach[1]);
        EXPECT_EQ(childEntityName, wheelPrefabName);
        AZ::ComponentApplicationBus::BroadcastResult(
            childEntityName, &AZ::ComponentApplicationRequests::GetEntityName, entityOrderArrayAfterDetach[2]);
        EXPECT_EQ(childEntityName, blackWheelEntityName);
    }
} // namespace UnitTest