/*
* All or portions of this file Copyright (c) Amazon.com, Inc. or its affiliates or
* its licensors.
*
* For complete copyright EntityRef 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.
*
*/

#pragma once

#include <AzTest/AzTest.h>

namespace ScriptCanvasTests
{
    class EntityRefTestEvents : public AZ::EBusTraits
    {
    public:
        static const AZ::EBusAddressPolicy AddressPolicy = AZ::EBusAddressPolicy::ById;
        using BusIdType = AZ::EntityId;

        virtual void TestEvent(AZ::EntityId, bool) = 0;
    };
    using EntityRefTestEventBus = AZ::EBus<EntityRefTestEvents>;


    class TestComponent : public AZ::Component
        , EntityRefTestEventBus::Handler
    {
    public:
        AZ_COMPONENT(TestComponent, "{527680AE-BF46-4BC8-A923-A39B458A3B53}", AZ::Component);

        void Init() override {}
        void Activate() override
        {
            EntityRefTestEventBus::Handler::BusConnect(GetEntityId());
        }
        void Deactivate() override
        {
            EntityRefTestEventBus::Handler::BusDisconnect();
        }

        void TestEvent(AZ::EntityId referencedEntity, bool shouldBeSelf) override
        {
            // validate entity
            if (shouldBeSelf)
            {
                EXPECT_TRUE(GetEntity()->GetId() == referencedEntity);
            }

            AZ_TracePrintf("Script Canvas", "TestEvent handled by: %s\n", GetEntity()->GetName().c_str());
        }

        static void Reflect(AZ::ReflectContext* reflection)
        {
            AZ::SerializeContext* serializeContext = azrtti_cast<AZ::SerializeContext*>(reflection);
            if (serializeContext)
            {
                serializeContext->Class<TestComponent, AZ::Component>()
                    ->Version(0)
                    ;
            }

            AZ::BehaviorContext* behaviorContext = azrtti_cast<AZ::BehaviorContext*>(reflection);
            if (behaviorContext)
            {
                behaviorContext->EBus<EntityRefTestEventBus>("EntityRefTestEventBus")
                    ->Event("TestEvent", &EntityRefTestEvents::TestEvent)
                    ;
            }
        }
    };
}