summaryrefslogtreecommitdiffstats
path: root/tests/utils.py
diff options
context:
space:
mode:
authordongwenjuan <dong.wenjuan@zte.com.cn>2017-08-14 16:58:13 +0800
committerdongwenjuan <dong.wenjuan@zte.com.cn>2017-09-04 22:30:39 +0800
commit19f7ba75850c52e1ae163766b64b6d153e8d7e1b (patch)
treef76b13d9d9092209788624609f65872ca8d0dea5 /tests/utils.py
parent836a8932d6c6a502980009b9578f0c6ecf64cb47 (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.py90
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