/*
* All or portions of this file Copyright (c) Amazon.com, Inc. or its affiliates or
* its licensors.
*
* For complete copyright and license terms please see the LICENSE at the root of this
* distribution (the "License"). All use of this software is governed by the License,
* or, if provided, by the license below or the license accompanying this file. Do not
* remove or modify any license notices. This file is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*
*/

#include <AzCore/Serialization/SerializeContext.h>
#include <AzTest/AzTest.h>

#include <AzToolsFramework/Application/ToolsApplication.h>
#include <AzToolsFramework/UI/PropertyEditor/EntityPropertyEditor.hxx>
#include <AzToolsFramework/API/EntityPropertyEditorRequestsBus.h>
#include <AzToolsFramework/UI/PropertyEditor/EntityIdQLineEdit.h>

#include <QApplication>

#include <AzToolsFramework/UnitTest/AzToolsFrameworkTestHelpers.h>

namespace UnitTest
{
    using namespace AZ;
    using namespace AzToolsFramework;

    // Test widget to store an EntityIdQLineEdit
    class TestEntityIdParentWidget
        : public QWidget
    {
    public:
        explicit TestEntityIdParentWidget(QWidget* parent = nullptr)
            : QWidget(nullptr)
        {
            AZ_UNUSED(parent);
            // ensure TestWidget can intercept and filter any incoming events itself
            installEventFilter(this);

            m_testLineEdit = new EntityIdQLineEdit(this);
        }

        EntityIdQLineEdit* m_testLineEdit = nullptr;

    };

    class EntityIdQLineEditTests
        : public ToolsApplicationFixture
    {
    };

    TEST_F(EntityIdQLineEditTests, DoubleClickWontSelectInvalidEntity)
    {
        AZ::Entity* entity = aznew AZ::Entity();
        ASSERT_TRUE(entity != nullptr);
        entity->Init();
        entity->Activate();

        AZ::EntityId entityId = entity->GetId();
        ASSERT_TRUE(entityId.IsValid());

        TestEntityIdParentWidget* widget = new TestEntityIdParentWidget(nullptr);
        ASSERT_TRUE(widget != nullptr);
        ASSERT_TRUE(widget->m_testLineEdit != nullptr);

        // Set a valid EntityId
        widget->m_testLineEdit->setFocus();
        widget->m_testLineEdit->SetEntityId(entityId, {});

        // Simulate double click which will cause the EntityId to be set as selected
        QTest::mouseDClick(widget->m_testLineEdit, Qt::LeftButton);

        // If successful we expect the label's entity to be selected.
        EntityIdList selectedEntities;
        ToolsApplicationRequestBus::BroadcastResult(selectedEntities, &ToolsApplicationRequests::GetSelectedEntities);

        EXPECT_EQ(selectedEntities.size(), 1) << "Double clicking on an EntityIdQLabel should only select a single entity";
        EXPECT_TRUE(selectedEntities[0] == entityId) << "The selected entity is not the one that was double clicked";

        selectedEntities.clear();
        ToolsApplicationRequestBus::Broadcast(&ToolsApplicationRequests::SetSelectedEntities, selectedEntities);

        // Now set an invalid EntityId
        widget->m_testLineEdit->SetEntityId(AZ::EntityId(), {});

        // Simulate double clicking again, which should not trigger a selection change since the EntityId is invalid
        QTest::mouseDClick(widget->m_testLineEdit, Qt::LeftButton);

        ToolsApplicationRequestBus::BroadcastResult(selectedEntities, &ToolsApplicationRequests::GetSelectedEntities);
        EXPECT_TRUE(selectedEntities.empty()) << "Double clicking on an EntityIdQLabel with an invalid entity ID shouldn't change anything";

        delete entity;
        delete widget;
    }
}