summaryrefslogtreecommitdiffstats
path: root/tests/scenario/common.py
blob: e880e8b23d244b0347ae799836f6be1696f5f2ec (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
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)