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

class TestAutobotFactory(PythonSourceFactory):
  """
  factory to test autobot
  """
  def __init__(self, slaveport=7357, htmlport=7358):

    # setup the environment
    PythonSourceFactory.__init__(self, name='autobot',
                                 hg_sources=['http://k0s.org/mozilla/hg/autobot'])

    # find the necessary scripts
    self.findScript('create-autobot')
    self.findScript('buildbot')
    self.findScript('buildslave')

    # make sure the template creator actually does something
    self.addStep(ShellCommand(command=[WithProperties('%(create-autobot)s'),
                                       '--list-factories'],
                              description='create-autobot --list-factories'))

    # make a master, slave pair
    self.addStep(ShellCommand(command=[WithProperties('%(create-autobot)s'),
                                       '-f', '-', # all factories,
                                       '--slaveport', str(slaveport),
                                       '--htmlport', str(htmlport)
                                       'bot'],
                              description="make an autobot",
                              workdir=WithProperties('%(virtualenv)s'),
                              haltOnFailure=True))



    # make sure they start
    self.addStep(ShellCommand(command=[WithProperties('%(buildbot)s'),
                                       'start', 'master'],
                              workdir=WithProperties('%(virtualenv)s/bot'),
                              description='start build master'))
    self.addStep(ShellCommand(command=[WithProperties('%(buildslave)s'),
                                       'start', 'slave'],
                              workdir=WithProperties('%(virtualenv)s/bot'),
                              description='start build slave'))
    

    # try to access the waterfall
    self.addStep(ShellCommand(command=['curl',
                                       'http://localhost:%s/' % htmlport]))

    # could try to actually build, but we won't here

    # stop them!
    self.addStep(ShellCommand(command=[WithProperties('%(buildbot)s'),
                                       'stop', 'master'],
                              workdir=WithProperties('%(virtualenv)s/bot'),
                              description='stop build master'))
    self.addStep(ShellCommand(command=[WithProperties('%(buildslave)s'),
                                       'stop', 'slave'],
                              workdir=WithProperties('%(virtualenv)s/bot'),
                              description='stop build slave'))