[Culprit-finder] Refactor findit testcase and move it to libs.

appengine/findit/libs/testcase.py

 # Copyright 2016 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.
 
 import os
 
 from testing_utils import testing

[stgao, 2016/11/23 20:24:53]

But this depends on App Engine local test setup, while libs is for non-appengine
code.

Ideally this should goes to gae_libs/ instead.

[Sharu Jiang, 2016/11/23 20:55:00]

Right, that makes sense, move it in another cl
https://codereview.chromium.org/2525003005/.

 
 from lib import time_util
+from libs.http import retry_http_client
 
 
-class FinditTestCase(testing.AppengineTestCase):  # pragma: no cover.
+class MockHttpClient(retry_http_client.RetryHttpClient):  # pragma: no cover.
+
+  def __init__(self, response_for_url=None):
+    super(MockHttpClient, self).__init__()
+    if response_for_url is None:
+      self.response_for_url = {}
+
+  def SetResponseForUrl(self, url, response):
+    self.response_for_url[url] = response
+
+  def GetBackoff(self, *_):
+    """Override to avoid sleep."""
+    return 0
+
+  def _Get(self, url, *_):
+    response = self.response_for_url.get(url)
+    if response is None:
+      return 404, 'Not Found'
+    else:
+      return 200, response
+
+  def _Post(self, *_):
+    pass
+
+  def _Put(self, *_):
+    pass
+
+
+class TestCase(testing.AppengineTestCase):  # pragma: no cover.
   # Setup the customized queues.
   taskqueue_stub_root_path = os.path.join(
     os.path.dirname(__file__), os.path.pardir)
 
   def MockPipeline(
       self, pipeline_class, result, expected_args, expected_kwargs=None):
     """Mocks a pipeline to return a value and asserts the expected parameters.
 
     Args:
       pipeline_class (class): The class of the pipeline to be mocked.
@@ skipping 12 matching lines @@
 
     self.mock(pipeline_class, 'run', Mocked_run)
 
   def MockUTCNow(self, mocked_utcnow):
     """Mocks utcnow with the given value for testing."""
     self.mock(time_util, 'GetUTCNow', lambda: mocked_utcnow)
 
   def MockUTCNowWithTimezone(self, mocked_utcnow):
     """Mocks utcnow with the given value for testing."""
     self.mock(time_util, 'GetUTCNowWithTimezone', lambda: mocked_utcnow)
+
+  def GetMockHttpClient(self, response_for_url=None):
+    """Return mocked http client class."""
+    return MockHttpClient(response_for_url)