From 19f7ba75850c52e1ae163766b64b6d153e8d7e1b Mon Sep 17 00:00:00 2001 From: dongwenjuan Date: Mon, 14 Aug 2017 16:58:13 +0800 Subject: refactor failure inject JIRA: DOCTOR-116 Change-Id: I14deda4ccb47414cff139a522a9196b68e92500e Signed-off-by: dongwenjuan --- tests/scenario/common.py | 40 ++++++++++++++++++++++++++++++++++++++++ 1 file changed, 40 insertions(+) create mode 100644 tests/scenario/common.py (limited to 'tests/scenario/common.py') diff --git a/tests/scenario/common.py b/tests/scenario/common.py new file mode 100644 index 00000000..e880e8b2 --- /dev/null +++ b/tests/scenario/common.py @@ -0,0 +1,40 @@ +############################################################################## +# Copyright (c) 2017 ZTE Corporation and others. +# +# All rights reserved. This program and the accompanying materials +# are made available under the terms of the Apache License, Version 2.0 +# which accompanies this distribution, and is available at +# http://www.apache.org/licenses/LICENSE-2.0 +############################################################################## +import os +import re +import sys + + +def match_rep_in_file(regex, full_path): + if not os.path.isfile(full_path): + raise Exception('File(%s) does not exist' % full_path) + + with open(full_path, 'r') as file: + for line in file: + result = re.search(regex, line) + if result: + return result.group(0) + + return None + + +def calculate_notification_time(): + log_file = '{0}/{1}'.format(sys.path[0], 'doctor.log') + + reg = '(?<=doctor monitor detected at )\d+.\d+' + detected = match_rep_in_file(reg, log_file) + if not detected: + raise Exception('Can not find detected time') + + reg = '(?<=doctor consumer notified at )\d+.\d+' + notified = match_rep_in_file(reg, log_file) + if not notified: + raise Exception('Can not find notified time') + + return float(notified) - float(detected) \ No newline at end of file -- cgit 1.2.3-korg