MockResponse.cs

Go to the documentation of this file.

using System;
using System.Collections.Generic;
using System.Net;
using RestSharp;
 
namespace AnalyzeReTesting.Utilities
{
    public class MockResponse : RestResponse
    {
        public RestResponse Response { get; set; }
 
        public static MockResponse Error(Exception requestException, ResponseStatus status = ResponseStatus.Error)
        {
            return new MockResponse
            {
                ResponseStatus = status,
                ErrorException = requestException,
            };
        }
 
        public static MockResponse Completed(HttpStatusCode httpStatus, ResponseStatus status = ResponseStatus.Completed)
        {
            return new MockResponse
            {
                StatusCode = httpStatus,
                ResponseStatus = status,
            };
        }
 
        public static MockResponse RetryAfter(double? retry_time, int? queue_position)
        {
            MockResponse response = new MockResponse
            {
                StatusCode = HttpStatusCode.ServiceUnavailable,
            };
            List<Parameter> mockHeaders = new List<Parameter>();
            if (retry_time.HasValue)
                mockHeaders.Add(new Parameter
                {
                    Type = ParameterType.HttpHeader,
                    Name = "Retry-After",
                    Value = retry_time.Value
                });
            if (queue_position.HasValue)
                mockHeaders.Add(new Parameter
                {
                    Type = ParameterType.HttpHeader,
                    Name = "Queue-Position",
                    Value = queue_position.Value
                });
            response.Headers = mockHeaders;
            return response;
        }
    }
}