Implemented state resetting on replay() for RunCounter and Orderer.

1 files changed, 6 insertions, 0 deletions

diff --git a/mocker.py b/mocker.py
index 71d3ac7..fe27892 100644
--- a/mocker.py
+++ b/mocker.py
@@ -1514,6 +1514,9 @@ class RunCounter(Task):
             self.max = max
         self._runs = 0
 
+    def replay(self):
+        self._runs = 0
+
     def run(self, path):
         self._runs += 1
         if self._runs > self.max:
@@ -1618,6 +1621,9 @@ class Orderer(Task):
         self._run = False 
         self._dependencies = []
 
+    def replay(self):
+        self._run = False
+
     def run(self, path):
         self._run = True