def main(): # pragma: no cover
args = parse_args()
matchlist = buildbot_state.construct_pattern_matcher()
logger = logging.getLogger(__name__)
logs.add_handler(logger)
if args.list_all_states:
matchlist.print_all_states()
return 0
abs_master_directory = os.path.abspath(args.directory)
state_machine = partial(run_state_machine_pass, logger,
matchlist, abs_master_directory, args.emergency_file,
args.desired_state, args.transition_time_utc, args.enable_gclient_sync,
args.prod, args.connection_timeout, args.hostname)
if args.loop:
loop_opts = outer_loop.process_argparse_options(args)
outer_loop.loop(
state_machine, lambda: args.loop_sleep_secs, **loop_opts)
else:
return state_machine()
return 0
def setUp(self):
super(TestBuildbotState, self).setUp()
self.matcher = buildbot_state.construct_pattern_matcher()
self.utcnow = 2000
self.last_boot = 1000
self.last_no_new_builds = 1100
self.buildbot_is_running = True
self.accepting_builds = True
self.current_running_builds = 10
self.desired_buildbot_state = 'running'
self.desired_transition_time = 900
def utcnow_handler():
return self.utcnow
def last_boot_handler(*_args):
return self.last_boot
def last_no_new_builds_handler(*_args):
return self.last_no_new_builds
def buildbot_is_running_handler(*_args):
return self.buildbot_is_running
def buildstate_handler(*_args, **_kwargs):
return self.accepting_builds, self.current_running_builds
self.mock(timestamp, 'utcnow_ts', utcnow_handler)
self.mock(master, 'get_last_boot', last_boot_handler)
self.mock(master, 'get_last_no_new_builds', last_no_new_builds_handler)
self.mock(master, 'buildbot_is_running', buildbot_is_running_handler)
self.mock(master, 'get_buildstate', buildstate_handler)
def setUp(self):
super(TestBuildbotState, self).setUp()
self.matcher = buildbot_state.construct_pattern_matcher()
self.utcnow = 2000
self.last_boot = 1000
self.last_no_new_builds = 1100
self.buildbot_is_running = True
self.accepting_builds = True
self.current_running_builds = set(('test', d) for d in xrange(10))
self.desired_buildbot_state = 'running'
self.desired_transition_time = 900
self.builder_filters = None
def utcnow_handler():
return self.utcnow
def last_boot_handler(*_args):
return self.last_boot
def last_no_new_builds_handler(*_args):
return self.last_no_new_builds
def buildbot_is_running_handler(*_args):
return self.buildbot_is_running
def buildstate_handler(*_args, **_kwargs):
return self.accepting_builds, self.current_running_builds
self.mock(timestamp, 'utcnow_ts', utcnow_handler)
self.mock(master, 'get_last_boot', last_boot_handler)
self.mock(master, 'get_last_no_new_builds', last_no_new_builds_handler)
self.mock(master, 'buildbot_is_running', buildbot_is_running_handler)
self.mock( master, 'get_buildstate', buildstate_handler)
def main(): # pragma: no cover
args = parse_args()
matchlist = buildbot_state.construct_pattern_matcher()
logger = logging.getLogger(__name__)
logs.add_handler(logger)
if args.list_all_states:
matchlist.print_all_states()
return 0
abs_master_directory = os.path.abspath(args.directory)
state_machine = partial(run_state_machine_pass, logger, matchlist,
abs_master_directory, args.emergency_file,
args.desired_state, args.transition_time_utc,
args.enable_gclient_sync, args.prod,
args.connection_timeout, args.hostname)
if args.loop:
loop_opts = outer_loop.process_argparse_options(args)
outer_loop.loop(state_machine, lambda: args.loop_sleep_secs,
**loop_opts)
else:
return state_machine()
return 0
