Adding unlimited=bool to files_varied -- if set to true, it will never issue StopIteration

s3tests/realistic.py

@@ -160,7 +160,7 @@ def names(mean, stddev, charset=None, seed=None):
         name = ''.join(rand.choice(charset) for _ in xrange(length))
         yield name
 
-def files_varied(groups):
+def files_varied(groups, unlimited=False):
     """ Yields a weighted-random selection of file-like objects. """
     # Quick data type sanity.
     assert groups and isinstance(groups, (list, tuple))
@@ -193,6 +193,7 @@ def files_varied(groups):
                 num -= file_set.num
                 continue
 
+            if not unlimited:
                 total_num -= 1
                 file_set.num -= 1