blob: c123c16b0d5e44e15629907a703fa9ea7db9e326 [file] [log] [blame]
# Copyright 2017 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 unittest
from libs import time_util
from libs.gitiles.gitiles_repository import GitilesRepository
from libs.http import retry_http_client
class MockHttpClient(retry_http_client.RetryHttpClient): # pragma: no cover.
def __init__(self, response_for_url=None):
super(MockHttpClient, self).__init__()
self.response_for_url = response_for_url or {}
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', {}
return 200, response, {}
def _Post(self, *_):
def _Put(self, *_):
class BaseTestCase(unittest.TestCase): # pragma: no cover.
def MockUTCNow(self, mocked_utcnow):
"""Mocks utcnow with the given value for testing."""
self.mock(time_util, 'GetUTCNow', lambda: mocked_utcnow)
def GetMockHttpClient(self, response_for_url=None):
"""Returns mocked http client class."""
return MockHttpClient(response_for_url or {})
def GetMockRepoFactory(self, response_for_url=None):
"""Returns mocked repository factory."""
return GitilesRepository.Factory(
self.GetMockHttpClient(response_for_url or {}))