optimize the fault notification test

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

diff --git a/doctor_tests/consumer/sample.py b/doctor_tests/consumer/sample.py
index d76a764..c7dcf4f 100644 (file)
--- a/doctor_tests/consumer/sample.py
+++ b/doctor_tests/consumer/sample.py
@@ -54,9 +54,11 @@ class ConsumerApp(Thread):
 
         @app.route('/failure', methods=['POST'])
         def event_posted():
-            self.log.info('doctor consumer notified at %s' % time.time())
-            self.log.info('sample consumer received data = %s' % request.data)
+            notified_time = time.time()
+            self.log.info('doctor consumer notified at %s' % notified_time)
+            self.consumer.notified_time = notified_time
             data = json.loads(request.data.decode('utf8'))
+            self.log.info('sample consumer received data = %s' % data)
             return 'OK'
 
         @app.route('/shutdown', methods=['POST'])