/*
 * 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 <Tests/ActionManager/ActionManagerFixture.h>
#include <AzToolsFramework/API/ToolsApplicationAPI.h>

#include <QToolBar>

namespace UnitTest
{
    TEST_F(ActionManagerFixture, RegisterToolBar)
    {
        auto outcome = m_toolBarManagerInterface->RegisterToolBar("o3de.toolbar.test", {});
        EXPECT_TRUE(outcome.IsSuccess());
    }

    TEST_F(ActionManagerFixture, RegisterToolBarTwice)
    {
        m_toolBarManagerInterface->RegisterToolBar("o3de.toolbar.test", {});
        auto outcome = m_toolBarManagerInterface->RegisterToolBar("o3de.toolbar.test", {});
        EXPECT_FALSE(outcome.IsSuccess());
    }

    TEST_F(ActionManagerFixture, RegisterToolBarArea)
    {
        auto outcome = m_toolBarManagerInterface->RegisterToolBarArea("o3de.toolbararea.test", m_mainWindow, Qt::ToolBarArea::TopToolBarArea);
        EXPECT_TRUE(outcome.IsSuccess());
    }

    TEST_F(ActionManagerFixture, RegisterToolBarAreaTwice)
    {
        m_toolBarManagerInterface->RegisterToolBarArea("o3de.toolbararea.test", m_mainWindow, Qt::ToolBarArea::TopToolBarArea);
        auto outcome = m_toolBarManagerInterface->RegisterToolBarArea("o3de.toolbararea.test", m_mainWindow, Qt::ToolBarArea::TopToolBarArea);
        EXPECT_FALSE(outcome.IsSuccess());
    }

    TEST_F(ActionManagerFixture, AddActionToUnregisteredToolBar)
    {
        m_actionManagerInterface->RegisterActionContext("o3de.context.test", {});
        m_actionManagerInterface->RegisterAction("o3de.context.test", "o3de.action.test", {}, []{});

        auto outcome = m_toolBarManagerInterface->AddActionToToolBar("o3de.toolbar.test", "o3de.action.test", 42);
        EXPECT_FALSE(outcome.IsSuccess());
    }

    TEST_F(ActionManagerFixture, AddActionToToolBar)
    {
        m_actionManagerInterface->RegisterActionContext("o3de.context.test", {});
        m_actionManagerInterface->RegisterAction("o3de.context.test", "o3de.action.test", {}, []{});
        m_toolBarManagerInterface->RegisterToolBar("o3de.toolbar.test", {});
        
        auto outcome = m_toolBarManagerInterface->AddActionToToolBar("o3de.toolbar.test", "o3de.action.test", 42);
        EXPECT_TRUE(outcome.IsSuccess());
    }

    TEST_F(ActionManagerFixture, AddActionToToolBarTwice)
    {
        m_actionManagerInterface->RegisterActionContext("o3de.context.test", {});
        m_actionManagerInterface->RegisterAction("o3de.context.test", "o3de.action.test", {}, []{});
        m_toolBarManagerInterface->RegisterToolBar("o3de.toolbar.test", {});
        
        m_toolBarManagerInterface->AddActionToToolBar("o3de.toolbar.test", "o3de.action.test", 42);
        auto outcome = m_toolBarManagerInterface->AddActionToToolBar("o3de.toolbar.test", "o3de.action.test", 42);
        EXPECT_FALSE(outcome.IsSuccess());
    }

    TEST_F(ActionManagerFixture, AddActionsToToolBar)
    {
        m_actionManagerInterface->RegisterActionContext("o3de.context.test", {});
        m_actionManagerInterface->RegisterAction("o3de.context.test", "o3de.action.test", {}, []{});
        m_actionManagerInterface->RegisterAction("o3de.context.test", "o3de.action.test2", {}, []{});
        m_toolBarManagerInterface->RegisterToolBar("o3de.toolbar.test", {});
        
        AZStd::vector<AZStd::pair<AZStd::string, int>> actions;
        actions.push_back(AZStd::make_pair("o3de.action.test", 42));
        actions.push_back(AZStd::make_pair("o3de.action.test2", 1));

        auto outcome = m_toolBarManagerInterface->AddActionsToToolBar("o3de.toolbar.test", actions);
        EXPECT_TRUE(outcome.IsSuccess());
    }

    TEST_F(ActionManagerFixture, RemoveActionFromToolBar)
    {
        m_actionManagerInterface->RegisterActionContext("o3de.context.test", {});
        m_actionManagerInterface->RegisterAction("o3de.context.test", "o3de.action.test", {}, []{});
        m_toolBarManagerInterface->RegisterToolBar("o3de.toolbar.test", {});
        
        m_toolBarManagerInterface->AddActionToToolBar("o3de.toolbar.test", "o3de.action.test", 42);
        
        auto outcome = m_toolBarManagerInterface->RemoveActionFromToolBar("o3de.toolbar.test", "o3de.action.test");
        EXPECT_TRUE(outcome.IsSuccess());
    }

    TEST_F(ActionManagerFixture, RemoveMissingActionFromToolBar)
    {
        m_toolBarManagerInterface->RegisterToolBar("o3de.toolbar.test", {});
        
        auto outcome = m_toolBarManagerInterface->RemoveActionFromToolBar("o3de.toolbar.test", "o3de.action.test");
        EXPECT_FALSE(outcome.IsSuccess());
    }

    TEST_F(ActionManagerFixture, RemoveActionsFromToolBar)
    {
        m_actionManagerInterface->RegisterActionContext("o3de.context.test", {});
        m_actionManagerInterface->RegisterAction("o3de.context.test", "o3de.action.test", {}, []{});
        m_actionManagerInterface->RegisterAction("o3de.context.test", "o3de.action.test2", {}, []{});
        m_toolBarManagerInterface->RegisterToolBar("o3de.toolbar.test", {});
        
        AZStd::vector<AZStd::pair<AZStd::string, int>> actions;
        actions.push_back(AZStd::make_pair("o3de.action.test", 42));
        actions.push_back(AZStd::make_pair("o3de.action.test2", 1));

        m_toolBarManagerInterface->AddActionsToToolBar("o3de.toolbar.test", actions);

        auto outcome = m_toolBarManagerInterface->RemoveActionsFromToolBar("o3de.toolbar.test", { "o3de.action.test", "o3de.action.test2" });
        EXPECT_TRUE(outcome.IsSuccess());
    }

    TEST_F(ActionManagerFixture, RemoveMissingActionsFromToolBar)
    {
        m_actionManagerInterface->RegisterActionContext("o3de.context.test", {});
        m_actionManagerInterface->RegisterAction("o3de.context.test", "o3de.action.test", {}, []{});
        m_actionManagerInterface->RegisterAction("o3de.context.test", "o3de.action.test2", {}, []{});
        m_toolBarManagerInterface->RegisterToolBar("o3de.toolbar.test", {});
        
        AZStd::vector<AZStd::pair<AZStd::string, int>> actions;
        actions.push_back(AZStd::make_pair("o3de.action.test", 42));

        m_toolBarManagerInterface->AddActionsToToolBar("o3de.toolbar.test", actions);

        auto outcome = m_toolBarManagerInterface->RemoveActionsFromToolBar("o3de.toolbar.test", { "o3de.action.test", "o3de.action.test2" });
        EXPECT_FALSE(outcome.IsSuccess());
    }

    TEST_F(ActionManagerFixture, GetUnregisteredToolBar)
    {
        QToolBar* toolBar = m_toolBarManagerInterface->GenerateToolBar("o3de.toolbar.test");
        EXPECT_TRUE(toolBar == nullptr);
    }

    TEST_F(ActionManagerFixture, GenerateToolBar)
    {
        m_toolBarManagerInterface->RegisterToolBar("o3de.toolbar.test", {});
        
        QToolBar* toolBar = m_toolBarManagerInterface->GenerateToolBar("o3de.toolbar.test");
        EXPECT_TRUE(toolBar != nullptr);
    }

    TEST_F(ActionManagerFixture, VerifyActionInToolBar)
    {
        // Register ToolBar, get it and verify it's empty.
        m_toolBarManagerInterface->RegisterToolBar("o3de.toolbar.test", {});
        QToolBar* toolBar = m_toolBarManagerInterface->GenerateToolBar("o3de.toolbar.test");
        EXPECT_EQ(toolBar->actions().size(), 0);

        // Register a new action and add it to the ToolBar.
        m_actionManagerInterface->RegisterActionContext("o3de.context.test", {});
        m_actionManagerInterface->RegisterAction("o3de.context.test", "o3de.action.test", {}, []{});
        auto outcome = m_toolBarManagerInterface->AddActionToToolBar("o3de.toolbar.test", "o3de.action.test", 42);

        // Manually trigger ToolBar refresh - Editor will call this once per tick.
        m_toolBarManagerInternalInterface->RefreshToolBars();

        // Verify the action is now in the ToolBar.
        EXPECT_EQ(toolBar->actions().size(), 1);
    }

    TEST_F(ActionManagerFixture, VerifyActionOrderInToolBar)
    {
        // Register ToolBar, get it and verify it's empty.
        m_toolBarManagerInterface->RegisterToolBar("o3de.toolbar.test", {});
        QToolBar* toolBar = m_toolBarManagerInterface->GenerateToolBar("o3de.toolbar.test");
        EXPECT_EQ(toolBar->actions().size(), 0);

        // Register a new action and add it to the ToolBar.
        m_actionManagerInterface->RegisterActionContext("o3de.context.test", {});
        m_actionManagerInterface->RegisterAction("o3de.context.test", "o3de.action.test1", {}, []{});
        m_actionManagerInterface->RegisterAction("o3de.context.test", "o3de.action.test2", {}, []{});
        m_toolBarManagerInterface->AddActionToToolBar("o3de.toolbar.test", "o3de.action.test2", 42);
        m_toolBarManagerInterface->AddActionToToolBar("o3de.toolbar.test", "o3de.action.test1", 1);

        // Manually trigger ToolBar refresh - Editor will call this once per tick.
        m_toolBarManagerInternalInterface->RefreshToolBars();

        // Verify the actions are now in the ToolBar.
        EXPECT_EQ(toolBar->actions().size(), 2);

        // Verify the order is correct.
        QAction* test1 = m_actionManagerInternalInterface->GetAction("o3de.action.test1");
        QAction* test2 = m_actionManagerInternalInterface->GetAction("o3de.action.test2");

        const auto& actions = toolBar->actions();
        EXPECT_EQ(actions[0], test1);
        EXPECT_EQ(actions[1], test2);
    }

    TEST_F(ActionManagerFixture, VerifyActionOrderInToolBarWithCollision)
    {
        // Register ToolBar, get it and verify it's empty.
        m_toolBarManagerInterface->RegisterToolBar("o3de.toolbar.test", {});
        QToolBar* toolBar = m_toolBarManagerInterface->GenerateToolBar("o3de.toolbar.test");
        EXPECT_EQ(toolBar->actions().size(), 0);
        
        // Register a new action and add it to the ToolBar.
        m_actionManagerInterface->RegisterActionContext("o3de.context.test", {});
        m_actionManagerInterface->RegisterAction("o3de.context.test", "o3de.action.test1", {}, []{});
        m_actionManagerInterface->RegisterAction("o3de.context.test", "o3de.action.test2", {}, []{});
        m_toolBarManagerInterface->AddActionToToolBar("o3de.toolbar.test", "o3de.action.test2", 42);
        m_toolBarManagerInterface->AddActionToToolBar("o3de.toolbar.test", "o3de.action.test1", 42);

        // Manually trigger ToolBar refresh - Editor will call this once per tick.
        m_toolBarManagerInternalInterface->RefreshToolBars();

        // Verify the actions are now in the ToolBar.
        EXPECT_EQ(toolBar->actions().size(), 2);

        // Verify the order is correct (when a collision happens, items should be in order of addition).
        QAction* test1 = m_actionManagerInternalInterface->GetAction("o3de.action.test1");
        QAction* test2 = m_actionManagerInternalInterface->GetAction("o3de.action.test2");

        const auto& actions = toolBar->actions();
        EXPECT_EQ(actions[0], test2);
        EXPECT_EQ(actions[1], test1);
    }

    TEST_F(ActionManagerFixture, VerifySeparatorInToolBar)
    {
        // Register ToolBar, get it and verify it's empty.
        m_toolBarManagerInterface->RegisterToolBar("o3de.toolbar.test", {});
        QToolBar* toolBar = m_toolBarManagerInterface->GenerateToolBar("o3de.toolbar.test");
        EXPECT_EQ(toolBar->actions().size(), 0);

        // Add a separator to the ToolBar.
        m_toolBarManagerInterface->AddSeparatorToToolBar("o3de.toolbar.test", 42);

        // Manually trigger ToolBar refresh - Editor will call this once per tick.
        m_toolBarManagerInternalInterface->RefreshToolBars();

        // Verify the separator is now in the ToolBar.
        const auto& actions = toolBar->actions();

        EXPECT_EQ(actions.size(), 1);
        EXPECT_TRUE(actions[0]->isSeparator());
    }

    TEST_F(ActionManagerFixture, AddUnregisteredWidgetInToolBar)
    {
        // Register ToolBar.
        m_toolBarManagerInterface->RegisterToolBar("o3de.toolbar.test", {});

        // Try to add a nullptr widget.
        auto outcome = m_toolBarManagerInterface->AddWidgetToToolBar("o3de.toolbar.test", "someUnregisteredWidgetIdentifier", 42);
        EXPECT_FALSE(outcome.IsSuccess());
    }

    TEST_F(ActionManagerFixture, VerifyWidgetInToolBar)
    {
        // Register ToolBar and widget action.
        m_toolBarManagerInterface->RegisterToolBar("o3de.toolbar.test", {});

        QWidget* widget = new QWidget();
        m_actionManagerInterface->RegisterWidgetAction(
            "o3de.widgetAction.test",
            {},
            [widget]()
            {
                // Note: the WidgetAction generator function should create a new widget every time it's called.
                // This implementation is technically incorrect, but it allows us to test the correct behavior.
                return widget;
            }
        );

        // Add the widget to the ToolBar.
        m_toolBarManagerInterface->AddWidgetToToolBar("o3de.toolbar.test", "o3de.widgetAction.test", 42);

        // Manually trigger ToolBar refresh - Editor will call this once per tick.
        m_toolBarManagerInternalInterface->RefreshToolBars();

        // Verify the separator is now in the ToolBar.
        QToolBar* toolBar = m_toolBarManagerInterface->GenerateToolBar("o3de.toolbar.test");
        const auto& actions = toolBar->actions();

        EXPECT_EQ(actions.size(), 1);

        QWidgetAction* widgetAction = qobject_cast<QWidgetAction*>(actions[0]);
        EXPECT_TRUE(widgetAction != nullptr);
        EXPECT_TRUE(widgetAction->defaultWidget() == widget);
    }

    TEST_F(ActionManagerFixture, VerifyComplexToolBar)
    {
        // Combine multiple actions and separators.
        m_toolBarManagerInterface->RegisterToolBar("o3de.toolbar.test", {});

        m_actionManagerInterface->RegisterActionContext("o3de.context.test", {});
        m_actionManagerInterface->RegisterAction("o3de.context.test", "o3de.action.test1", {}, []{});
        m_actionManagerInterface->RegisterAction("o3de.context.test", "o3de.action.test2", {}, []{});

        // Create a ToolBar with this setup. Order of addition is intentionally scrambled to verify sortKeys.
        // - Test 1 Action
        // - Separator
        // - Test 2 Action
        m_toolBarManagerInterface->AddActionToToolBar("o3de.toolbar.test", "o3de.action.test2", 15);
        m_toolBarManagerInterface->AddActionToToolBar("o3de.toolbar.test", "o3de.action.test1", 1);
        m_toolBarManagerInterface->AddSeparatorToToolBar("o3de.toolbar.test", 10);

        // Verify the actions are now in the ToolBar in the expected order.
        QToolBar* toolBar = m_toolBarManagerInterface->GenerateToolBar("o3de.toolbar.test");
        QAction* test1 = m_actionManagerInternalInterface->GetAction("o3de.action.test1");
        QAction* test2 = m_actionManagerInternalInterface->GetAction("o3de.action.test2");

        // Manually trigger ToolBar refresh - Editor will call this once per tick.
        m_toolBarManagerInternalInterface->RefreshToolBars();

        // Note: separators are still QActions in the context of the ToolBar.
        const auto& actions = toolBar->actions();
        EXPECT_EQ(actions.size(), 3);

        // Verify the order is correct.
        EXPECT_EQ(actions[0], test1);
        EXPECT_TRUE(actions[1]->isSeparator());
        EXPECT_EQ(actions[2], test2);
    }

    TEST_F(ActionManagerFixture, AddToolBarToUnregisteredToolBarArea)
    {
        m_toolBarManagerInterface->RegisterToolBar("o3de.toolbar.test", {});

        auto outcome = m_toolBarManagerInterface->AddToolBarToToolBarArea("o3de.toolbararea.test", "o3de.toolbar.test", 42);
        EXPECT_FALSE(outcome.IsSuccess());
    }

    TEST_F(ActionManagerFixture, AddToolBarToToolBarArea)
    {
        m_toolBarManagerInterface->RegisterToolBarArea("o3de.toolbararea.test", m_mainWindow, Qt::ToolBarArea::TopToolBarArea);
        m_toolBarManagerInterface->RegisterToolBar("o3de.toolbar.test", {});

        auto outcome = m_toolBarManagerInterface->AddToolBarToToolBarArea("o3de.toolbararea.test", "o3de.toolbar.test", 42);
        EXPECT_TRUE(outcome.IsSuccess());
    }

    TEST_F(ActionManagerFixture, AddToolBarToToolBarAreaTwice)
    {
        m_toolBarManagerInterface->RegisterToolBarArea("o3de.toolbararea.test", m_mainWindow, Qt::ToolBarArea::TopToolBarArea);
        m_toolBarManagerInterface->RegisterToolBar("o3de.toolbar.test", {});

        m_toolBarManagerInterface->AddToolBarToToolBarArea("o3de.toolbararea.test", "o3de.toolbar.test", 42);
        auto outcome = m_toolBarManagerInterface->AddToolBarToToolBarArea("o3de.toolbararea.test", "o3de.toolbar.test", 42);
        EXPECT_FALSE(outcome.IsSuccess());
    }

    TEST_F(ActionManagerFixture, VerifyToolBarInToolBarArea)
    {
        const char* TestToolBarName = "Test ToolBar";

        m_toolBarManagerInterface->RegisterToolBarArea("o3de.toolbararea.test", m_mainWindow, Qt::ToolBarArea::TopToolBarArea);
        AzToolsFramework::ToolBarProperties toolBarProperties;
        toolBarProperties.m_name = TestToolBarName;
        m_toolBarManagerInterface->RegisterToolBar("o3de.toolbar.test", toolBarProperties);

        // Add the ToolBar to the toolbar area.
        m_toolBarManagerInterface->AddToolBarToToolBarArea("o3de.toolbararea.test", "o3de.toolbar.test", 42);

        // Manually trigger ToolBar refresh - Editor will call this once per tick.
        m_toolBarManagerInternalInterface->RefreshToolBarAreas();

        // Verify the ToolBar is now in the ToolBar Area.
        auto toolBars = m_mainWindow->findChildren<QToolBar*>("");
        EXPECT_EQ(toolBars.size(), 1);
        EXPECT_EQ(toolBars[0]->windowTitle(), TestToolBarName);
        EXPECT_EQ(m_mainWindow->toolBarArea(toolBars[0]), Qt::ToolBarArea::TopToolBarArea);
    }

    TEST_F(ActionManagerFixture, GetSortKeyOfActionInToolBar)
    {
        m_toolBarManagerInterface->RegisterToolBar("o3de.toolbar.test", {});
        m_actionManagerInterface->RegisterActionContext("o3de.context.test", {});
        m_actionManagerInterface->RegisterAction("o3de.context.test", "o3de.action.test", {}, []{});

        // Add the action to the ToolBar.
        m_toolBarManagerInterface->AddActionToToolBar("o3de.toolbar.test", "o3de.action.test", 42);

        // Verify the API returns the correct sort key.
        auto outcome = m_toolBarManagerInterface->GetSortKeyOfActionInToolBar("o3de.toolbar.test", "o3de.action.test");
        EXPECT_TRUE(outcome.IsSuccess());
        EXPECT_EQ(outcome.GetValue(), 42);
    }

    TEST_F(ActionManagerFixture, GetSortKeyOfUnregisteredActionInToolBar)
    {
        m_toolBarManagerInterface->RegisterToolBar("o3de.toolbar.test", {});

        // Verify the API fails as the action is not registered.
        auto outcome = m_toolBarManagerInterface->GetSortKeyOfActionInToolBar("o3de.toolbar.test", "o3de.action.test");
        EXPECT_FALSE(outcome.IsSuccess());
    }

    TEST_F(ActionManagerFixture, GetSortKeyOfActionNotInToolBar)
    {
        m_toolBarManagerInterface->RegisterToolBar("o3de.toolbar.test", {});
        m_actionManagerInterface->RegisterActionContext("o3de.context.test", {});
        m_actionManagerInterface->RegisterAction("o3de.context.test", "o3de.action.test", {}, []{});

        // Verify the API fails as the action is registered but was not added to the ToolBar.
        auto outcome = m_toolBarManagerInterface->GetSortKeyOfActionInToolBar("o3de.toolbar.test", "o3de.action.test");
        EXPECT_FALSE(outcome.IsSuccess());
    }

    TEST_F(ActionManagerFixture, GetSortKeyOfWidgetInToolBar)
    {
        m_toolBarManagerInterface->RegisterToolBar("o3de.toolbar.test", {});
        m_actionManagerInterface->RegisterWidgetAction(
            "o3de.widgetAction.test",
            {},
            []() -> QWidget*
            {
                return nullptr;
            }
        );

        // Add the widget to the ToolBar.
        m_toolBarManagerInterface->AddWidgetToToolBar("o3de.toolbar.test", "o3de.widgetAction.test", 42);

        // Verify the API returns the correct sort key.
        auto outcome = m_toolBarManagerInterface->GetSortKeyOfWidgetInToolBar("o3de.toolbar.test", "o3de.widgetAction.test");
        EXPECT_TRUE(outcome.IsSuccess());
        EXPECT_EQ(outcome.GetValue(), 42);
    }

    TEST_F(ActionManagerFixture, GetSortKeyOfUnregisteredWidgetInToolBar)
    {
        m_toolBarManagerInterface->RegisterToolBar("o3de.toolbar.test", {});

        // Verify the API fails as the widget is not registered.
        auto outcome = m_toolBarManagerInterface->GetSortKeyOfWidgetInToolBar("o3de.toolbar.test", "o3de.widgetAction.test");
        EXPECT_FALSE(outcome.IsSuccess());
    }

    TEST_F(ActionManagerFixture, GetSortKeyOfWidgetNotInToolBar)
    {
        m_toolBarManagerInterface->RegisterToolBar("o3de.toolbar.test", {});
        m_actionManagerInterface->RegisterWidgetAction(
            "o3de.widgetAction.test",
            {},
            []() -> QWidget*
            {
                return nullptr;
            }
        );

        // Verify the API fails as the widget is registered but was not added to the ToolBar.
        auto outcome = m_toolBarManagerInterface->GetSortKeyOfWidgetInToolBar("o3de.toolbar.test", "o3de.widgetAction.test");
        EXPECT_FALSE(outcome.IsSuccess());
    }

    TEST_F(ActionManagerFixture, VerifyToolBarVisibilityHideWhenDisabled)
    {
        // Register toolbar, get it and verify it's empty.
        m_toolBarManagerInterface->RegisterToolBar("o3de.toolbar.test", {});
        QToolBar* toolBar = m_toolBarManagerInterface->GenerateToolBar("o3de.toolbar.test");
        EXPECT_EQ(toolBar->actions().size(), 0);

        // Register a new action and add it to the ToolBar. Have ToolBarVisibility set to HideWhenDisabled.
        AzToolsFramework::ActionProperties actionProperties;
        actionProperties.m_toolBarVisibility = AzToolsFramework::ActionVisibility::HideWhenDisabled;

        m_actionManagerInterface->RegisterActionContext("o3de.context.test", {});
        m_actionManagerInterface->RegisterAction("o3de.context.test", "o3de.action.test", actionProperties, []{});
        m_toolBarManagerInterface->AddActionToToolBar("o3de.toolbar.test", "o3de.action.test", 42);

        // Add enabled state callback.
        bool enabledState = true;
        m_actionManagerInterface->InstallEnabledStateCallback(
            "o3de.action.test",
            [&]()
            {
                return enabledState;
            }
        );

        // Manually trigger ToolBar refresh - Editor will call this once per tick.
        m_toolBarManagerInternalInterface->RefreshToolBars();

        // Verify the action is now in the ToolBar.
        EXPECT_EQ(toolBar->actions().size(), 1);

        // Set the action as disabled.
        enabledState = false;
        m_actionManagerInterface->UpdateAction("o3de.action.test");

        // Manually trigger ToolBar refresh - Editor will call this once per tick.
        m_toolBarManagerInternalInterface->RefreshToolBars();

        // Verify the action is no longer in the ToolBar.
        EXPECT_EQ(toolBar->actions().size(), 0);
    }

    TEST_F(ActionManagerFixture, VerifyDefaultToolBarVisibility)
    {
        // Register ToolBar, get it and verify it's empty.
        m_toolBarManagerInterface->RegisterToolBar("o3de.toolbar.test", {});
        QToolBar* toolBar = m_toolBarManagerInterface->GenerateToolBar("o3de.toolbar.test");
        EXPECT_EQ(toolBar->actions().size(), 0);

        // Register a new action and add it to the menu. ToolBarVisibility is set to OnlyInActiveMode by default.
        m_actionManagerInterface->RegisterActionContext("o3de.context.test", {});
        m_actionManagerInterface->RegisterAction("o3de.context.test", "o3de.action.test", {}, []{});
        m_toolBarManagerInterface->AddActionToToolBar("o3de.toolbar.test", "o3de.action.test", 42);

        // Add enabled state callback.
        bool enabledState = true;
        m_actionManagerInterface->InstallEnabledStateCallback(
            "o3de.action.test",
            [&]()
            {
                return enabledState;
            }
        );

        // Manually trigger ToolBar refresh - Editor will call this once per tick.
        m_toolBarManagerInternalInterface->RefreshToolBars();

        // Verify the action is now in the ToolBar.
        EXPECT_EQ(toolBar->actions().size(), 1);

        // Set the action as disabled.
        enabledState = false;
        m_actionManagerInterface->UpdateAction("o3de.action.test");

        // Manually trigger ToolBar refresh - Editor will call this once per tick.
        m_toolBarManagerInternalInterface->RefreshToolBars();

        // Verify the action is still in the ToolBar.
        EXPECT_EQ(toolBar->actions().size(), 1);
    }

    TEST_F(ActionManagerFixture, VerifyToolBarVisibilityAlwaysShowWhenChangingMode)
    {
        // Register ToolBar, get it and verify it's empty.
        m_toolBarManagerInterface->RegisterToolBar("o3de.toolbar.test", {});
        QToolBar* toolBar = m_toolBarManagerInterface->GenerateToolBar("o3de.toolbar.test");
        EXPECT_EQ(toolBar->actions().size(), 0);

        // Register a new action and add it to the default mode. Set ToolBarVisibility to AlwaysShow.
        AzToolsFramework::ActionProperties actionProperties;
        actionProperties.m_toolBarVisibility = AzToolsFramework::ActionVisibility::AlwaysShow;

        m_actionManagerInterface->RegisterActionContext("o3de.context.test", {});
        m_actionManagerInterface->RegisterAction("o3de.context.test", "o3de.action.test", actionProperties, []{});
        m_actionManagerInterface->AssignModeToAction(AzToolsFramework::DefaultActionContextModeIdentifier, "o3de.action.test");

        // Add the action to the ToolBar.
        m_toolBarManagerInterface->AddActionToToolBar("o3de.toolbar.test", "o3de.action.test", 42);

        // Manually trigger ToolBar refresh - Editor will call this once per tick.
        m_toolBarManagerInternalInterface->RefreshToolBars();

        // Verify the action is now in the ToolBar.
        EXPECT_EQ(toolBar->actions().size(), 1);

        // Register a new mode and switch to it.
        m_actionManagerInterface->RegisterActionContextMode("o3de.context.test", "testMode");
        m_actionManagerInterface->SetActiveActionContextMode("o3de.context.test", "testMode");

        // Manually trigger ToolBar refresh - Editor will call this once per tick.
        m_toolBarManagerInternalInterface->RefreshToolBars();

        // Verify the action is still in the ToolBar.
        EXPECT_EQ(toolBar->actions().size(), 1);
    }

} // namespace UnitTest