Files
UnrealEngine/Engine/Source/Programs/AutoRTFMEngineTests/Private/AutomationTestRunner.cpp
2025-05-18 13:04:45 +08:00

97 lines
2.5 KiB
C++

// Copyright Epic Games, Inc. All Rights Reserved.
#include "AutomationTestRunner.h"
#include "Logging/LogVerbosity.h"
#include "Misc/AutomationEvent.h"
#if WITH_AUTOMATION_WORKER
namespace UE::AutoRTFM
{
FAutomationTestRunner::FAutomationTestRunner()
{
}
bool
FAutomationTestRunner::RunTests(const TCHAR* TestFilter)
{
constexpr int32 ExpectedTestCount = 2048;
TArray<FAutomationTestInfo> TestInfos;
TestInfos.Empty(ExpectedTestCount);
FAutomationTestFramework& TestFramework = FAutomationTestFramework::Get();
TestFramework.SetRequestedTestFilter(EAutomationTestFlags::SmokeFilter | EAutomationTestFlags::EngineFilter);
TestFramework.GetValidTestNames(TestInfos);
int TestCount = TestInfos.Num();
if (TestCount <= 0)
{
return true;
}
// Stack walking doesn't work properly on Windows when omitting frame pointers. See WindowsPlatformStackWalk.cpp.
const bool bCaptureStack = TestFramework.GetCaptureStack();
TestFramework.SetCaptureStack(false);
bool AllPassed = true;
const double TestStartTime = FPlatformTime::Seconds();
for (int TestIt = 0; TestIt != TestCount; ++TestIt)
{
const FAutomationTestInfo& TestInfo = TestInfos[TestIt];
const FString& TestFullPath = TestInfo.GetFullTestPath();
if (!(TestInfo.GetTestFlags() & EAutomationTestFlags::SupportsAutoRTFM) && !TestFullPath.Contains("AutoRTFM"))
{
continue;
}
if (TestFilter && !TestFullPath.Contains(TestFilter))
{
continue;
}
constexpr int32 RoleIndex = 0;
TestFramework.StartTestByName(TestInfo.GetTestName(), RoleIndex);
TestFramework.ExecuteLatentCommands();
FAutomationTestExecutionInfo ExecutionInfo;
if (!TestFramework.StopTest(ExecutionInfo))
{
for (const FAutomationExecutionEntry& Entry : ExecutionInfo.GetEntries())
{
switch(Entry.Event.Type)
{
case EAutomationEventType::Info:
UE_LOG(LogAutoRTFMEngineTests, Display, TEXT("%s"), *Entry.Event.Message);
break;
case EAutomationEventType::Warning:
UE_LOG(LogAutoRTFMEngineTests, Warning, TEXT("%s"), *Entry.Event.Message);
break;
case EAutomationEventType::Error:
UE_LOG(LogAutoRTFMEngineTests, Error, TEXT("%s"), *Entry.Event.Message);
break;
}
}
AllPassed = false;
}
}
const double TestEndTime = FPlatformTime::Seconds();
const double TestTime = TestEndTime - TestStartTime;
UE_LOG(LogAutoRTFMEngineTests, Display, TEXT("Tests took %.3lf seconds to execute"), TestTime);
TestFramework.SetCaptureStack(bCaptureStack);
return AllPassed;
}
}
#endif