Implemented Mocker.generate().

1 files changed, 6 insertions, 0 deletions

diff --git a/mocker.py b/mocker.py
index 1c05f92..b8582ec 100644
--- a/mocker.py
+++ b/mocker.py
@@ -460,6 +460,12 @@ class MockerBase(object):
         """
         self.call(lambda *args, **kwargs: value)
 
+    def generate(self, sequence):
+        def generate(*args, **kwargs):
+            for value in sequence:
+                yield value
+        self.call(generate)
+
     def throw(self, exception):
         """Make the last recorded event raise the given exception on replay.