Add log and link aliasing to BuildBot.

scripts/master/unittests/annotator_test.py

 #!/usr/bin/env python
 # Copyright (c) 2012 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.
 
 """ Source file for annotated command testcases."""
 
 import os
 import time
 import unittest
@@ skipping 53 matching lines @@
 
   def finish(self):
     self.finished = True
 
 
 class FakeBuildstepStatus(mock.Mock):
   def __init__(self, name, build):
     mock.Mock.__init__(self)
     self.name = name
     self.urls = {}
+    self.aliases = {}
     self.build = build
     self.text = None
     self.step = None
     self.logs = []
     self.started = False
     self.finished = False
 
   def stepStarted(self):
     self.started = True
 
   def isStarted(self):
     return self.started
 
   def setText(self, text):
     self.text = text
 
   def setText2(self, text):
     self.text = text
 
   def getBuild(self):
     return self.build
 
   def getURLs(self):
     return self.urls.copy()
 
+  def getAliases(self):
+    return self.aliases.copy()
+
   def addURL(self, label, url):
     self.urls[label] = url
 
+  def addAlias(self, label, url, text=None):
+    self.aliases.setdefault(label, []).append((text, url))
+
   def addLog(self, log):
     l = FakeLog(log)
     self.logs.append(l)
     return l
 
   def getLogs(self):
     return self.logs
 
   def getLog(self, log):
     candidates = [x for x in self.logs if x.name == log]
@@ skipping 98 matching lines @@
     self.handleOutputLine('@@@BUILD_STEP step@@@')
 
     statuses = [x['status'] for x in self.step.script_observer.sections]
 
     self.assertEquals(statuses, [builder.EXCEPTION, builder.SUCCESS])
     self.assertEquals(self.step.script_observer.annotate_status,
                       builder.EXCEPTION)
 
   def testStepLink(self):
     self.handleOutputLine('@@@STEP_LINK@label@http://localhost/@@@')
-    testurls = [('label', 'http://localhost/')]
+    testurls = [('label', 'http://localhost/', None)]
     testurl_hash = {'label': 'http://localhost/'}
 
     annotatedLinks = [x['links'] for x in self.step.script_observer.sections]
     stepLinks = [x['step'].getURLs() for x in
                  self.step.script_observer.sections]
 
     self.assertEquals(annotatedLinks, [testurls])
     self.assertEquals(stepLinks, [testurl_hash])
 
+  def testStepAlias(self):
+    self.handleOutputLine('@@@STEP_LINK@alias-->label@http://localhost/@@@')
+    testurls = [('label', 'http://localhost/', 'alias')]
+    testalias_hash = {'label': [('alias', 'http://localhost/')]}
+
+    annotatedLinks = [x['links'] for x in self.step.script_observer.sections]
+    stepAliases = [x['step'].getAliases() for x in
+                   self.step.script_observer.sections]
+
+    self.assertEquals(annotatedLinks, [testurls])
+    self.assertEquals(stepAliases, [testalias_hash])
+
   def testStepWarning(self):
     self.handleOutputLine('@@@STEP_WARNINGS@@@')
     self.handleOutputLine('@@@BUILD_STEP step@@@')
 
     statuses = [x['status'] for x in self.step.script_observer.sections]
 
     self.assertEquals(statuses, [builder.WARNINGS, builder.SUCCESS])
     self.assertEquals(self.step.script_observer.annotate_status,
                       builder.WARNINGS)
 
@@ skipping 315 matching lines @@
     self.handleOutputLine('@@@STEP_CURSOR annotated_steps@@@')
     self.handleOutputLine('@@@STEP_CLOSED@@@')
 
     self.step.script_observer.handleReturnCode(0)
 
     self.assertNothingToWait()
 
 
 if __name__ == '__main__':
   unittest.main()