o3de/AutomatedTesting/Gem/PythonTests/editor/EditorScripts/EntityOutliner_EntityOrdering.py

"""
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
"""


class Tests:
    entities_sorted = (
        "Entities sorted in the expected order",
        "Entities sorted in an incorrect order",
    )


def EntityOutliner_EntityOrdering():
    """
    Summary:
    Verify that manual entity ordering in the entity outliner works and is stable.

    Expected Behavior:
    Several entities are created, some are manually ordered, and their order
    is maintained, even when new entities are added.

    Test Steps:
    1) Open the empty Prefab Base level
    2) Add 5 entities to the outliner
    3) Move "Entity1" to the top of the order
    4) Move "Entity4" to the bottom of the order
    5) Add another new entity, ensure the rest of the order is unchanged
    """

    from PySide2 import QtCore
    
    import editor_python_test_tools.hydra_editor_utils as hydra
    import pyside_utils
    from editor_python_test_tools.utils import Report
    from editor_python_test_tools.wait_utils import PrefabWaiter

    # Grab the Editor, Entity Outliner, and Outliner Model
    editor_window = pyside_utils.get_editor_main_window()
    entity_outliner = pyside_utils.find_child_by_hierarchy(
        editor_window, ..., "EntityOutlinerWidgetUI", ..., "m_objectTree"
    )
    entity_outliner_model = entity_outliner.model()

    # Get the outliner index for the root prefab container entity
    def get_root_prefab_container_index():
        return entity_outliner_model.index(0, 0)

    # Get the outliner index for the top level entity of a given name
    def index_for_name(name):
        root_index = get_root_prefab_container_index()
        for row in range(entity_outliner_model.rowCount(root_index)):
            row_index = entity_outliner_model.index(row, 0, root_index)
            if row_index.data() == name:
                return row_index
        return None

    # Validate that the outliner top level entity order matches the expected order
    def verify_entities_sorted(expected_order):
        actual_order = []
        root_index = get_root_prefab_container_index()
        for row in range(entity_outliner_model.rowCount(root_index)):
            row_index = entity_outliner_model.index(row, 0, root_index)
            actual_order.append(row_index.data())

        sorted_correctly = actual_order == expected_order
        Report.result(Tests.entities_sorted, sorted_correctly)
        if not sorted_correctly:
            print(f"Expected entity order: {expected_order}")
            print(f"Actual entity order: {actual_order}")

    # Creates an entity from the outliner context menu
    def create_entity():
        # Create entity
        pyside_utils.trigger_context_menu_entry(
            entity_outliner, "Create entity", index=get_root_prefab_container_index()
        )
        # Wait a tick after entity creation to let events process
        PrefabWaiter.wait_for_propagation()

    # Moves an entity (wrapped by move_entity_before and move_entity_after)
    def _move_entity(source_name, target_name, move_after=False):
        source_index = index_for_name(source_name)
        target_index = index_for_name(target_name)
        if source_index is None:
            print(f"Failed to retrieve index for {source_name}")
            return
        if target_index is None:
            print(f"Failed to retrieve index for {target_name}")
            return

        target_row = target_index.row()
        if move_after:
            target_row += 1

        # Generate MIME data and directly inject it into the model instead of
        # generating mouse click operations, as it's more reliable and we're
        # testing the underlying drag & drop logic as opposed to Qt's mouse
        # handling here
        mime_data = entity_outliner_model.mimeData([source_index])
        entity_outliner_model.dropMimeData(
            mime_data, QtCore.Qt.MoveAction, target_row, 0, target_index.parent()
        )
        # Wait after move to let events (i.e. prefab propagation) process
        PrefabWaiter.wait_for_propagation()

    # Move an entity before another entity in the order by dragging the source above the target
    move_entity_before = lambda source_name, target_name: _move_entity(
        source_name, target_name, move_after=False
    )
    # Move an entity after another entity in the order by dragging the source below the target
    move_entity_after = lambda source_name, target_name: _move_entity(
        source_name, target_name, move_after=True
    )

    expected_order = []

    # 1) Open the empty Prefab Base level
    hydra.open_base_level()

    # 2) Add 5 entities to the outliner
    ENTITIES_TO_ADD = 5
    for i in range(ENTITIES_TO_ADD):
        create_entity()

        # Our new entity should be given a name with a number automatically
        new_entity = f"Entity{i+1}"
        # The new entity should be added to the bottom of its parent entity
        expected_order = expected_order + [new_entity]

        verify_entities_sorted(expected_order)

    # 3) Move "Entity5" to the top of the order
    move_entity_before("Entity5", "Entity1")
    expected_order = ["Entity5", "Entity1", "Entity2", "Entity3", "Entity4"]
    verify_entities_sorted(expected_order)

    # 4) Move "Entity2" to the bottom of the order
    move_entity_after("Entity2", "Entity4")
    expected_order = ["Entity5", "Entity1", "Entity3", "Entity4", "Entity2"]
    verify_entities_sorted(expected_order)

    # 5) Add another new entity, ensure the rest of the order is unchanged
    create_entity()
    expected_order = ["Entity5", "Entity1", "Entity3", "Entity4", "Entity2", "Entity6"]
    verify_entities_sorted(expected_order)


if __name__ == "__main__":
    from editor_python_test_tools.utils import Report

    Report.start_test(EntityOutliner_EntityOrdering)