1 files changed, 27 insertions, 0 deletions
diff --git a/rdiff-backup/testing/iterfiletest.py b/rdiff-backup/testing/iterfiletest.py
new file mode 100644
index 0000000..38dca4d
--- /dev/null
+++ b/rdiff-backup/testing/iterfiletest.py
@@ -0,0 +1,27 @@
+import unittest, StringIO
+execfile("commontest.py")
+rbexec("iterfile.py")
+
+
+class testIterFile(unittest.TestCase):
+	def setUp(self):
+		self.iter1maker = lambda: iter(range(50))
+		self.iter2maker = lambda: iter(map(str, range(50)))
+
+	def testConversion(self):
+		"""Test iter to file conversion"""
+		for itm in [self.iter1maker, self.iter2maker]:
+			assert Iter.equal(itm(),
+							  IterWrappingFile(FileWrappingIter(itm())))
+
+class testBufferedRead(unittest.TestCase):
+	def testBuffering(self):
+		"""Test buffering a StringIO"""
+		fp = StringIO.StringIO("12345678"*10000)
+		bfp = BufferedRead(fp)
+		assert bfp.read(5) == "12345"
+		assert bfp.read(4) == "6781"
+		assert len(bfp.read(75000)) == 75000
+
+
+if __name__ == "__main__": unittest.main()