/*
 * 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 <AzTest/AzTest.h>
#include <AzCore/Memory/SystemAllocator.h>
#include <AzQtComponents/Utilities/QtPluginPaths.h>
#include <QApplication>

namespace UnitTest
{
    // Handle asserts
    class ToolsFrameworkHook
        : public AZ::Test::ITestEnvironment
    {
    public:
        void SetupEnvironment() override
        {
            AZ::AllocatorInstance<AZ::SystemAllocator>::Create();
        }

        void TeardownEnvironment() override
        {
            AZ::AllocatorInstance<AZ::SystemAllocator>::Destroy();
        }
    };

    AZTEST_EXPORT int AZ_UNIT_TEST_HOOK_NAME(int argc, char** argv)
    {
        ::testing::InitGoogleMock(&argc, argv);
        AzQtComponents::PrepareQtPaths();
        QApplication app(argc, argv);
        AZ::Test::excludeIntegTests();
        AZ::Test::printUnusedParametersWarning(argc, argv);
        AZ::Test::addTestEnvironments({ new ToolsFrameworkHook });
        int result = RUN_ALL_TESTS();
        return result;
    }
} // namespace UnitTest