1 // OpenTween - Client of Twitter
2 // Copyright (c) 2014 kim_upsilon (@kim_upsilon) <https://upsilo.net/~upsilon/>
3 // All rights reserved.
5 // This file is part of OpenTween.
7 // This program is free software; you can redistribute it and/or modify it
8 // under the terms of the GNU General public License as published by the Free
9 // Software Foundation; either version 3 of the License, or (at your option)
12 // This program is distributed in the hope that it will be useful, but
13 // WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
14 // or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General public License
17 // You should have received a copy of the GNU General public License along
18 // with this program. If not, see <http://www.gnu.org/licenses/>, or write to
19 // the Free Software Foundation, Inc., 51 Franklin Street - Fifth Floor,
20 // Boston, MA 02110-1301, USA.
23 using System.Collections.Generic;
25 using System.Net.Http;
27 using System.Threading;
28 using System.Threading.Tasks;
32 public class HttpMessageHandlerMock : HttpMessageHandler
34 private readonly Queue<Func<HttpRequestMessage, Task<HttpResponseMessage>>> Queue =
35 new Queue<Func<HttpRequestMessage, Task<HttpResponseMessage>>>();
37 public void Enqueue(Func<HttpRequestMessage, Task<HttpResponseMessage>> handler)
39 this.Queue.Enqueue(handler);
42 public void Enqueue(Func<HttpRequestMessage, HttpResponseMessage> handler)
44 this.Queue.Enqueue(x => Task.FromResult(handler(x)));
47 protected override Task<HttpResponseMessage> SendAsync(HttpRequestMessage request, CancellationToken cancellationToken)
49 var handler = this.Queue.Dequeue();
50 return handler(request);