Add a check for a connection terminated by server (#21)

SharpXMPP.NUnit/TestUtils/MockedXmppTcpConnection.cs

@@ -0,0 +1,14 @@
+using System.IO;
+using System.Xml;
+using SharpXMPP.XMPP;
+
+namespace SharpXMPP.NUnit.TestUtils
+{
+    public class MockedXmppTcpConnection : XmppTcpConnection
+    {
+        public MockedXmppTcpConnection(Stream stream) : base("", new JID(), "")
+        {
+            Reader = XmlReader.Create(stream);
+        }
+    }
+}

SharpXMPP.NUnit/XmppTcpConnectionTests.cs

@@ -0,0 +1,30 @@
+using System.IO;
+using System.Text;
+using NUnit.Framework;
+using SharpXMPP.Errors;
+using SharpXMPP.NUnit.TestUtils;
+
+namespace SharpXMPP.NUnit
+{
+    [TestFixture]
+    public class XmppTcpConnectionTests
+    {
+        private const string EmptyXmppStream = @"<?xml version='1.0'?>
+<stream:stream from='example.com'
+               id='someid'
+               xmlns='jabber:client'
+               xmlns:stream='http://etherx.jabber.org/streams'
+               version='1.0'>
+</stream:stream>";
+
+        [Test]
+        public void ConnectionShouldStopAfterStreamClosed()
+        {
+            using (var stream = new MemoryStream(Encoding.UTF8.GetBytes(EmptyXmppStream)))
+            using (var connection = new MockedXmppTcpConnection(stream))
+            {
+                Assert.Throws<XmppConnectionTerminatedException>(() => connection.NextElement());
+            }
+        }
+    }
+}

SharpXMPP.Shared/Errors/XmppConnectionTerminatedException.cs

@@ -0,0 +1,9 @@
+using System;
+
+namespace SharpXMPP.Errors
+{
+    public class XmppConnectionTerminatedException : Exception
+    {
+        public override string Message => "XMPP connection was terminated by server";
+    }
+}

SharpXMPP.Shared/Properties/AssemblyInfo.cs

@@ -0,0 +1,3 @@
+using System.Runtime.CompilerServices;
+
+[assembly: InternalsVisibleTo("SharpXMPP.NUnit")]

SharpXMPP.Shared/SharpXMPP.Shared.csproj

@@ -5,6 +5,7 @@
     <TargetFrameworks>netstandard1.6;net451</TargetFrameworks>
     <DebugType>portable</DebugType>
     <AssemblyName>SharpXMPP.Shared</AssemblyName>
+    <RootNamespace>SharpXMPP</RootNamespace>
     <NetStandardImplicitPackageVersion Condition=" '$(TargetFramework)' == 'netstandard1.6' ">1.6.0</NetStandardImplicitPackageVersion>
   </PropertyGroup>
 

SharpXMPP.Shared/XmppTcpConnection.cs

@@ -8,6 +8,7 @@
 using System.Threading.Tasks;
 using System.Xml;
 using System.Xml.Linq;
+using SharpXMPP.Errors;
 using SharpXMPP.XMPP;
 using SharpXMPP.XMPP.Bind;
 using SharpXMPP.XMPP.Client;
@@ -41,7 +42,7 @@ protected XmppTcpConnection(string ns, JID jid, string password) : base(ns)
             Password = password;
         }
 
-        public System.IO.Stream ConnectionStream { get; private set; }
+        public System.IO.Stream ConnectionStream { get; private protected set; }
 
         protected XmlReader Reader;
         protected XmlWriter Writer;
@@ -76,7 +77,10 @@ public override XElement NextElement()
             }
             do
             {
-                Reader.Read();
+                if (!Reader.Read())
+                {
+                    throw new XmppConnectionTerminatedException();
+                }
             } while (Reader.NodeType != XmlNodeType.Element);
             var result = XElement.Load(Reader.ReadSubtree());
             OnElement(new ElementArgs { Stanza = result, IsInput = true });