/*
 * 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/Component/ComponentApplication.h>
#include <AzCore/Component/Entity.h>
#include <AzCore/Memory/MemoryComponent.h>
#include <AzCore/RTTI/BehaviorContext.h>
#include <AzCore/Serialization/SerializeContext.h>
#include <AzTest/AzTest.h>
#include <AzToolsFramework/Application/ToolsApplication.h>
#include <AzCore/UnitTest/TestTypes.h>

namespace UnitTest
{
    class GraphModelPythonBindingsFixture
        : public ::testing::Test
    {
    };

    TEST_F(GraphModelPythonBindingsFixture, GraphModelGraphManagerRequests_ApiExists)
    {
        AZ::BehaviorContext* behaviorContext(nullptr);
        AZ::ComponentApplicationBus::BroadcastResult(behaviorContext, &AZ::ComponentApplicationRequests::GetBehaviorContext);
        ASSERT_TRUE(behaviorContext);

        auto graphManagerRequestBus = behaviorContext->m_ebuses.find("GraphManagerRequestBus");
        EXPECT_TRUE(behaviorContext->m_ebuses.end() != graphManagerRequestBus);

        auto* behaviorBus = graphManagerRequestBus->second;
        auto eventIt = behaviorBus->m_events.find("GetGraph");
        EXPECT_TRUE(behaviorBus->m_events.end() != eventIt);
    }

    TEST_F(GraphModelPythonBindingsFixture, GraphModelGraphControllerRequests_ApiExists)
    {
        AZ::BehaviorContext* behaviorContext(nullptr);
        AZ::ComponentApplicationBus::BroadcastResult(behaviorContext, &AZ::ComponentApplicationRequests::GetBehaviorContext);
        ASSERT_TRUE(behaviorContext);

        auto graphControllerRequestBus = behaviorContext->m_ebuses.find("GraphControllerRequestBus");
        EXPECT_TRUE(behaviorContext->m_ebuses.end() != graphControllerRequestBus);

        const AZStd::vector<AZStd::string> eventNames = {
            "AddNode",
            "RemoveNode",
            "AddConnection",
            "AddConnectionBySlotId",
            "RemoveConnection"
        };
        auto* behaviorBus = graphControllerRequestBus->second;
        for (const AZStd::string& name : eventNames)
        {
            auto eventIt = behaviorBus->m_events.find(name);
            EXPECT_TRUE(behaviorBus->m_events.end() != eventIt);
        }
    }
}