[Findit] Retrieve recent builds of a given builder.

appengine/findit/waterfall/test/buildbot_test.py

 # Copyright 2014 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 datetime import datetime
 import mock
 import os
 import unittest
 
 from common.retry_http_client import RetryHttpClient
@@ skipping 17 matching lines @@
 
   def _Post(self, *_):  # pragma: no cover
     pass
 
   def _Put(self, *_):  # pragma: no cover
     pass
 
 
 class BuildBotTest(unittest.TestCase):
 
+  def testFailToGetRecentCompletedBuilds(self):
+    mocked_http_client = mock.Mock()
+    mocked_http_client.Get.side_effect = [(404, '404'), None]
+    self.assertEqual(
+        [], buildbot.GetRecentCompletedBuilds('m', 'b', mocked_http_client))
+    mocked_http_client.assert_has_calls(
+        mock.call('https://build.chromium.org/p/m/json/builders/'))
+
+  def testGetRecentCompletedBuilds(self):
+    builders_json_data = """
+    {
+      "b": {
+        "cachedBuilds": [33, 32, 34, 35],
+        "currentBuilds": [35]
+      }
+    }
+    """
+    mocked_http_client = mock.Mock()
+    mocked_http_client.Get.return_value = (200, builders_json_data)
+    self.assertEqual(
+        [34, 33, 32],
+        buildbot.GetRecentCompletedBuilds('m', 'b', mocked_http_client))
+    mocked_http_client.assert_has_calls(
+        mock.call('https://build.chromium.org/p/m/json/builders/'))
+
   def testGetMasternameFromUrl(self):
     cases = {
         None: None,
         '': None,
         'https://unknown.host/p/chromium': None,
         'http://build.chromium.org/p/chromium': 'chromium',
         'http://build.chromium.org/p/chromium/builders/Linux': 'chromium',
     }
 
     for url, expected_result in cases.iteritems():
@@ skipping 368 matching lines @@
 
     expected_blame_list = [
         '449cdbd05616de91fcf7e8b4282e300336d6d7c5',
         '6addffac2601ab1083a55d085847d9bf8e66da02'
     ]
 
     build_info = buildbot.ExtractBuildInfo(
         master_name, builder_name, build_number, build_data)
 
     self.assertEqual(expected_blame_list, build_info.blame_list)