from autobot.process.factory import VirtualenvFactory
from buildbot.steps.shell import ShellCommand
from buildbot.steps.shell import WithProperties

class TestMozbase(VirtualenvFactory):
    """
    Base utilties for mozilla test harnesses:
    https://github.com/mozilla/mozbase
    """

    sources = {'git': ['https://github.com/mozilla/mozbase'],
               'hg': []}

    def __init__(self, platform=None):
        VirtualenvFactory.__init__(self)

        # setup for development
        self.addStep(ShellCommand(command=[WithProperties('%(python)s'), 'setup_development.py'], workdir=WithProperties('%(virtualenv)s/src/mozbase')))

        # add %(scripts)s to $PATH
        self.addScripts()

        # run the tests
        self.addStep(ShellCommand(command=[WithProperties('%(python)s'), 'test.py'], workdir=WithProperties('%(virtualenv)s/src/mozbase'),
                     env={'PATH': WithProperties('%(PATH)s')}))