1 files changed, 4 insertions, 0 deletions
diff --git a/Swiften/Component/UnitTest/ComponentSessionTest.cpp b/Swiften/Component/UnitTest/ComponentSessionTest.cpp
index c27ade5..1541cce 100644
--- a/Swiften/Component/UnitTest/ComponentSessionTest.cpp
+++ b/Swiften/Component/UnitTest/ComponentSessionTest.cpp
@@ -110,70 +110,74 @@ class ComponentSessionTest : 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 false;
 				}
 
 				virtual void addTLSEncryption() {
 					assert(false);
 				}
 
 				virtual bool isTLSEncrypted() {
 					return false;
 				}
 
 				virtual ByteArray getTLSFinishMessage() const {
 					return ByteArray();
 				}
 
 				virtual Certificate::ref getPeerCertificate() const {
 					return Certificate::ref();
 				}
 
 				virtual boost::shared_ptr<CertificateVerificationError> getPeerCertificateVerificationError() const {
 					return boost::shared_ptr<CertificateVerificationError>();
 				}
 
+				virtual bool supportsZLibCompression() {
+					return true;
+				}
+
 				virtual void addZLibCompression() {
 					assert(false);
 				}
 
 				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 sendStreamStart() {
 					ProtocolHeader header;
 					header.setFrom("service.foo.com");
 					return onStreamStartReceived(header);
 				}
 
 				void sendHandshakeResponse() {
 					onElementReceived(ComponentHandshake::ref(new ComponentHandshake()));
 				}
 
 				void sendHandshakeError() {
 					// FIXME: This isn't the correct element
 					onElementReceived(AuthFailure::ref(new AuthFailure()));
 				}
 
 				void receiveStreamStart() {
 					Event event = popEvent();
 					CPPUNIT_ASSERT(event.header);
 				}