47 lines
973 B
C++
47 lines
973 B
C++
// Copyright Epic Games, Inc. All Rights Reserved.
|
|
|
|
#pragma once
|
|
|
|
#include "CoreMinimal.h"
|
|
#include "AITestsCommon.h"
|
|
#include "Misc/App.h"
|
|
#include "Managers/MLAdapterManager.h"
|
|
|
|
|
|
#if WITH_RPCLIB
|
|
namespace rpc
|
|
{
|
|
class client;
|
|
}
|
|
|
|
struct FRPCTestBase : public FAITestBase
|
|
{
|
|
enum
|
|
{
|
|
DefaultServerPort = 10101
|
|
};
|
|
|
|
EMLAdapterServerMode Mode = EMLAdapterServerMode::Client;
|
|
FDelegateHandle BindClientHandle;
|
|
FDelegateHandle BindServerHandle;
|
|
rpc::client* RPCClient = nullptr;
|
|
|
|
FRPCTestBase()
|
|
{
|
|
BindClientHandle = UMLAdapterManager::Get().GetOnAddClientFunctions().AddLambda([this](FRPCServer& Server)
|
|
{
|
|
SetUpClientBinds(Server);
|
|
});
|
|
BindServerHandle = UMLAdapterManager::Get().GetOnAddServerFunctions().AddLambda([this](FRPCServer& Server)
|
|
{
|
|
SetUpServerBinds(Server);
|
|
});
|
|
}
|
|
|
|
virtual void SetUpClientBinds(FRPCServer& Server) {}
|
|
virtual void SetUpServerBinds(FRPCServer& Server) {}
|
|
|
|
virtual void TearDown() override;
|
|
};
|
|
#endif // WITH_RPCLIB
|