TestManager.h

#pragma once
 
#include <LkEngine/Core/CoreTypes.h>
#include <LkEngine/Core/CoreMacros.h>
#include <LkEngine/Core/LObject/Object.h>
#include <LkEngine/Core/LObject/ObjectPtr.h>
 
#include "LTesto/Core/AutomationTest.h"
 
namespace LkEngine {
 
    class LTestManager
    {
    public:
        LTestManager() = default;
        ~LTestManager() = default;
 
        static LTestManager& Get();
 
        CORE_API bool RegisterAutomationTest(const std::string& TestName, Test::LAutomationTest* InTestInstance);
 
        CORE_API bool UnregisterAutomationTest(const std::string& TestName, Test::LAutomationTest* InTestInstance);
 
        CORE_API void RunTests(const Test::ETestSuite Suite = Test::ETestSuite::All) const;
 
        CORE_API bool StartTest(const std::string& TestName);
 
        CORE_API bool StopTest(const std::string& TestName);
 
        Test::LAutomationTest* GetCurrentTest() const
        {
            return CurrentTest;
        }
 
        FORCEINLINE CORE_API void SetCaptureStack(const bool Enabled)
        {
            bCaptureStack = Enabled;
        }
 
        using FTestCreator = std::function<std::shared_ptr<Test::LAutomationTest>()>;
 
        void AddToSuite(const Test::ETestSuite Suite, FTestCreator CreatorFunc)
        {
            /* TODO: Cannot use log macros here because the registration takes place before the logger is instantiated.
             *       Needs to be fixed. */
            //LK_PRINTLN("[TestManager] Registering test {} to suite '{}'", TestCreatorMap[Suite].size() + 1, Enum::ToString(Suite));
            TestCreatorMap[Suite].push_back(std::move(CreatorFunc));
        }
 
        const std::vector<FTestCreator>& GetTests(const Test::ETestSuite Suite) const
        {
            static const std::vector<FTestCreator> EmptySuite;
            auto Iter = TestCreatorMap.find(Suite);
            return (Iter != TestCreatorMap.end()) ? Iter->second : EmptySuite;
        }
 
    public:
        std::unordered_map<std::string, Test::LAutomationTest*> TestInstanceMap{};
 
        std::unordered_map<Test::ETestSuite, std::vector<FTestCreator>> TestCreatorMap;
 
    private:
        CORE_API static bool bCaptureStack;
 
        Test::LAutomationTest* CurrentTest = nullptr;
    };
 
}