X-Git-Url: http://git.onelab.eu/?a=blobdiff_plain;f=system%2FTestSsh.py;h=c9d1e25a33cab6d3816f0a1194ab4b291539a531;hb=0cb2702996186784eb841109f41dcd99c9d0c3c2;hp=6d721fa7d7863f2615eb6ffce4b601feabc7011d;hpb=d53f90e066e8afbccaa5f1d5ae8b2fab2db01d66;p=tests.git diff --git a/system/TestSsh.py b/system/TestSsh.py index 6d721fa..c9d1e25 100644 --- a/system/TestSsh.py +++ b/system/TestSsh.py @@ -1,5 +1,5 @@ -# -# Thierry Parmentelat - INRIA +# Thierry Parmentelat +# Copyright (C) 2010 INRIA # # class for issuing commands on a box, either local or remote # @@ -15,6 +15,7 @@ # (*) or copy a local file into the remote 'buildname' # +import sys import os.path import utils import shutil @@ -47,16 +48,22 @@ class TestSsh: utils.header("WARNING : something wrong in is_local_hostname with hostname=%s"%hostname) return False - def __init__(self,hostname,buildname=None,key=None, username=None): + # some boxes have their working space in user's homedir (/root), + # some others in a dedicated area with max. space (/vservers) + # when root is not specified we use the homedir + def __init__(self,hostname,buildname=None,key=None, username=None,unknown_host=True, root=None): self.hostname=hostname self.buildname=buildname self.key=key self.username=username + self.unknown_host=unknown_host + self.root=root def is_local(self): return TestSsh.is_local_hostname(self.hostname) - std_options="-o BatchMode=yes -o StrictHostKeyChecking=no -o CheckHostIP=no -o ConnectTimeout=5 -o UserKnownHostsFile=/dev/null " + std_options="-o BatchMode=yes -o StrictHostKeyChecking=no -o CheckHostIP=no -o ConnectTimeout=5 " + unknown_option="-o UserKnownHostsFile=/dev/null " def key_part (self): if not self.key: @@ -70,38 +77,75 @@ class TestSsh: return "%s@%s"%(self.username,self.hostname) # command gets run on the right box - def actual_command (self, command,keep_stdin=False): + def actual_command (self, command, keep_stdin=False, dry_run=False): if self.is_local(): return command ssh_command = "ssh " - if not keep_stdin: - ssh_command += "-n " - ssh_command += TestSsh.std_options + if not dry_run: + if not keep_stdin: + ssh_command += "-n " + ssh_command += TestSsh.std_options + if self.unknown_host: ssh_command += TestSsh.unknown_option ssh_command += self.key_part() ssh_command += "%s %s" %(self.hostname_part(),TestSsh.backslash_shell_specials(command)) return ssh_command - def run(self, command,background=False): - local_command = self.actual_command(command) - return utils.system(local_command,background) - - def clean_dir (self,dirname): + # same in argv form + def actual_argv (self, argv,keep_stdin=False, dry_run=False): if self.is_local(): + return argv + ssh_argv=[] + ssh_argv.append('ssh') + if not dry_run: + if not keep_stdin: ssh_argv.append('-n') + ssh_argv += TestSsh.std_options.split() + if self.unknown_host: ssh_argv += TestSsh.unknown_option.split() + ssh_argv += self.key_part().split() + ssh_argv.append(self.hostname_part()) + ssh_argv += argv + return ssh_argv + + def header (self,message): + if not message: return + print "===============",message + sys.stdout.flush() + + def run(self, command,message=None,background=False,dry_run=False): + local_command = self.actual_command(command, dry_run=dry_run) + if dry_run: + utils.header("DRY RUN " + local_command) return 0 - return self.run("rm -rf %s"%dirname) + else: + self.header(message) + return utils.system(local_command,background) - def mkdir (self,dirname=None): + def run_in_buildname (self,command, background=False, dry_run=False): + if self.is_local(): + return utils.system(command,background) + self.create_buildname_once(dry_run) + return self.run("cd %s ; %s"%(self.fullname(self.buildname),command), + background=background, dry_run=dry_run) + + def fullname (self,dirname): + if self.root==None: return dirname + else: return os.path.join(self.root,dirname) + + def mkdir (self,dirname=None,abs=False,dry_run=False): if self.is_local(): if dirname: return os.path.mkdir(dirname) return 0 - if dirname: - dirname="%s/%s"%(self.buildname,dirname) - else: - dirname=self.buildname - return self.run("mkdir -p %s"%dirname) + # ab. paths remain as-is + if not abs: + if dirname: + dirname="%s/%s"%(self.buildname,dirname) + else: + dirname=self.buildname + dirname=self.fullname(dirname) + if dirname=='.': return + return self.run("mkdir -p %s"%dirname,dry_run=dry_run) - def rmdir (self,dirname=None): + def rmdir (self,dirname=None, dry_run=False): if self.is_local(): if dirname: return shutil.rmtree(dirname) @@ -110,34 +154,33 @@ class TestSsh: dirname="%s/%s"%(self.buildname,dirname) else: dirname=self.buildname - return self.run("rm -rf %s"%dirname) + dirname=self.fullname(dirname) + return self.run("rm -rf %s"%dirname, dry_run=dry_run) - def create_buildname_once (self): + def create_buildname_once (self, dry_run): if self.is_local(): return # create remote buildname on demand try: self.buildname_created except: - self.mkdir() + self.mkdir(dry_run=dry_run) self.buildname_created=True - def run_in_buildname (self,command, background=False): - if self.is_local(): - return utils.system(command,background) - self.create_buildname_once() - return self.run("cd %s ; %s"%(self.buildname,command),background) - - def copy (self,local_file,recursive=False): + def copy (self,local_file,recursive=False,dry_run=False): if self.is_local(): return 0 - self.create_buildname_once() + self.create_buildname_once(dry_run) scp_command="scp " - scp_command += TestSsh.std_options + if not dry_run: + scp_command += TestSsh.std_options if recursive: scp_command += "-r " scp_command += self.key_part() scp_command += "%s %s:%s/%s"%(local_file,self.hostname_part(), - self.buildname,os.path.basename(local_file) or ".") + self.fullname(self.buildname),os.path.basename(local_file) or ".") + if dry_run: + utils.header ("DRY RUN TestSsh.copy %s"%scp_command) + return True return utils.system(scp_command) def copy_abs (self,local_file,remote_file,recursive=False): @@ -155,14 +198,15 @@ class TestSsh: def copy_home (self, local_file, recursive=False): return self.copy_abs(local_file,os.path.basename(local_file),recursive) - def fetch (self, remote_file, local_file, recursive=False): + def fetch (self, remote_file, local_file, recursive=False, dry_run=False): if self.is_local(): command="cp " if recursive: command += "-r " command += "%s %s"%(remote_file,local_file) else: command="scp " - command += TestSsh.std_options + if not dry_run: + command += TestSsh.std_options if recursive: command += "-r " command += self.key_part() # absolute path - do not preprend buildname @@ -170,6 +214,7 @@ class TestSsh: remote_path=remote_file else: remote_path="%s/%s"%(self.buildname,remote_file) + remote_path=self.fullname(remote_path) command += "%s:%s %s"%(self.hostname_part(),remote_path,local_file) return utils.system(command)