Change Builder Names, MkIII

tools/tests/factory_configuration/verify_mapping.py

+# Copyright (c) 2013 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.
+
+""" Runs a series of tests on a mapping file. """
+
+
+import json
+import os
+import sys
+
+
+EXPECTED_DIR = os.path.join(os.path.dirname(__file__), 'expected')
+
+
+def VerifyMapping(old_name, new_name, slaves_cfg, mapping):
+  try:
+    assert os.path.isfile(os.path.join(EXPECTED_DIR, old_name))
+    if old_name.endswith('Trybot'):
+      assert new_name.endswith('Trybot')
+    else:
+      assert new_name in slaves_cfg
+
+    # Assert no duplicate values
+    match_count = 0
+    for val in mapping.itervalues():
+      if val == new_name:
+        match_count += 1
+    assert match_count == 1
+
+    # ADD YOUR ASSERTIONS HERE
+
+  except Exception:
+    print 'Failed assertion for (%s to %s)' % (old_name, new_name)
+    raise
+
+
+def main(argv):
+  if len(argv) != 2:
+    raise Exception('Invalid arguments given; you must provide a mapping file.')
+  with open(argv[1]) as f:
+    mapping = json.load(f)
+
+  slaves_cfg_file = os.path.join('master', 'slaves.cfg')
+  with open(slaves_cfg_file) as f:
+    slaves_cfg = f.read()
+
+  for old_name, new_name in mapping.iteritems():
+    VerifyMapping(old_name, new_name, slaves_cfg, mapping)
+
+
+if '__main__' == __name__:
+  sys.exit(main(sys.argv))