zfec/zunfec: make the cmdline programs take input from stdin

[tahoe-lafs/zfec.git] / zfec / test / test_zfec.py

diff --git a/zfec/test/test_zfec.py b/zfec/test/test_zfec.py
index 46bbb1cb6bbdeaf8c12a2982a5c326ceafcf02f4..b2d9a5b9e923bba1d2f982b1d540410550cf4df1 100755 (executable)
--- a/zfec/test/test_zfec.py
+++ b/zfec/test/test_zfec.py
@@ -246,6 +246,10 @@ class FileFec(unittest.TestCase):
                         assert (rm, rk, rpad, rsh,) == (m, k, pad, sh,), h
 
     def _help_test_filefec(self, teststr, k, m, numshs=None):
+        self._do_help_test_filefec(teststr, k, m, numshs=numshs, withsize=False)
+        self._do_help_test_filefec(teststr, k, m, numshs=numshs, withsize=True)
+
+    def _do_help_test_filefec(self, teststr, k, m, numshs=None, withsize=True):
         if numshs == None:
             numshs = m
 
@@ -253,7 +257,10 @@ class FileFec(unittest.TestCase):
         PREFIX = "test"
         SUFFIX = ".fec"
 
-        fsize = len(teststr)
+        if withsize:
+          fsize = len(teststr)
+        else:
+          fsize = None
 
         tempdir = fileutil.NamedTemporaryDirectory(cleanup=True)
         try: