testsetup.py

import os
import subprocess
import tarfile
import shutil
from subprocess import PIPE, Popen, call
from preparetest import TestParpare


class TestSetup(TestParpare):
    '''
        Test project settings
    '''

    def decompressfile(self, filepath, filename, homepath):
        fileformat = filepath.split('.')[-1]
        installdir = os.path.join(homepath, 'tmp')
        tarfilepath = self.mkinstalldir(installdir)
        print "test is %s" % tarfilepath
        tarfilepath = os.path.join(homepath, tarfilepath)
        tmpfilepath = os.path.join(tarfilepath, 'packet')
        try:
            filedecompress = tarfile.open("%s" % filepath,
                                          "r:%s" % fileformat)
            filedecompress.extractall(path=tmpfilepath)
            filedecompress.close()
        except IOError as err:
            print "%s decompressfile  error : %s" % (filename, err)
        except tarfile.CompressionError as err:
            print "Decompressfile faild, %s" % err
        bindir = os.path.join(tarfilepath, filename)
        call('mv %s/* %s' % (tmpfilepath, bindir), shell=True)
        return bindir

    def _runsh(self, shcmd):
        try:
            call('sh %s' % shcmd, shell=True)
        except:
            pass

    def _configure(self, args):
        try:
            configure = Popen('./configure %s' % args, stdout=PIPE, stderr=PIPE, shell=True)
            stdout = configure.communicate()[0]
            print stdout
        except:
            pass

    def _make(self, args):
        try:
            make = Popen('make clean', stdout=PIPE, stderr=PIPE,shell=True)
            stdout = make.communicate()[0]
            print stdout
        except:
            pass
        try:
            make = Popen('make %s' % args, stdout=PIPE, stderr=PIPE, shell=True)
            stdout, stderr = make.communicate()
            print stdout, stderr
        except:
            pass
        try:
            make = Popen('make install %s' % args, stdout=PIPE, stderr=PIPE, shell=True)
            stdout, stderr = make.communicate()[0]
            print stdout
        except:
            pass

    def _rmfile(self, args):
        try:
            os.remove(args)
        except OSError:
            pass

    def _cpfile(self, src, dst):
        try:
            shutil.copy('%s' % src, '%s' % dst)
            print "mv ok"
        except:
            pass

    def pacagemanger(self, *args):
        '''
           Check test tool based on
        '''
        for arg in args:
            try:
                retcode = call('which %s' % arg, shell=True)
                if retcode == 0:
                    return arg
                    break
            except OSError:
                pass

    def packageinstall(defectlist):
        packagetool = self.pacagemanger('dnf', 'yum', 'apt-get')
        for defect in defectlist:
            call('%s -y install %s' % (packagetool, defect), shell=True)

# testcase
# test = testsetup()
# test.packageinstall()
# test.toolinstall('sysbench-0..12', 'tmp')