optimize the fault notification test

[doctor.git] / doctor_tests / consumer / base.py

diff --git a/doctor_tests/consumer/base.py b/doctor_tests/consumer/base.py
index b6c4b34..e21b080 100644 (file)
--- a/doctor_tests/consumer/base.py
+++ b/doctor_tests/consumer/base.py
@@ -16,6 +16,15 @@ class BaseConsumer(object):
     def __init__(self, conf, log):
         self.conf = conf
         self.log = log
+        self._notified_time = None
+
+    @property
+    def notified_time(self):
+        return self._notified_time
+
+    @notified_time.setter
+    def notified_time(self, notified_time):
+        self._notified_time = notified_time
 
     @abc.abstractmethod
     def start(self):