Files
UnrealEngine/Engine/Source/Runtime/Online/BuildPatchServices/Private/Tests/Mock/DownloadServiceStat.mock.h
2025-05-18 13:04:45 +08:00

43 lines
1.1 KiB
C++

// Copyright Epic Games, Inc. All Rights Reserved.
#pragma once
#include "Common/StatsCollector.h"
#include "Installer/DownloadService.h"
#if WITH_DEV_AUTOMATION_TESTS
namespace BuildPatchServices
{
class FMockDownloadServiceStat
: public IDownloadServiceStat
{
public:
typedef TTuple<double, int32, FString> FDownloadStarted;
typedef TTuple<double, int32, int32> FDownloadProgress;
typedef TTuple<double, FDownloadRecord> FDownloadComplete;
public:
virtual void OnDownloadStarted(int32 RequestId, const FString& Uri) override
{
RxDownloadStarted.Emplace(FStatsCollector::GetSeconds(), RequestId, Uri);
}
virtual void OnDownloadProgress(int32 RequestId, uint64 BytesReceived) override
{
RxDownloadProgress.Emplace(FStatsCollector::GetSeconds(), RequestId, BytesReceived);
}
virtual void OnDownloadComplete(const FDownloadRecord& DownloadRecord) override
{
RxDownloadComplete.Emplace(FStatsCollector::GetSeconds(), DownloadRecord);
}
public:
TArray<FDownloadStarted> RxDownloadStarted;
TArray<FDownloadProgress> RxDownloadProgress;
TArray<FDownloadComplete> RxDownloadComplete;
};
}
#endif //WITH_DEV_AUTOMATION_TESTS