def write_to_big_query(testcase, regression_range_start, regression_range_end):
"""Write the regression range to BigQuery."""
big_query.write_range(table_id='regressions',
testcase=testcase,
range_name='regression',
start=regression_range_start,
end=regression_range_end)
def test_error(self):
"""Tests error."""
self.client.insert.return_value = {'insertErrors': ['exception']}
big_query.write_range('regressions', self.testcase, 'regression', 456,
789)
self.mock.log_error.assert_called_once_with(
("Ignoring error writing the testcase's regression range (%s) to "
'BigQuery.' % self.testcase.key.id()),
exception=mock.ANY)
self.client.insert.assert_called_once_with([
big_query.Insert(row={
'testcase_id': str(self.testcase.key.id()),
'crash_type': 'type',
'crash_state': 'state',
'security_flag': True,
'parent_fuzzer_name': 'libfuzzer',
'fuzzer_name': 'libfuzzer_pdf',
'job_type': 'some_job',
'created_at': 99,
'regression_range_start': 456,
'regression_range_end': 789,
},
insert_id='%s:456:789' % self.testcase.key.id())
])
def _write_to_bigquery(testcase, progression_range_start,
progression_range_end):
"""Write the fixed range to BigQuery."""
big_query.write_range(table_id='fixeds',
testcase=testcase,
range_name='fixed',
start=progression_range_start,
end=progression_range_end)
def test_write(self):
"""Tests write."""
self.client.insert.return_value = {}
big_query.write_range('regressions', self.testcase, 'regression', 456,
789)
self.assertEqual(0, self.mock.log_error.call_count)
self.client.insert.assert_called_once_with([
big_query.Insert(row={
'testcase_id': str(self.testcase.key.id()),
'crash_type': 'type',
'crash_state': 'state',
'security_flag': True,
'parent_fuzzer_name': 'libfuzzer',
'fuzzer_name': 'libfuzzer_pdf',
'job_type': 'some_job',
'created_at': 99,
'regression_range_start': 456,
'regression_range_end': 789,
},
insert_id='%s:456:789' % self.testcase.key.id())
])
