public void ShouldTriggerBlockReserved()
{
BitfieldChanged bitfieldChanged = new BitfieldChanged
{
Peer = PeerHash.Random(),
Bitfield = Bitfield.Sequence(false, true)
};
using (OmnibusFixture fixture = new OmnibusFixture())
using (OmnibusSession session = fixture.Start())
{
Trigger trigger = Trigger.Bind(ref session.Hooks.OnBlockReserved, data =>
{
data.Hash.Should().Be(session.Hash);
data.Peer.Should().Be(bitfieldChanged.Peer);
data.Block.Should().NotBeNull();
data.Block.Piece.Index.Should().Be(1);
data.Block.Offset.Should().Be(0);
data.Block.Size.Should().Be(3616);
});
session.Service.Start();
session.Service.HandleMetafileVerified(session.Metainfo);
session.Service.HandleDataVerified(session.Metainfo.Pieces.Length);
session.Service.Handle(bitfieldChanged);
session.Service.Schedule(OmnibusStrategy.RarestFirst, bitfieldChanged.Peer, 1);
session.Pipeline.Process();
trigger.Wait().Should().BeTrue();
}
}
public void ShouldTriggerBlockReservedSequentialForNotCompletedPieces()
{
BitfieldChanged bitfieldChangedA = new BitfieldChanged
{
Peer = PeerHash.Random(),
Bitfield = Bitfield.Sequence(true, true)
};
BitfieldChanged bitfieldChangedB = new BitfieldChanged
{
Peer = PeerHash.Random(),
Bitfield = Bitfield.Sequence(true, false)
};
using (OmnibusFixture fixture = new OmnibusFixture())
using (OmnibusSession session = fixture.Start())
{
DataVerified verified = new DataVerified
{
Hash = session.Hash,
Bitfield = Bitfield.Sequence(true, false)
};
Trigger trigger = Trigger.Bind(ref session.Hooks.OnBlockReserved, data =>
{
data.Block.Should().NotBeNull();
data.Block.Piece.Index.Should().Be(1);
});
session.Service.Start();
session.Service.HandleMetafileVerified(session.Metainfo);
session.Service.Handle(verified);
session.Service.Handle(bitfieldChangedA);
session.Service.Handle(bitfieldChangedB);
session.Pipeline.Process();
session.Service.Schedule(OmnibusStrategy.Sequential, bitfieldChangedA.Peer, 1);
session.Pipeline.Process();
trigger.Wait().Should().BeTrue();
}
}
public void ShouldTriggerPieceReadyWhenAllBlocksCompleted()
{
using (OmnibusFixture fixture = new OmnibusFixture())
using (OmnibusSession session = fixture.Start())
{
Trigger trigger = Trigger.Bind(ref session.Hooks.OnPieceReady, data =>
{
data.Hash.Should().Be(session.Hash);
data.Piece.Index.Should().Be(1);
});
session.Service.Start();
session.Service.HandleMetafileVerified(session.Metainfo);
session.Service.HandleDataVerified(session.Metainfo.Pieces.Length);
session.Service.Complete(new BlockIndex(1, 0, 16384));
session.Pipeline.Process();
trigger.Wait().Should().BeTrue();
}
}
public void ShouldTriggerDataCompletedWhenAllPiecesCompleted()
{
using (OmnibusFixture fixture = new OmnibusFixture())
using (OmnibusSession session = fixture.Start())
{
Trigger trigger = Trigger.Bind(ref session.Hooks.OnDataCompleted, data =>
{
data.Hash.Should().Be(session.Hash);
});
session.Service.Start();
session.Service.HandleMetafileVerified(session.Metainfo);
session.Service.HandleDataVerified(session.Metainfo.Pieces.Length);
session.Service.Complete(new PieceInfo(0));
session.Service.Complete(new PieceInfo(1));
session.Pipeline.Process();
trigger.Wait().Should().BeTrue();
}
}
