SystemExtensions/SystemExtensions.Tests/Utils/MockHttpMessageHandler.cs

using System;
using System.Net.Http;
using System.Threading;
using System.Threading.Tasks;

namespace SystemExtensionsTests.Utils
{
    public sealed class MockHttpMessageHandler : HttpMessageHandler
    {
        public Func<HttpRequestMessage, HttpResponseMessage> ResponseResolver { get; set; }
        public Func<HttpRequestMessage, Task<HttpResponseMessage>> ResponseResolverAsync { get; set; }

        public MockHttpMessageHandler()
        {
        }

        protected override Task<HttpResponseMessage> SendAsync(HttpRequestMessage request, CancellationToken cancellationToken)
        {
            if (this.ResponseResolver is object)
            {
                return Task.FromResult(this.ResponseResolver(request));
            }
            else
            {
                return this.ResponseResolverAsync(request);
            }
        }
    }
}