// Copyright Epic Games, Inc. All Rights Reserved.
#pragma once

#include "Installer/CloudChunkSource.h"
#include "Common/StatsCollector.h"

#if WITH_DEV_AUTOMATION_TESTS

namespace BuildPatchServices
{
	class FMockCloudChunkSourceStat
		: public ICloudChunkSourceStat
	{
	public:
		typedef TTuple<double, FGuid> FDownloadRequested;
		typedef TTuple<double, FGuid> FDownloadSuccess;
		typedef TTuple<double, FGuid, FString> FDownloadFailed;
		typedef TTuple<double, FGuid, FString, EChunkLoadResult> FDownloadCorrupt;
		typedef TTuple<double, FGuid, FString, double, double, double, double> FDownloadAborted;
		typedef TTuple<double, int64> FReceivedDataUpdated;
		typedef TTuple<double, int64> FRequiredDataUpdated;
		typedef TTuple<double, EBuildPatchDownloadHealth> FDownloadHealthUpdated;
		typedef TTuple<double, float> FSuccessRateUpdated;
		typedef TTuple<double, int32> FActiveRequestCountUpdated;

	public:
		virtual void OnDownloadRequested(const FGuid& ChunkId) override
		{
			RxDownloadRequested.Emplace(FStatsCollector::GetSeconds(), ChunkId);
		}

		virtual void OnDownloadSuccess(const FGuid& ChunkId) override
		{
			RxDownloadSuccess.Emplace(FStatsCollector::GetSeconds(), ChunkId);
		}

		virtual void OnDownloadFailed(const FGuid& ChunkId, const FString& Url) override
		{
			RxDownloadFailed.Emplace(FStatsCollector::GetSeconds(), ChunkId, Url);
		}

		virtual void OnDownloadCorrupt(const FGuid& ChunkId, const FString& Url, EChunkLoadResult LoadResult) override
		{
			RxDownloadCorrupt.Emplace(FStatsCollector::GetSeconds(), ChunkId, Url, LoadResult);
		}

		virtual void OnDownloadAborted(const FGuid& ChunkId, const FString& Url, double DownloadTimeMean, double DownloadTimeStd, double DownloadTime, double BreakingPoint) override
		{
			RxDownloadAborted.Emplace(FStatsCollector::GetSeconds(), ChunkId, Url, DownloadTimeMean, DownloadTimeStd, DownloadTime, BreakingPoint);
		}

		virtual void OnReceivedDataUpdated(int64 TotalBytes) override
		{
			RxReceivedDataUpdated.Emplace(FStatsCollector::GetSeconds(), TotalBytes);
		}

		virtual void OnRequiredDataUpdated(int64 TotalBytes) override
		{
			RxRequiredDataUpdated.Emplace(FStatsCollector::GetSeconds(), TotalBytes);
		}

		virtual void OnDownloadHealthUpdated(EBuildPatchDownloadHealth DownloadHealth) override
		{
			RxDownloadHealthUpdated.Emplace(FStatsCollector::GetSeconds(), DownloadHealth);
		}

		virtual void OnSuccessRateUpdated(float SuccessRate) override
		{
			RxSuccessRateUpdated.Emplace(FStatsCollector::GetSeconds(), SuccessRate);
		}

		virtual void OnActiveRequestCountUpdated(uint32 RequestCount) override
		{
			RxActiveRequestCountUpdated.Emplace(FStatsCollector::GetSeconds(), RequestCount);
		}

		virtual void OnAcceptedNewRequirements(const TSet<FGuid>& ChunkIds) override
		{
		}

	public:
		TArray<FDownloadRequested> RxDownloadRequested;
		TArray<FDownloadSuccess> RxDownloadSuccess;
		TArray<FDownloadFailed> RxDownloadFailed;
		TArray<FDownloadCorrupt> RxDownloadCorrupt;
		TArray<FDownloadAborted> RxDownloadAborted;
		TArray<FReceivedDataUpdated> RxReceivedDataUpdated;
		TArray<FRequiredDataUpdated> RxRequiredDataUpdated;
		TArray<FDownloadHealthUpdated> RxDownloadHealthUpdated;
		TArray<FSuccessRateUpdated> RxSuccessRateUpdated;
		TArray<FActiveRequestCountUpdated> RxActiveRequestCountUpdated;
	};
}

#endif //WITH_DEV_AUTOMATION_TESTS