OpenTween.Tests/HttpMessageHandlerMock.cs

   1 // OpenTween - Client of Twitter
   2 // Copyright (c) 2014 kim_upsilon (@kim_upsilon) <https://upsilo.net/~upsilon/>
   3 // All rights reserved.
   4 //
   5 // This file is part of OpenTween.
   6 //
   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)
  10 // any later version.
  11 //
  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
  15 // for more details.
  16 //
  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.
  21
  22 using System;
  23 using System.Collections.Generic;
  24 using System.Linq;
  25 using System.Net.Http;
  26 using System.Text;
  27 using System.Threading;
  28 using System.Threading.Tasks;
  29
  30 namespace OpenTween
  31 {
  32     public class HttpMessageHandlerMock : HttpMessageHandler
  33     {
  34         private readonly Queue<Func<HttpRequestMessage, Task<HttpResponseMessage>>> Queue =
  35             new Queue<Func<HttpRequestMessage, Task<HttpResponseMessage>>>();
  36
  37         public int QueueCount
  38             => this.Queue.Count;
  39
  40         public void Enqueue(Func<HttpRequestMessage, Task<HttpResponseMessage>> handler)
  41             => this.Queue.Enqueue(handler);
  42
  43         public void Enqueue(Func<HttpRequestMessage, HttpResponseMessage> handler)
  44             => this.Queue.Enqueue(x => Task.Run(() => handler(x)));
  45
  46         protected override async Task<HttpResponseMessage> SendAsync(HttpRequestMessage request, CancellationToken cancellationToken)
  47         {
  48             var handler = this.Queue.Dequeue();
  49             return await handler(request);
  50         }
  51     }
  52 }