/// <summary>
/// Initializes a new instance of the <see cref="Distribox.Network.RequestManager"/> class.
/// </summary>
public RequestManager()
{
this.todoFileToPeer = new Dictionary<FileEvent, List<Peer>>();
this.todoPeerToFile = new Dictionary<Peer,List<FileEvent>>();
this.doing = new List<DoingQueueItem>();
this.estimator = new BandwidthEstimator();
this.usedBandwidth = 0;
}
/// <summary>
/// Initializes a new instance of the <see cref="Distribox.Network.RequestManager"/> class.
/// </summary>
public RequestManager()
{
this.todoFileToPeer = new Dictionary <FileEvent, List <Peer> >();
this.todoPeerToFile = new Dictionary <Peer, List <FileEvent> >();
this.doing = new List <DoingQueueItem>();
this.estimator = new BandwidthEstimator();
this.usedBandwidth = 0;
}
public void Restore()
{
if (File.Exists(Config.PeerBandwidthFilePath))
{
File.Delete(Config.PeerBandwidthFilePath);
}
BandwidthEstimator estimator = new BandwidthEstimator();
Peer peer1 = new Peer("127.0.0.1", 1111);
estimator.BeginRequest(peer1, 0x1234, 1024 * 1024);
Thread.Sleep(1150);
estimator.FinishRequest(0x1234);
BandwidthEstimator estimator2 = new BandwidthEstimator();
Assert.AreEqual(1024 * 1024, estimator2.GetPeerBandwidth(peer1));
}
public void MultiPeer()
{
if (File.Exists(Config.PeerBandwidthFilePath))
{
File.Delete(Config.PeerBandwidthFilePath);
}
BandwidthEstimator estimator = new BandwidthEstimator();
Peer peer1 = new Peer("127.0.0.1", 1111);
Peer peer2 = new Peer("127.0.0.1", 1111);
estimator.BeginRequest(peer1, 0x1234, 1024 * 1024);
estimator.BeginRequest(peer2, 0x1235, 2048 * 1024);
estimator.FinishRequest(0x1234);
Assert.AreEqual(1024 * 1024, estimator.TotalBandwidth);
estimator.FinishRequest(0x1235);
Assert.AreEqual(3072 * 1024, estimator.TotalBandwidth);
}
public void SinglePeer()
{
if (File.Exists(Config.PeerBandwidthFilePath))
{
File.Delete(Config.PeerBandwidthFilePath);
}
BandwidthEstimator estimator = new BandwidthEstimator();
Peer peer1 = new Peer("127.0.0.1", 1111);
// Invalid peer
Assert.AreEqual(Properties.DefaultConnectionSpeed, estimator.GetPeerBandwidth(peer1));
// Do a task
estimator.BeginRequest(peer1, 0x1234, 1024 * 1024);
Thread.Sleep(1150);
estimator.FinishRequest(0x1234);
Assert.AreEqual(1024 * 1024, estimator.GetPeerBandwidth(peer1));
// Do another slower task
estimator.BeginRequest(peer1, 0x1234, 512 * 1024);
Thread.Sleep(1150);
estimator.FinishRequest(0x1234);
Assert.AreEqual(512 * 1024, estimator.GetPeerBandwidth(peer1));
// Do two tasks
estimator.BeginRequest(peer1, 0x1234, 1024 * 1024);
Thread.Sleep(1150);
estimator.BeginRequest(peer1, 0x1235, 768 * 1024);
Assert.AreEqual(512 * 1024, estimator.GetPeerBandwidth(peer1));
estimator.FinishRequest(0x1235);
Assert.AreEqual(768 * 1024, estimator.GetPeerBandwidth(peer1));
Thread.Sleep(3150);
Assert.AreEqual(768 * 1024, estimator.GetPeerBandwidth(peer1));
estimator.FinishRequest(0x1234);
Assert.AreEqual(256 * 1024, estimator.GetPeerBandwidth(peer1));
// Fail a task
estimator.BeginRequest(peer1, 0x1234, 1024 * 1024);
estimator.FailRequest(0x1234);
}
