fix package path and move files under doctor_tests

[doctor.git] / doctor_tests / scenario / common.py

diff --git a/doctor_tests/scenario/common.py b/doctor_tests/scenario/common.py
new file mode 100644 (file)
index 0000000..a5cbe48
--- /dev/null
+++ b/doctor_tests/scenario/common.py
@@ -0,0 +1,29 @@
+##############################################################################
+# 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 sys
+
+from doctor_tests.common.utils import match_rep_in_file
+
+
+def calculate_notification_time():
+    log_file = '{0}/{1}'.format(sys.path[0], 'doctor.log')
+
+    reg = '(?<=doctor monitor detected at )\d+.\d+'
+    result = match_rep_in_file(reg, log_file)
+    if not result:
+        raise Exception('Can not match detected time')
+    detected = result.group(0)
+
+    reg = '(?<=doctor consumer notified at )\d+.\d+'
+    result = match_rep_in_file(reg, log_file)
+    if not result:
+        raise Exception('Can not match notified time')
+    notified = result.group(0)
+
+    return float(notified) - float(detected)