* MonoTorrent.Tests/Client/TestRig.cs:

* MonoTorrent.Tests/MonoTorrent.Tests.csproj:
* MonoTorrent.Tests/Client/InitialSeedingModeTest.cs:
* MonoTorrent/MonoTorrent.Client/Modes/InitialSeedingMode.cs: When
  switching from InitialSeeding mode to any other mode, ensure that
  all connected peers get an updated bitfield message.

svn path=/trunk/bitsharp/; revision=157404

src/MonoTorrent.Tests/Client/InitialSeedingModeTest.cs

@@ -0,0 +1,53 @@
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using System.Text;
+
+using MonoTorrent.Client.Messages.FastPeer;
+using MonoTorrent.Client.Messages.Standard;
+using MonoTorrent.Common;
+using NUnit.Framework;
+
+namespace MonoTorrent.Client
+{
+    [TestFixture]
+    public class InitialSeedingModeTest
+    {
+        InitialSeedingMode Mode {
+            get { return Rig.Manager.Mode as InitialSeedingMode; }
+        }
+
+        TestRig Rig {
+            get; set;
+        }
+
+        [SetUp]
+        public void Setup()
+        {
+            Rig = TestRig.CreateSingleFile(Piece.BlockSize * 20, Piece.BlockSize * 2);
+            Rig.Manager.Mode = new InitialSeedingMode(Rig.Manager);
+        }
+
+        [TearDown]
+        public void Teardown()
+        {
+            Rig.Dispose();
+        }
+
+        [Test]
+        public void SwitchingModesSendsHaves()
+        {
+            Rig.Manager.Peers.ConnectedPeers.Add(Rig.CreatePeer(true, true));
+            Rig.Manager.Peers.ConnectedPeers.Add(Rig.CreatePeer(true, false));
+
+            var peer = Rig.CreatePeer(true);
+            peer.BitField.SetAll(true);
+            Mode.HandlePeerConnected(peer, Direction.Incoming);
+            Mode.Tick(0);
+
+            Assert.IsTrue(Rig.Manager.Peers.ConnectedPeers[0].Dequeue() is HaveAllMessage, "#1");
+            BitfieldMessage m = (BitfieldMessage) Rig.Manager.Peers.ConnectedPeers[1].Dequeue();
+            Assert.IsTrue(m.BitField.AllTrue, "#2");
+        }
+    }
+}

src/MonoTorrent.Tests/Client/TestRig.cs

@@ -407,14 +407,19 @@ public void AddConnection(IConnection connection)
             else
                 listener.Add(manager, connection);
         }
-
         public PeerId CreatePeer(bool processingQueue)
+        {
+            return CreatePeer(processingQueue, true);
+        }
+
+        public PeerId CreatePeer(bool processingQueue, bool supportsFastPeer)
         {
             StringBuilder sb = new StringBuilder();
             for (int i = 0; i < 20; i++)
                 sb.Append((char)Random.Next((int)'a', (int)'z'));
             Peer peer = new Peer(sb.ToString(), new Uri("tcp://127.0.0.1:" + (port++)));
             PeerId id = new PeerId(peer, Manager);
+            id.SupportsFastPeer = supportsFastPeer;
             id.ProcessingQueue = processingQueue;
             return id;
         }

src/MonoTorrent.Tests/MonoTorrent.Tests.csproj

@@ -123,6 +123,7 @@
     <Compile Include="Client\NetworkIOTests.cs" />
     <Compile Include="Common\MagnetLinkTest.cs" />
     <Compile Include="Common\SpeedMonitorTest.cs" />
+    <Compile Include="Client\InitialSeedingModeTest.cs" />
   </ItemGroup>
   <ItemGroup>
     <ProjectReference Include="..\MonoTorrent.Dht\MonoTorrent.Dht.csproj">

src/MonoTorrent/MonoTorrent.Client/Modes/InitialSeedingMode.cs

@@ -97,8 +97,16 @@ public override void HandlePeerDisconnected(PeerId id)
         public override void Tick(int counter)
         {
             base.Tick(counter);
-            if (unchoker.Complete)
+            if (unchoker.Complete) {
+                PeerMessage bitfieldMessage = new BitfieldMessage (Manager.Bitfield);
+                PeerMessage haveAllMessage = new HaveAllMessage();
+                foreach (var peer in Manager.Peers.ConnectedPeers)
+                {
+                    PeerMessage message = peer.SupportsFastPeer && Manager.Complete ? haveAllMessage : bitfieldMessage;
+                    peer.Enqueue(message);
+                }
                 Manager.Mode = new DownloadMode(Manager);
+            }
         }
     }
 }