1#!/usr/bin/env python
2# Copyright 2015 The Chromium Authors. All rights reserved.
3# Use of this source code is governed by a BSD-style license that can be
4# found in the LICENSE file.
5
6import os
7import sys
8import unittest
9
10_CATAPULT_PATH = os.path.abspath(os.path.join(
11    os.path.dirname(__file__), os.path.pardir, os.path.pardir))
12sys.path.insert(0, os.path.join(_CATAPULT_PATH, 'third_party', 'mock'))
13
14import mock
15
16from bisect_lib import fetch_intervening_revisions
17
18_TEST_DATA = os.path.join(os.path.dirname(__file__), 'test_data')
19
20
21class FetchInterveningRevisionsTest(unittest.TestCase):
22
23  def testFetchInterveningRevisions(self):
24    response = open(os.path.join(_TEST_DATA, 'MOCK_RANGE_RESPONSE_1'))
25    with mock.patch('urllib2.urlopen', mock.MagicMock(return_value=response)):
26      revs = fetch_intervening_revisions.FetchInterveningRevisions(
27          '53fc07eb478520a80af6bf8b62be259bb55db0f1',
28          'c89130e28fd01062104e1be7f3a6fc3abbb80ca9',
29          depot_name='chromium')
30    self.assertEqual(
31        revs, [
32            ('32ce3b13924d84004a3e05c35942626cbe93cbbd', '356382'),
33            ('07a6d9854efab6677b880defa924758334cfd47d', '356383'),
34            ('22e49fb496d6ffa122c470f6071d47ccb4ccb672', '356384'),
35            ('5dbc149bebecea186b693b3d780b6965eeffed0f', '356385'),
36            ('ebd5f102ee89a4be5c98815c02c444fbf2b6b040', '356386'),
37            ('84f6037e951c21a3b00bd3ddd034f258da6839b5', '356387'),
38            ('48c1471f1f503246dd66753a4c7588d77282d2df', '356388'),
39            ('66aeb2b7084850d09f3fccc7d7467b57e4da1882', '356389'),
40            ('01542ac6d0fbec6aa78e33e6c7ec49a582072ea9', '356390'),
41            ('8414732168a8867a5d6bd45eaade68a5820a9e34', '356391'),
42            ('4f81be50501fbc02d7e44df0d56032e5885e19b6', '356392'),
43            ('7bd1741893bd4e233b5562a6926d7e395d558343', '356393'),
44            ('ee261f306c3c66e96339aa1026d62a6d953302fe', '356394'),
45            ('f1c777e3f97a16cc6a3aa922a23602fa59412989', '356395'),
46            ('8fcc8af20a3d41b0512e3b1486e4dc7de528a72b', '356396'),
47            ('3861789af25e2d3502f0fb7080da5785d31308aa', '356397'),
48            ('6feaa73a54d0515ad2940709161ca0a5ad91d1f8', '356398'),
49            ('2e93263dc74f0496100435e1fd7232e9e8323af0', '356399')
50        ])
51
52  def testFetchInterveningRevisionsPagination(self):
53
54    def MockUrlopen(url):
55      if 's=' not in url:
56        return open(os.path.join(_TEST_DATA, 'MOCK_RANGE_RESPONSE_2_PAGE_1'))
57      return open(os.path.join(_TEST_DATA, 'MOCK_RANGE_RESPONSE_2_PAGE_2'))
58
59    with mock.patch('urllib2.urlopen', MockUrlopen):
60      revs = fetch_intervening_revisions.FetchInterveningRevisions(
61          '7bd1741893bd4e233b5562a6926d7e395d558343',
62          '3861789af25e2d3502f0fb7080da5785d31308aa',
63          depot_name='chromium')
64    self.assertEqual(
65        revs, [
66            ('ee261f306c3c66e96339aa1026d62a6d953302fe', '356394'),
67            ('f1c777e3f97a16cc6a3aa922a23602fa59412989', '356395'),
68            ('8fcc8af20a3d41b0512e3b1486e4dc7de528a72b', '356396'),
69        ])
70
71
72if __name__ == '__main__':
73  unittest.main()
74