3 files changed, 3 insertions, 2 deletions
diff --git a/Swiften/FileTransfer/FileWriteBytestream.h b/Swiften/FileTransfer/FileWriteBytestream.h
index c563789..26ab98e 100644
--- a/Swiften/FileTransfer/FileWriteBytestream.h
+++ b/Swiften/FileTransfer/FileWriteBytestream.h
@@ -19,7 +19,7 @@ namespace Swift {
             virtual ~FileWriteBytestream();
 
             virtual bool write(const std::vector<unsigned char>&);
-            void close();
+            virtual void close();
 
         private:
             boost::filesystem::path file;
diff --git a/Swiften/FileTransfer/WriteBytestream.h b/Swiften/FileTransfer/WriteBytestream.h
index 5d9c3f8..5452317 100644
--- a/Swiften/FileTransfer/WriteBytestream.h
+++ b/Swiften/FileTransfer/WriteBytestream.h
@@ -26,6 +26,7 @@ namespace Swift {
              * On success true is returned and \ref onWrite is called. On failure false is returned.
              */
             virtual bool write(const std::vector<unsigned char>&) = 0;
+            virtual void close() {}
 
             boost::signals2::signal<void (const std::vector<unsigned char>&)> onWrite;
     };
diff --git a/Swiften/QA/StorageTest/FileWriteBytestreamTest.cpp b/Swiften/QA/StorageTest/FileWriteBytestreamTest.cpp
index 7823519..393d1d7 100644
--- a/Swiften/QA/StorageTest/FileWriteBytestreamTest.cpp
+++ b/Swiften/QA/StorageTest/FileWriteBytestreamTest.cpp
@@ -34,7 +34,7 @@ class FileWriteBytestreamTest : public CppUnit::TestFixture {
 
             CPPUNIT_ASSERT_EQUAL(true, writeBytestream->write(createByteArray("Some data.")));
             CPPUNIT_ASSERT_EQUAL(true, onWriteWasCalled);
-
+            writeBytestream->close();
             boost::filesystem::remove(filename);
         }