ChunkEvictionPolicy.mock.h

Go to the documentation of this file.

// Copyright Epic Games, Inc. All Rights Reserved.
#pragma once
 
#include "Installer/ChunkEvictionPolicy.h"
#include "Common/StatsCollector.h"
 
#if WITH_DEV_AUTOMATION_TESTS
 
namespace BuildPatchServices
{
    class FMockChunkEvictionPolicy
        : public IChunkEvictionPolicy
    {
    public:
        typedef TTuple<double, TSet<FGuid>, int32, TSet<FGuid>, TSet<FGuid>> FQuery;
 
    public:
        virtual void Query(const TMap<FGuid, TUniquePtr<IChunkDataAccess>>& CurrentMap, int32 DesiredMax, TSet<FGuid>& OutCleanable, TSet<FGuid>& OutBootable) const override
        {
            OutCleanable = Cleanable;
            OutBootable = Bootable;
            RxQuery.Emplace(FStatsCollector::GetSeconds(), CurrentMapToKeySet(CurrentMap), DesiredMax, OutCleanable, OutBootable);
        }
 
    public:
        TSet<FGuid> CurrentMapToKeySet(const TMap<FGuid, TUniquePtr<IChunkDataAccess>>& CurrentMap) const
        {
            TArray<FGuid> Result;
            CurrentMap.GetKeys(Result);
            return TSet<FGuid>(MoveTemp(Result));
        }
 
    public:
        TSet<FGuid> Cleanable;
        TSet<FGuid> Bootable;
        mutable TArray<FQuery> RxQuery;
    };
}
 
#endif //WITH_DEV_AUTOMATION_TESTS