Add buildbot state machine and master_manager tool.

infra/services/master_lifecycle/test/buildbot_state_test.py

+# Copyright 2015 The Chromium Authors. All rights reserved.
+# Use of this source code is governed by a BSD-style license that can be
+# found in the LICENSE file.
+
+from testing_support import auto_stub
+
+from infra.libs.buildbot import master
+from infra.libs.time_functions import timestamp
+from infra.services.master_lifecycle import buildbot_state
+
+class TestBuildbotState(auto_stub.TestCase):
+  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.desired_buildbot_state = 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 accepting_builds_handler(*_args):
+      return self.accepting_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_accepting_builds', accepting_builds_handler)
+
+  def _get_evidence(self):
+    evidence = buildbot_state.collect_evidence('fake_dir')
+    evidence['desired_buildbot_state'] = {
+        'desired_state': self.desired_buildbot_state,
+    }
+    return evidence
+
+  def _get_state(self):
+    return self.matcher.get_state(self._get_evidence())
+
+  def _get_execution_list(self):
+    return self.matcher.execution_list(self._get_evidence())
+
+  def testPatternMatcherIsSane(self):
+    self.assertTrue(self.matcher.is_correct)
+
+
+  #### Tests.
+
+  def testBuildbotIsOffline(self):
+    self.buildbot_is_running = False
+    state = self._get_state()['buildbot']
+    self.assertEqual(state, 'offline')
+
+  def testBuildbotIsBooting(self):
+    self.utcnow = self.last_boot + 4 * 60
+    self.accepting_builds = None
+    state = self._get_state()['buildbot']
+    self.assertEqual(state, 'booting')
+
+  def testBuildbotIsRunning(self):
+    state = self._get_state()['buildbot']
+    self.assertEqual(state, 'started_serving_accept')
+
+  def testBuildbotIsDraining(self):
+    self.accepting_builds = False
+    self.utcnow = self.last_no_new_builds + 4 * 60
+    state = self._get_state()['buildbot']
+    self.assertEqual(state, 'started_serving_not_accept')
+
+  def testBuildbotIsDrained(self):
+    self.accepting_builds = False
+    state = self._get_state()['buildbot']
+    self.assertEqual(state, 'started_serving_not_accept_drained')
+
+  def testBuildbotIsCrashing(self):
+    self.accepting_builds = None
+    state = self._get_state()['buildbot']
+    self.assertEqual(state, 'started_not_serving')
+
+  def testDesiredUpToDate(self):
+    state = self._get_state()['desired_buildbot_state']
+    self.assertEqual(state, 'up_to_date')
+
+  def testDesiredOffline(self):
+    self.desired_buildbot_state = None
+    state = self._get_state()['desired_buildbot_state']
+    self.assertEqual(state, 'offline')
+
+  def testDesiredFuture(self):
+    self.desired_buildbot_state = 3000
+    state = self._get_state()['desired_buildbot_state']
+    self.assertEqual(state, 'future')
+
+  def testDesiredReboot(self):
+    self.desired_buildbot_state = 1100
+    state = self._get_state()['desired_buildbot_state']
+    self.assertEqual(state, 'needs_reboot')
+
+  def testNoLastBoot(self):
+    self.last_boot = None
+    state = self._get_state()['desired_buildbot_state']
+    self.assertEqual(state, 'needs_reboot')
+
+  def testOfflineStaysOffline(self):
+    self.desired_buildbot_state = None
+    self.buildbot_is_running = False
+    _, _, execution_list = self._get_execution_list()
+    self.assertEqual(execution_list, [])
+
+  def testRestartKickedOff(self):
+    self.desired_buildbot_state = 1100
+    _, _, execution_list = self._get_execution_list()
+    self.assertEqual(execution_list, [master.MakeNoNewBuilds])
+
+  def testTurnDown(self):
+    self.desired_buildbot_state = None
+    self.accepting_builds = False
+    _, _, execution_list = self._get_execution_list()
+    self.assertEqual(execution_list, [master.MakeStop])
+
+  def testStartUp(self):
+    self.buildbot_is_running = False
+    _, _, execution_list = self._get_execution_list()
+    self.assertEqual(execution_list, [master.GclientSync, master.MakeStart])
+
+  def testRestart(self):
+    self.accepting_builds = False
+    _, _, execution_list = self._get_execution_list()
+    self.assertEqual(execution_list, [
+      master.GclientSync,
+      master.MakeStop,
+      master.MakeWait,
+      master.MakeStart])