Files
UnrealEngine/Engine/Plugins/Runtime/StateTree/Source/StateTreeTestSuite/Private/StateTreeTest.h
Brandyn / Techy fcc1b09210 init
2026-04-04 15:40:51 -05:00

49 lines
1.2 KiB
C++

// Copyright Epic Games, Inc. All Rights Reserved.
#pragma once
#include "StateTreeSchema.h"
#include "StateTreeTypes.h"
#include "StateTreeTest.generated.h"
UCLASS(HideDropdown)
class UStateTreeTestSchema : public UStateTreeSchema
{
GENERATED_BODY()
public:
virtual bool IsStructAllowed(const UScriptStruct* InScriptStruct) const override
{
return true;
}
virtual bool IsExternalItemAllowed(const UStruct& InStruct) const override
{
return true;
}
virtual bool IsScheduledTickAllowed() const
{
return true;
}
virtual EStateTreeStateSelectionRules GetStateSelectionRules() const
{
return DefaultRules;
}
void SetStateSelectionRules(EStateTreeStateSelectionRules Rules)
{
DefaultRules = Rules;
}
private:
UPROPERTY()
EStateTreeStateSelectionRules DefaultRules = EStateTreeStateSelectionRules::Default;
};
UCLASS(HideDropdown)
class UStateTreeTestSchema2 : public UStateTreeSchema
{
GENERATED_BODY()
};
#define IMPLEMENT_STATE_TREE_INSTANT_TEST(TestClass, PrettyName) \
IMPLEMENT_AI_INSTANT_TEST_WITH_FLAGS(TestClass, PrettyName, EAutomationTestFlags::ClientContext | EAutomationTestFlags::EditorContext | EAutomationTestFlags::EngineFilter | EAutomationTestFlags::SupportsAutoRTFM)