1 files changed, 4 insertions, 0 deletions
diff --git a/Swiften/Client/UnitTest/ClientSessionTest.cpp b/Swiften/Client/UnitTest/ClientSessionTest.cpp
index e9d1b21..22db8fc 100644
--- a/Swiften/Client/UnitTest/ClientSessionTest.cpp
+++ b/Swiften/Client/UnitTest/ClientSessionTest.cpp
@@ -371,70 +371,74 @@ class ClientSessionTest : public CppUnit::TestFixture {
 				virtual void writeFooter() {
 					receivedEvents.push_back(Event());
 				}
 
 				virtual void writeElement(boost::shared_ptr<Element> element) {
 					receivedEvents.push_back(Event(element));
 				}
 
 				virtual void writeData(const std::string&) {
 				}
 
 				virtual bool supportsTLSEncryption() {
 					return canTLSEncrypt;
 				}
 
 				virtual void addTLSEncryption() {
 					tlsEncrypted = true;
 				}
 
 				virtual bool isTLSEncrypted() {
 					return tlsEncrypted;
 				}
 
 				virtual ByteArray getTLSFinishMessage() const {
 					return ByteArray();
 				}
 
 				virtual Certificate::ref getPeerCertificate() const {
 					return Certificate::ref(new SimpleCertificate());
 				}
 
 				virtual boost::shared_ptr<CertificateVerificationError> getPeerCertificateVerificationError() const {
 					return boost::shared_ptr<CertificateVerificationError>();
 				}
 
+				virtual bool supportsZLibCompression() {
+					return true;
+				}
+			
 				virtual void addZLibCompression() {
 					compressed = true;
 				}
 
 				virtual void setWhitespacePingEnabled(bool enabled) {
 					whitespacePingEnabled = enabled;
 				}
 
 				virtual void resetXMPPParser() {
 					resetCount++;
 				}
 
 				void breakConnection() {
 					onClosed(boost::shared_ptr<SessionStream::Error>(new SessionStream::Error(SessionStream::Error::ConnectionReadError)));
 				}
 
 				void breakTLS() {
 					onClosed(boost::shared_ptr<SessionStream::Error>(new SessionStream::Error(SessionStream::Error::TLSError)));
 				}
 
 
 				void sendStreamStart() {
 					ProtocolHeader header;
 					header.setTo("foo.com");
 					return onStreamStartReceived(header);
 				}
 
 				void sendStreamFeaturesWithStartTLS() {
 					boost::shared_ptr<StreamFeatures> streamFeatures(new StreamFeatures());
 					streamFeatures->setHasStartTLS();
 					onElementReceived(streamFeatures);
 				}
 
 				void sendStreamError() {
 					onElementReceived(boost::make_shared<StreamError>());