// OpenTween - Client of Twitter
// Copyright (c) 2014 kim_upsilon (@kim_upsilon)
// All rights reserved.
//
// This file is part of OpenTween.
//
// This program is free software; you can redistribute it and/or modify it
// under the terms of the GNU General public License as published by the Free
// Software Foundation; either version 3 of the License, or (at your option)
// any later version.
//
// This program is distributed in the hope that it will be useful, but
// WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
// or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General public License
// for more details.
//
// You should have received a copy of the GNU General public License along
// with this program. If not, see , or write to
// the Free Software Foundation, Inc., 51 Franklin Street - Fifth Floor,
// Boston, MA 02110-1301, USA.
using System;
using System.Collections.Generic;
using System.Linq;
using System.Net.Http;
using System.Text;
using System.Threading;
using System.Threading.Tasks;
namespace OpenTween
{
public class HttpMessageHandlerMock : HttpMessageHandler
{
private readonly Queue>> Queue =
new Queue>>();
public int QueueCount
{
get { return this.Queue.Count; }
}
public void Enqueue(Func> handler)
{
this.Queue.Enqueue(handler);
}
public void Enqueue(Func handler)
{
this.Queue.Enqueue(x => Task.Run(() => handler(x)));
}
protected override async Task SendAsync(HttpRequestMessage request, CancellationToken cancellationToken)
{
var handler = this.Queue.Dequeue();
return await handler(request);
}
}
}