2 files changed, 74 insertions, 0 deletions
diff --git a/Swiften/QA/StorageTest/FileReadBytestreamTest.cpp b/Swiften/QA/StorageTest/FileReadBytestreamTest.cpp
new file mode 100644
index 0000000..b8e9fd4
--- /dev/null
+++ b/Swiften/QA/StorageTest/FileReadBytestreamTest.cpp
@@ -0,0 +1,73 @@
+/*
+ * Copyright (c) 2010 Remko Tronçon
+ * Licensed under the GNU General Public License v3.
+ * See Documentation/Licenses/GPLv3.txt for more information.
+ */
+
+#include <cppunit/extensions/HelperMacros.h>
+#include <cppunit/extensions/TestFactoryRegistry.h>
+
+#include "Swiften/FileTransfer/FileReadBytestream.h"
+#include "SwifTools/Application/PlatformApplicationPathProvider.h"
+
+using namespace Swift;
+
+class FileReadBytestreamTest : public CppUnit::TestFixture {
+		CPPUNIT_TEST_SUITE(FileReadBytestreamTest);
+		CPPUNIT_TEST(testRead);
+		CPPUNIT_TEST(testRead_Twice);
+		CPPUNIT_TEST(testIsFinished_NotFinished);
+		CPPUNIT_TEST(testIsFinished_IsFinished);
+		CPPUNIT_TEST_SUITE_END();
+
+	public:
+		void setUp() {
+			pathProvider = new PlatformApplicationPathProvider("FileReadBytestreamTest");
+		}
+
+		void tearDown() {
+			delete pathProvider;
+		}
+
+		void testRead() {
+			std::auto_ptr<FileReadBytestream> testling(createTestling());
+
+			ByteArray result = testling->read(10);
+
+			CPPUNIT_ASSERT_EQUAL(String("/*\n * Copy"), result.toString());
+		}
+
+		void testRead_Twice() {
+			std::auto_ptr<FileReadBytestream> testling(createTestling());
+
+			testling->read(10);
+			ByteArray result = testling->read(10);
+
+			CPPUNIT_ASSERT_EQUAL(String("right (c) "), result.toString());
+		}
+
+		void testIsFinished_NotFinished() {
+			std::auto_ptr<FileReadBytestream> testling(createTestling());
+
+			testling->read(10);
+
+			CPPUNIT_ASSERT(!testling->isFinished());
+		}
+
+		void testIsFinished_IsFinished() {
+			std::auto_ptr<FileReadBytestream> testling(createTestling());
+
+			testling->read(4096);
+
+			CPPUNIT_ASSERT(testling->isFinished());
+		}
+
+	private:
+		FileReadBytestream* createTestling() {
+			return new FileReadBytestream(pathProvider->getExecutableDir() / "FileReadBytestreamTest.cpp");
+		}
+
+		PlatformApplicationPathProvider* pathProvider;
+};
+
+CPPUNIT_TEST_SUITE_REGISTRATION(FileReadBytestreamTest);
diff --git a/Swiften/QA/StorageTest/SConscript b/Swiften/QA/StorageTest/SConscript
index bb8646e..4d0a197 100644
--- a/Swiften/QA/StorageTest/SConscript
+++ b/Swiften/QA/StorageTest/SConscript
@@ -15,5 +15,6 @@ if env["TEST"] :
 
 	tester = myenv.Program("StorageTest", [
 			"VCardFileStorageTest.cpp",
+			"FileReadBytestreamTest.cpp",
 		])
 	myenv.Test(tester, "system", is_checker = True)