adding max_amount to realistic.names generator

Set max_amount to limit the number of names yeilded by the generator.

diff --git a/s3tests/realistic.py b/s3tests/realistic.py
index 325113656643eae7c6840b720876f0983b27385a..458fa08beb9b5503a6acc6a896f1405a7bc4ccb2 100644 (file)
--- a/s3tests/realistic.py
+++ b/s3tests/realistic.py
@@ -137,12 +137,13 @@ def files(mean, stddev, seed=None):
                 break
         yield RandomContentFile(size=size, seed=rand.getrandbits(32))
 
-def names(mean, stddev, charset=None, seed=None):
+def names(mean, stddev, charset=None, seed=None, max_amount=None):
     """
     Yields strings that are somewhat plausible as file names, where
     the lenght of each filename follows the normal distribution with
     `mean` and `stddev`.
     """
+    count = 0
     if charset is None:
         charset = string.ascii_lowercase
     rand = random.Random(seed)
@@ -154,6 +155,11 @@ def names(mean, stddev, charset=None, seed=None):
         name = ''.join(rand.choice(charset) for _ in xrange(length))
         yield name
 
+        count += 1
+        if count == max_amount:
+            return
+
+
 def files_varied(groups, unlimited=False):
     """ Yields a weighted-random selection of file-like objects. """
     # Quick data type sanity.