1 files changed, 51 insertions, 47 deletions
diff --git a/Swiften/Network/FakeConnection.h b/Swiften/Network/FakeConnection.h
index eca45da..08c1d75 100644
--- a/Swiften/Network/FakeConnection.h
+++ b/Swiften/Network/FakeConnection.h
@@ -1,60 +1,64 @@
 /*
- * Copyright (c) 2010 Remko Tronçon
- * Licensed under the GNU General Public License v3.
- * See Documentation/Licenses/GPLv3.txt for more information.
+ * Copyright (c) 2010-2016 Isode Limited.
+ * All rights reserved.
+ * See the COPYING file for more information.
  */
 
 #pragma once
 
-#include <boost/optional.hpp>
-#include <boost/enable_shared_from_this.hpp>
+#include <memory>
 #include <vector>
 
+#include <boost/optional.hpp>
+
 #include <Swiften/Base/API.h>
+#include <Swiften/EventLoop/EventLoop.h>
+#include <Swiften/EventLoop/EventOwner.h>
 #include <Swiften/Network/Connection.h>
 #include <Swiften/Network/HostAddressPort.h>
-#include <Swiften/EventLoop/EventOwner.h>
-#include <Swiften/EventLoop/EventLoop.h>
 
 namespace Swift {
-	class SWIFTEN_API FakeConnection : 
-			public Connection, 
-			public EventOwner,
-			public boost::enable_shared_from_this<FakeConnection> {
-		public:
-			enum State {
-				Initial,
-				Connecting,
-				Connected,
-				Disconnected,
-				DisconnectedWithError
-			};
-
-			FakeConnection(EventLoop* eventLoop);
-			~FakeConnection();
-
-			virtual void listen();
-			virtual HostAddressPort getLocalAddress() const {
-				return HostAddressPort();
-			}
-
-			void setError(const Error& e);
-			virtual void connect(const HostAddressPort& address);
-			virtual void disconnect();
-
-			virtual void write(const SafeByteArray& data) {
-				dataWritten.push_back(data);
-			}
-
-			void setDelayConnect() {
-				delayConnect = true;
-			}
-
-			EventLoop* eventLoop;
-			boost::optional<HostAddressPort> connectedTo;
-			std::vector<SafeByteArray> dataWritten;
-			boost::optional<Error> error;
-			State state;
-			bool delayConnect;
-	};
+    class SWIFTEN_API FakeConnection :
+            public Connection,
+            public EventOwner,
+            public std::enable_shared_from_this<FakeConnection> {
+        public:
+            enum State {
+                Initial,
+                Connecting,
+                Connected,
+                Disconnected,
+                DisconnectedWithError
+            };
+
+            FakeConnection(EventLoop* eventLoop);
+            virtual ~FakeConnection();
+
+            virtual void listen();
+            virtual HostAddressPort getLocalAddress() const {
+                return HostAddressPort();
+            }
+            virtual HostAddressPort getRemoteAddress() const {
+                return HostAddressPort();
+            }
+
+            void setError(const Error& e);
+            virtual void connect(const HostAddressPort& address);
+            virtual void disconnect();
+
+            virtual void write(const SafeByteArray& data) {
+                dataWritten.push_back(data);
+            }
+
+            void setDelayConnect() {
+                delayConnect = true;
+            }
+
+            EventLoop* eventLoop;
+            boost::optional<HostAddressPort> connectedTo;
+            std::vector<SafeByteArray> dataWritten;
+            boost::optional<Error> error;
+            State state;
+            bool delayConnect;
+    };
 }