Class RequestManagerMock

valdi/test/utils/RequestManagerMock.hpp:27–57

class RequestTask;

class RequestManagerMock : public snap::valdi_core::HTTPRequestManager,
                           public std::enable_shared_from_this<RequestManagerMock> {
public:
    RequestManagerMock(ILogger& logger);
    ~RequestManagerMock();

    Result<snap::valdi_core::HTTPResponse> getResponse(const snap::valdi_core::HTTPRequest& request);

    void addMockedResponse(StringBox url, StringBox method, BytesView successBody);
    void addMockedResponseForURLSuffix(StringBox urlSuffix, StringBox method, BytesView successBody);

    void addMockedResponse(const snap::valdi_core::HTTPRequest& request,
                           const snap::valdi_core::HTTPResponse& response);

    void addMockedResponse(const snap::valdi_core::HTTPRequest& request,
                           const Result<snap::valdi_core::HTTPResponse>& response);

    void setDisabled(bool disabled);

    std::shared_ptr<snap::valdi_core::Cancelable> performRequest(
        const snap::valdi_core::HTTPRequest& request,
        const std::shared_ptr<snap::valdi_core::HTTPRequestManagerCompletion>& completion) override;

    std::vector<Shared<RequestTask>> getAllPerformedTasks() const;

private:
    Ref<DispatchQueue> _queue;
    std::vector<std::pair<snap::valdi_core::HTTPRequest, Result<snap::valdi_core::HTTPResponse>>> _mockedResponses;
    std::vector<std::pair<snap::valdi_core::HTTPRequest, Result<snap::valdi_core::HTTPResponse>>>
        _mockedSuffixResponses;
    [[maybe_unused]] ILogger& _logger;
    std::vector<Shared<RequestTask>> _tasks;
    bool _disabled = false;
};