def test_normal(self):
"""Test when there's no end revision."""
self.mock.get_start_and_end_revision.return_value = (1, 100)
start, end = impact_task.get_start_and_end_revision('123:456', 'job')
self.assertEqual(1, start)
self.assertEqual(100, end)
self.mock.get_start_and_end_revision.assert_has_calls(
[mock.call('123:456')])
self.mock.get_component_range_list.assert_has_calls([])
def test_android(self):
"""Test android."""
self.mock.get_start_and_end_revision.side_effect = [(1, 100), (9, 90)]
self.mock.get_component_range_list.return_value = [{
'component': 'test'
}, {
'component': 'Chromium',
'link_text': 'somelink'
}, {
'component': 'test'
}]
start, end = impact_task.get_start_and_end_revision('123:456',
'android_job')
self.assertEqual(9, start)
self.assertEqual(90, end)
self.mock.get_start_and_end_revision.assert_has_calls(
[mock.call('123:456'), mock.call('somelink')])
self.mock.get_component_range_list.assert_has_calls(
[mock.call(1, 100, 'android_job')])
