58 lines
1.6 KiB
C++
58 lines
1.6 KiB
C++
// Copyright Epic Games, Inc. All Rights Reserved.
|
|
|
|
#pragma once
|
|
|
|
#include "Iris/DataStream/DataStream.h"
|
|
#include "Net/Core/Misc/ResizableCircularQueue.h"
|
|
#include "MockDataStream.generated.h"
|
|
|
|
UCLASS()
|
|
class UMockDataStream : public UDataStream
|
|
{
|
|
GENERATED_BODY()
|
|
|
|
public:
|
|
struct FFunctionCallStatus
|
|
{
|
|
uint32 WriteDataCallCount;
|
|
uint32 ReadDataCallCount;
|
|
uint32 ProcessPacketDeliveryStatusCallCount;
|
|
uint32 ProcessPacketDeliveryStatusMagicValue;
|
|
uint32 UpdateCallCount;
|
|
};
|
|
|
|
struct FFunctionCallSetup
|
|
{
|
|
// WriteData
|
|
EWriteResult WriteDataReturnValue;
|
|
uint32 WriteDataBitCount;
|
|
uint32 WriteDataRecordMagicValue;
|
|
|
|
// ReadData
|
|
uint32 ReadDataBitCount;
|
|
};
|
|
|
|
inline void SetFunctionCallSetup(const FFunctionCallSetup& Setup) { CallSetup = Setup; }
|
|
inline const FFunctionCallStatus& GetFunctionCallStatus() const { return CallStatus; }
|
|
inline void ResetFunctionCallStatus() { CallStatus = FFunctionCallStatus({}); }
|
|
|
|
protected:
|
|
virtual EWriteResult WriteData(UE::Net::FNetSerializationContext& context, FDataStreamRecord const*& OutRecord) override;
|
|
virtual void ReadData(UE::Net::FNetSerializationContext& context) override;
|
|
virtual void ProcessPacketDeliveryStatus(UE::Net::EPacketDeliveryStatus Status, FDataStreamRecord const* Record) override;
|
|
virtual bool HasAcknowledgedAllReliableData() const override;
|
|
virtual void Update(const FUpdateParameters& Params) override;
|
|
|
|
private:
|
|
struct FRecord : public FDataStreamRecord
|
|
{
|
|
uint32 MagicValue;
|
|
};
|
|
|
|
UMockDataStream();
|
|
|
|
FFunctionCallStatus CallStatus;
|
|
FFunctionCallSetup CallSetup;
|
|
TResizableCircularQueue<FRecord, TFixedAllocator<256>> Records;
|
|
};
|