DownloadServiceStat.mock.h

Go to the documentation of this file.

// 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