Commit 6b4801ce authored by Leif Walsh's avatar Leif Walsh Committed by Yoni Fogel
Browse files

refactoring stress test thing some more 

git-svn-id: file:///svn/toku/tokudb@42142 c7de825b-a66e-492c-adef-691d508d4ae1
parent 94f2e9c5

scripts/run.stress-tests.py

+16 −19
Original line number Diff line number Diff line
@@ -52,12 +52,6 @@ def setlimits(): 
    setrlimit(RLIMIT_CORE, (-1, -1))

    os.nice(7)



def timestr(timeval):

    if timeval == 0:

        return 'None'

    else:

        return time.ctime(timeval)



class TestFailure(Exception):

    pass
@@ -98,12 +92,18 @@ class TestRunnerBase(object): 
    def __str__(self):

        return 'TestRunner<%s, %d, %d>' % (self.execf, self.tsize, self.csize)



    def infostr(self):

        return '%(execf)s\t%(rev)s\t%(tsize)d\t%(csize)d\t%(num_ptquery)d\t%(num_update)d\t%(time)d' % self



    def __getitem__(self, k):

        return self.__getattribute__(k)



    def infostr(self):

        return '\t'.join(['%(execf)s',

                          '%(rev)s',

                          '%(tsize)d',

                          '%(csize)d',

                          '%(num_ptquery)d',

                          '%(num_update)d',

                          '%(time)d']) % self



    @property

    def time(self):

        if self.times[0] != 0 and self.times[1] != 0:
@@ -125,6 +125,10 @@ class TestRunnerBase(object): 
        else:

            return randrange(16)



    @property

    def envdir(self):

        return os.path.join(self.rundir, 'envdir')



    @property

    def prepareloc(self):

        preparename = 'dir.%(execf)s-%(tsize)d-%(csize)d' % self
@@ -133,21 +137,16 @@ class TestRunnerBase(object): 
    def prepare(self):

        if os.path.isdir(self.prepareloc):

            debug('%s found existing environment.', self)

            copytree(self.prepareloc, os.path.join(self.rundir, self.envdir))

            copytree(self.prepareloc, self.envdir)

        else:

            debug('%s preparing an environment.', self)

            self.run_prepare()

            debug('%s copying environment to %s.', self, self.prepareloc)

            copytree(os.path.join(self.rundir, self.envdir), self.prepareloc)



    def delete_prepared_env(self):

        if os.path.isdir(self.prepareloc):

            rmtree(self.prepareloc)

            copytree(self.envdir, self.prepareloc)



    def run(self):

        srctests = os.path.join(self.tokudb, 'src', 'tests')

        self.rundir = mkdtemp(dir=srctests)

        self.envdir = '%(execf)s-%(tsize)d-%(csize)d-%(num_ptquery)d-%(num_update)d' % self



        try:

            outname = os.path.join(self.rundir, 'output.txt')
@@ -209,7 +208,7 @@ class TestRunnerBase(object): 


    def prepareargs(self):

        return ['-v',

                '--envdir', self.envdir,

                '--envdir', 'envdir',

                '--num_elements', str(self.tsize),

                '--cachetable_size', str(self.csize)]
@@ -424,8 +423,6 @@ def main(opts): 
            if scheduler.error is not None:

                error('Scheduler reported an error.')

                raise scheduler.error

            for runner in runners:

                runner.delete_prepared_env()

            rebuild(opts.tokudb, opts.cc)

            rev = revfor(opts.tokudb)

            for runner in runners: