HttpManager.fake.h

Go to the documentation of this file.

// Copyright Epic Games, Inc. All Rights Reserved.
#pragma once
 
#include "Containers/Ticker.h"
#include "Tests/Mock/HttpManager.mock.h"
#include "Tests/Mock/HttpRequest.mock.h"
#include "Tests/Mock/HttpResponse.mock.h"
 
#if WITH_DEV_AUTOMATION_TESTS
 
namespace BuildPatchServices
{
    class FFakeHttpRequest
        : public FMockHttpRequest
    {
    public:
        virtual bool ProcessRequest() override;
 
    public:
        class FFakeHttpManager* FakeHttpModule;
    };
 
    class FFakeHttpResponse
        : public FMockHttpResponse
    {
    public:
        virtual int32 GetResponseCode() const override;
        virtual const TArray<uint8>& GetContent() const override;
        virtual FString GetContentAsString() const override;
        virtual uint64 GetContentLength() const override;
 
    public:
        TArray<uint8> Data;
        EHttpResponseCodes::Type Code;
    };
 
    class FFakeHttpManager
        : public FMockHttpManager
    {
    public:
        FFakeHttpManager(FTSTicker& Ticker);
        virtual TSharedRef<IHttpRequest, ESPMode::ThreadSafe> CreateRequest() override;
        virtual bool Tick(float Delta);
        virtual bool OnProcessRequest(FFakeHttpRequest* FakeHttpRequest);
 
    public:
        TArray<TSharedRef<FFakeHttpRequest, ESPMode::ThreadSafe>> NewRequests;
        TArray<TSharedRef<FFakeHttpRequest, ESPMode::ThreadSafe>> RunningRequests;
        TArray<TSharedRef<FFakeHttpRequest, ESPMode::ThreadSafe>> ProgressedRequests;
        TMap<FString, TArray<uint8>> DataServed;
    };
 
    bool FFakeHttpRequest::ProcessRequest()
    {
        ++RxProcessRequest;
        return FakeHttpModule->OnProcessRequest(this);
    }
 
    int32 FFakeHttpResponse::GetResponseCode() const
    {
        return static_cast<int32>(Code);
    }
 
    const TArray<uint8>& FFakeHttpResponse::GetContent() const
    {
        return Data;
    }
 
    FString FFakeHttpResponse::GetContentAsString() const
    {
        TArray<uint8> ZeroTerminatedPayload(GetContent());
        ZeroTerminatedPayload.Add(0);
        return UTF8_TO_TCHAR(ZeroTerminatedPayload.GetData());
    }
 
    uint64 FFakeHttpResponse::GetContentLength() const
    {
        return Data.Num();
    }
 
    FFakeHttpManager::FFakeHttpManager(FTSTicker& Ticker)
    {
        Ticker.AddTicker(FTickerDelegate::CreateRaw(this, &FFakeHttpManager::Tick));
    }
 
    TSharedRef<IHttpRequest, ESPMode::ThreadSafe> FFakeHttpManager::CreateRequest()
    {
        ++RxCreateRequest;
        NewRequests.Emplace(new FFakeHttpRequest());
        NewRequests.Last()->FakeHttpModule = this;
        return NewRequests.Last();
    }
 
    bool FFakeHttpManager::Tick(float Delta)
    {
        // Complete progressed.
        for (const TSharedRef<FFakeHttpRequest, ESPMode::ThreadSafe>& ProgressedRequest : ProgressedRequests)
        {
            FFakeHttpResponse* FakeHttpResponse = new FFakeHttpResponse();
            FakeHttpResponse->Code = EHttpResponseCodes::Ok;
            const FString& Url = ProgressedRequest->RxSetURL.Last().Get<0>();
            if (DataServed.Contains(Url))
            {
                FakeHttpResponse->Data = DataServed[Url];
            }
            ProgressedRequest->HttpRequestCompleteDelegate.ExecuteIfBound(
                ProgressedRequest,
                MakeShareable(FakeHttpResponse),
                true);
        }
 
        // Progress started.
        ProgressedRequests = MoveTemp(RunningRequests);
        for (const TSharedRef<FFakeHttpRequest, ESPMode::ThreadSafe>& ProgressedRequest : ProgressedRequests)
        {
            const FString& Url = ProgressedRequest->RxSetURL.Last().Get<0>();
            int32 Progress = 0;
            if (DataServed.Contains(Url))
            {
                Progress = DataServed[Url].Num() / 2;
            }
            ProgressedRequest->HttpRequestProgressDelegate.ExecuteIfBound(
                ProgressedRequest,
                0,
                Progress);
        }
 
        return true;
    }
 
    bool FFakeHttpManager::OnProcessRequest(FFakeHttpRequest* FakeHttpRequest)
    {
        int32 Index = NewRequests.IndexOfByPredicate([FakeHttpRequest](const TSharedRef<FFakeHttpRequest, ESPMode::ThreadSafe>& Element)
        {
            return &Element.Get() == FakeHttpRequest;
        });
        bool bValidRequest = Index >= 0;
        if (bValidRequest)
        {
            const TSharedRef<FFakeHttpRequest, ESPMode::ThreadSafe>& Request = NewRequests[Index];
            bValidRequest = Request->RxSetURL.Num() > 0;
            if (bValidRequest)
            {
                RunningRequests.Add(Request);
                NewRequests.RemoveAtSwap(Index);
            }
        }
        return bValidRequest;
    }
}
 
#endif //WITH_DEV_AUTOMATION_TESTS