catapult/experimental/bisect_lib/fetch_intervening_revisions_test.py

#!/usr/bin/env python
# Copyright 2015 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
import sys
import unittest

_CATAPULT_PATH = os.path.abspath(os.path.join(
    os.path.dirname(__file__), os.path.pardir, os.path.pardir))
sys.path.insert(0, os.path.join(_CATAPULT_PATH, 'third_party', 'mock'))

import mock

from bisect_lib import fetch_intervening_revisions

_TEST_DATA = os.path.join(os.path.dirname(__file__), 'test_data')


class FetchInterveningRevisionsTest(unittest.TestCase):

  def testFetchInterveningRevisions(self):
    response = open(os.path.join(_TEST_DATA, 'MOCK_RANGE_RESPONSE_1'))
    with mock.patch('urllib2.urlopen', mock.MagicMock(return_value=response)):
      revs = fetch_intervening_revisions.FetchInterveningRevisions(
          '53fc07eb478520a80af6bf8b62be259bb55db0f1',
          'c89130e28fd01062104e1be7f3a6fc3abbb80ca9',
          depot_name='chromium')
    self.assertEqual(
        revs, [
            ('32ce3b13924d84004a3e05c35942626cbe93cbbd', '356382'),
            ('07a6d9854efab6677b880defa924758334cfd47d', '356383'),
            ('22e49fb496d6ffa122c470f6071d47ccb4ccb672', '356384'),
            ('5dbc149bebecea186b693b3d780b6965eeffed0f', '356385'),
            ('ebd5f102ee89a4be5c98815c02c444fbf2b6b040', '356386'),
            ('84f6037e951c21a3b00bd3ddd034f258da6839b5', '356387'),
            ('48c1471f1f503246dd66753a4c7588d77282d2df', '356388'),
            ('66aeb2b7084850d09f3fccc7d7467b57e4da1882', '356389'),
            ('01542ac6d0fbec6aa78e33e6c7ec49a582072ea9', '356390'),
            ('8414732168a8867a5d6bd45eaade68a5820a9e34', '356391'),
            ('4f81be50501fbc02d7e44df0d56032e5885e19b6', '356392'),
            ('7bd1741893bd4e233b5562a6926d7e395d558343', '356393'),
            ('ee261f306c3c66e96339aa1026d62a6d953302fe', '356394'),
            ('f1c777e3f97a16cc6a3aa922a23602fa59412989', '356395'),
            ('8fcc8af20a3d41b0512e3b1486e4dc7de528a72b', '356396'),
            ('3861789af25e2d3502f0fb7080da5785d31308aa', '356397'),
            ('6feaa73a54d0515ad2940709161ca0a5ad91d1f8', '356398'),
            ('2e93263dc74f0496100435e1fd7232e9e8323af0', '356399')
        ])

  def testFetchInterveningRevisionsPagination(self):

    def MockUrlopen(url):
      if 's=' not in url:
        return open(os.path.join(_TEST_DATA, 'MOCK_RANGE_RESPONSE_2_PAGE_1'))
      return open(os.path.join(_TEST_DATA, 'MOCK_RANGE_RESPONSE_2_PAGE_2'))

    with mock.patch('urllib2.urlopen', MockUrlopen):
      revs = fetch_intervening_revisions.FetchInterveningRevisions(
          '7bd1741893bd4e233b5562a6926d7e395d558343',
          '3861789af25e2d3502f0fb7080da5785d31308aa',
          depot_name='chromium')
    self.assertEqual(
        revs, [
            ('ee261f306c3c66e96339aa1026d62a6d953302fe', '356394'),
            ('f1c777e3f97a16cc6a3aa922a23602fa59412989', '356395'),
            ('8fcc8af20a3d41b0512e3b1486e4dc7de528a72b', '356396'),
        ])


if __name__ == '__main__':
  unittest.main()