[Findit] Use unittest.mock for testing and create a shared method to mock pipeline.

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
 from waterfall import buildbot
 
 
 class DummyHttpClient(RetryHttpClient):
 
   def __init__(self, status_code, response_content):
@@ skipping 123 matching lines @@
         expected_stdio_log_url,
         buildbot.CreateGtestResultPath(
             master_name, builder_name, build_number, step_name))
 
   def testGetBuildDataFromArchiveSuccess(self):
     master_name = 'a'
     builder_name = 'b c'
     build_number = 1
     expected_url = ('https://chrome-build-extract.appspot.com/p/a/builders/'
                     'b%20c/builds/1?json=1')
-    http_client = DummyHttpClient(200, 'abc')
+    http_client = mock.create_autospec(RetryHttpClient)
+    http_client.Get.return_value = (200, 'abc')
     data = buildbot.GetBuildDataFromArchive(
         master_name, builder_name, build_number, http_client)
-    self.assertEqual(http_client.response_content, data)
-    self.assertEqual(1, len(http_client.requests))
-    self.assertEqual(expected_url, http_client.requests[0])
+    self.assertEqual('abc', data)
+    http_client.assert_has_calls([mock.call.Get(expected_url)])
 
   def testGetBuildDataFromArchiveFailure(self):
     master_name = 'a'
     builder_name = 'b c'
     build_number = 1
     expected_url = ('https://chrome-build-extract.appspot.com/p/a/builders/'
                     'b%20c/builds/1?json=1')
     http_client = DummyHttpClient(404, 'Not Found')
     data = buildbot.GetBuildDataFromArchive(
         master_name, builder_name, build_number, http_client)
@@ skipping 251 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)