Exemplo n.º 1
0
        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();
                }
        }
Exemplo n.º 2
0
        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();
                }
        }
Exemplo n.º 3
0
        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();
                }
        }
Exemplo n.º 4
0
        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();
                }
        }