summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--Swiften/FileTransfer/FileWriteBytestream.h2
-rw-r--r--Swiften/FileTransfer/WriteBytestream.h1
-rw-r--r--Swiften/QA/StorageTest/FileWriteBytestreamTest.cpp2
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);
}