diff options
author | 2017-08-14 16:58:13 +0800 | |
---|---|---|
committer | 2017-09-04 22:30:39 +0800 | |
commit | 19f7ba75850c52e1ae163766b64b6d153e8d7e1b (patch) | |
tree | f76b13d9d9092209788624609f65872ca8d0dea5 /tests/utils.py | |
parent | 836a8932d6c6a502980009b9578f0c6ecf64cb47 (diff) |
refactor failure inject
JIRA: DOCTOR-116
Change-Id: I14deda4ccb47414cff139a522a9196b68e92500e
Signed-off-by: dongwenjuan <dong.wenjuan@zte.com.cn>
Diffstat (limited to 'tests/utils.py')
-rw-r--r-- | tests/utils.py | 90 |
1 files changed, 0 insertions, 90 deletions
diff --git a/tests/utils.py b/tests/utils.py deleted file mode 100644 index fd8c4cd7..00000000 --- a/tests/utils.py +++ /dev/null @@ -1,90 +0,0 @@ -############################################################################## -# 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 json -import os -import paramiko - - -def load_json_file(full_path): - """Loads JSON from file - :param target_filename: - :return: - """ - if not os.path.isfile(full_path): - raise Exception('File(%s) does not exist' % full_path) - - with open(full_path, 'r') as file: - return json.load(file) - - -def write_json_file(full_path, data): - """write JSON from file - :param target_filename: - :return: - """ - - with open(full_path, 'w+') as file: - file.write(json.dumps(data)) - - -class SSHClient(object): - def __init__(self, ip, username, password=None, pkey=None, - key_filename=None, log=None, look_for_keys=False, - allow_agent=False): - self.client = paramiko.SSHClient() - self.client.load_system_host_keys() - self.client.set_missing_host_key_policy(paramiko.AutoAddPolicy()) - self.client.connect(ip, username=username, password=password, - pkey=pkey, key_filename=key_filename, - look_for_keys=look_for_keys, - allow_agent=allow_agent) - self.log = log - - def __del__(self): - self.client.close() - - def ssh(self, command): - if self.log: - self.log.debug("Executing: %s" % command) - stdin, stdout, stderr = self.client.exec_command(command) - ret = stdout.channel.recv_exit_status() - output = list() - for line in stdout.read().splitlines(): - output.append(line.decode('utf-8')) - if ret: - if self.log: - self.log.debug("*** FAILED to run command %s (%s)" % (command, ret)) - raise Exception( - "Unable to run \ncommand: %s\nret: %s" - % (command, ret)) - if self.log: - self.log.debug("*** SUCCESSFULLY run command %s" % command) - return ret, output - - def scp(self, source, dest, method='put'): - if self.log: - self.log.info("Copy %s -> %s" % (source, dest)) - ftp = self.client.open_sftp() - if method == 'put': - ftp.put(source, dest) - elif method == 'get': - ftp.get(source, dest) - ftp.close() - -def run_async(func): - from threading import Thread - from functools import wraps - - @wraps(func) - def async_func(*args, **kwargs): - thread = Thread(target=func, args=args, kwargs=kwargs) - thread.start() - return thread - - return async_func |