public async void UpdateLimits_given_headers_with_reset_limits_updates_limits()
{
var limitSourceMock = new Mock <ISource <ImgurRatelimitResponse> >();
var limitResponse = new ImgurRatelimitResponse {
ClientLimit = 12500, ClientRemaining = 1000, UserLimit = 500, UserRemaining = 50, StatusCode = HttpStatusCode.OK
};
limitSourceMock.Setup(m => m.GetContent(It.IsAny <string>())).ReturnsAsync(limitResponse);
var source = new ImgurRatelimiter(limitSourceMock.Object);
await source.LoadLimits();
var headers = new List <KeyValuePair <string, IEnumerable <string> > >
{
new KeyValuePair <string, IEnumerable <string> >("X-RateLimit-ClientRemaining", Enumerable.Repeat("12490", 1)),
new KeyValuePair <string, IEnumerable <string> >("X-RateLimit-UserLimit", Enumerable.Repeat("498", 1))
};
source.UpdateLimit(headers);
var vals = source.GetLimiterValues();
Assert.Equal(limitResponse.ClientLimit, vals.Item1);
Assert.Equal(limitResponse.UserLimit, vals.Item2);
Assert.Equal(12490, vals.Item3);
Assert.Equal(498, vals.Item4);
}
public async void LoadLimits_given_successful_ratelimitresponse_sets_values()
{
var limitSourceMock = new Mock <ISource <ImgurRatelimitResponse> >();
var limitResponse = new ImgurRatelimitResponse {
ClientLimit = 10, ClientRemaining = 20, UserLimit = 30, UserRemaining = 40, StatusCode = HttpStatusCode.OK
};
limitSourceMock.Setup(m => m.GetContent(It.IsAny <string>())).ReturnsAsync(limitResponse);
var source = new ImgurRatelimiter(limitSourceMock.Object);
await source.LoadLimits();
var vals = source.GetLimiterValues();
Assert.Equal(limitResponse.ClientLimit, vals.Item1);
Assert.Equal(limitResponse.UserLimit, vals.Item2);
Assert.Equal(limitResponse.ClientRemaining, vals.Item3);
Assert.Equal(limitResponse.UserRemaining, vals.Item4);
limitSourceMock.Verify(i => i.GetContent(It.IsAny <string>()), Times.Once);
}
public async void LoadLimits_when_connection_forbidden_sets_expected_values()
{
var limitSourceMock = new Mock <ISource <ImgurRatelimitResponse> >();
var limitResponse = new ImgurRatelimitResponse {
StatusCode = HttpStatusCode.Forbidden
};
limitSourceMock.Setup(m => m.GetContent(It.IsAny <string>())).ReturnsAsync(limitResponse);
var source = new ImgurRatelimiter(limitSourceMock.Object);
await Assert.ThrowsAsync <InvalidClientIDException>(source.LoadLimits);
var vals = source.GetLimiterValues();
Assert.Equal(-1, vals.Item1);
Assert.Equal(-1, vals.Item2);
Assert.Equal(-1, vals.Item3);
Assert.Equal(-1, vals.Item4);
limitSourceMock.Verify(i => i.GetContent(It.IsAny <string>()), Times.Once);
}
/// <summary>
/// Loads the limits imposed on the program and returns the limiter values.
/// </summary>
/// <returns>
/// A tuple of ints whose contents are:
/// Item 1: The max number of client-wide requests.
/// Item 2: The max number of user-specific requests.
/// Item 3: The remaining number of client-wide requests.
/// Item 4: The remaining number of user-specific requests.
/// </returns>
public async Task <Tuple <int, int, int, int> > GetImgurLimitInfo()
{
await _imgurLimiter.LoadLimits();
return(_imgurLimiter.GetLimiterValues());
}