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')])