• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1#!/usr/bin/env python3
2#coding: utf-8
3
4"""
5Copyright (c) 2021 Huawei Device Co., Ltd.
6Licensed under the Apache License, Version 2.0 (the "License");
7you may not use this file except in compliance with the License.
8You may obtain a copy of the License at
9
10    http://www.apache.org/licenses/LICENSE-2.0
11
12Unless required by applicable law or agreed to in writing, software
13distributed under the License is distributed on an "AS IS" BASIS,
14WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
15See the License for the specific language governing permissions and
16limitations under the License.
17"""
18
19"""
20Description: run script
21    expect_output will get run result, expect_file will get print string
22"""
23
24import argparse
25import subprocess
26import time
27
28
29def parse_args():
30    """parse arguments."""
31    parser = argparse.ArgumentParser()
32    parser.add_argument('--script-file', help='execute script file')
33    parser.add_argument('--script-options', help='execute script options')
34    parser.add_argument('--script-args', help='args of script')
35    parser.add_argument('--expect-output', help='expect output')
36    parser.add_argument('--expect-file', help='expect file')
37    parser.add_argument('--env-path', help='LD_LIBRARY_PATH env')
38    args = parser.parse_args()
39    return args
40
41
42def judge_output(args):
43    """run testcase and judge is success or not."""
44    start_time = time.time()
45    cmd = input_args.script_file
46    if input_args.script_options:
47        cmd += input_args.script_options
48    if input_args.script_args:
49        cmd += " " + input_args.script_args
50    subp = subprocess.Popen(cmd, shell=True, stdout=subprocess.PIPE,
51        env={'LD_LIBRARY_PATH': str(input_args.env_path)})
52    try:
53        out, err = subp.communicate(timeout=300) # units: s
54    except subprocess.TimeoutExpired:
55        subp.kill()
56        out, err = subp.communicate()
57
58    if args.expect_output:
59        returncode = str(subp.returncode)
60        if returncode != args.expect_output:
61            print_str = out.decode('UTF-8')
62            print(print_str)
63            raise RuntimeError("Run [" + cmd + "] failed!")
64    elif args.expect_file:
65        with open(args.expect_file, mode='r') as file:
66            # skip license header
67            expect_output = ''.join(file.readlines()[13:])
68            file.close()
69            print_str = out.decode('UTF-8')
70            if print_str != expect_output:
71                raise RuntimeError("\n>>>>> Expect : [" + expect_output \
72                    + "]\n>>>>> But got: [" + print_str + "]")
73    else:
74        raise RuntimeError("Run [" + cmd + "] with no expect !")
75
76    print("Run [" + cmd + "] success!")
77    print("used: %.5f seconds" % (time.time() - start_time))
78
79
80if __name__ == '__main__':
81    input_args = parse_args()
82    judge_output(input_args)
83