def test_records_change_to_ready_for_processing(
test_perf_alert,
create_record,
record_from_mature_report,
report_maintainer_mock,
backfill_tool_mock,
secretary,
sheriff_settings,
notify_client_mock,
):
# create new report with records - the report will not be mature
create_record(test_perf_alert)
preliminary_records = BackfillRecord.objects.filter(
status=BackfillRecord.PRELIMINARY)
ready_records = BackfillRecord.objects.filter(
status=BackfillRecord.READY_FOR_PROCESSING)
frozen_reports = BackfillReport.objects.filter(frozen=True)
assert preliminary_records.count() == 2
assert ready_records.count() == 0
assert frozen_reports.count() == 0
sheriff_bot = PerfSheriffBot(
report_maintainer_mock,
backfill_tool_mock,
secretary,
notify_client_mock,
)
sheriff_bot.sheriff(since=EPOCH,
frameworks=['raptor', 'talos'],
repositories=['autoland'])
assert preliminary_records.count() == 1
assert ready_records.count() == 1
assert frozen_reports.count() == 1
def test_no_email_is_sent_if_context_is_too_corrupt_to_be_actionable(
self,
report_maintainer_mock,
backfill_tool_mock,
secretary,
record_ready_for_processing,
sheriff_settings,
notify_client_mock,
broken_context_str,
# Note: parametrizes the test
):
record_ready_for_processing.context = broken_context_str
record_ready_for_processing.save()
sheriff_bot = PerfSheriffBot(
report_maintainer_mock,
backfill_tool_mock,
secretary,
notify_client_mock,
)
sheriff_bot.sheriff(since=EPOCH,
frameworks=['raptor', 'talos'],
repositories=['autoland'])
assert notify_client_mock.email.call_count == 0
def test_backfilling_gracefully_handles_invalid_json_contexts_without_blowing_up(
report_maintainer_mock,
backfill_tool_mock,
secretary,
record_ready_for_processing,
sheriff_settings,
notify_client_mock,
broken_context_str, # Note: parametrizes the test
):
record_ready_for_processing.context = broken_context_str
record_ready_for_processing.save()
sheriff_bot = PerfSheriffBot(report_maintainer_mock, backfill_tool_mock,
secretary, notify_client_mock)
try:
sheriff_bot.sheriff(since=EPOCH,
frameworks=['raptor', 'talos'],
repositories=['autoland'])
except (JSONDecodeError, KeyError, Job.DoesNotExist, Push.DoesNotExist):
pytest.fail()
record_ready_for_processing.refresh_from_db()
assert record_ready_for_processing.status == BackfillRecord.FAILED
assert not has_changed(sheriff_settings)
def test_records_remain_unchanged_if_no_backfills_left(
report_maintainer_mock,
backfill_tool_mock,
secretary,
notify_client_mock,
record_ready_for_processing,
empty_sheriff_settings,
):
sheriff_bot = PerfSheriffBot(report_maintainer_mock, backfill_tool_mock,
secretary, notify_client_mock)
sheriff_bot._backfill()
assert not has_changed(record_ready_for_processing)
def test_records_and_db_limits_remain_unchanged_if_no_records_suitable_for_backfill(
report_maintainer_mock,
backfill_tool_mock,
secretary,
notify_client_mock,
sheriff_settings,
record_unsuited_for_backfill,
):
sheriff_bot = PerfSheriffBot(report_maintainer_mock, backfill_tool_mock,
secretary, notify_client_mock)
sheriff_bot._backfill()
assert not has_changed(record_unsuited_for_backfill)
assert not has_changed(sheriff_settings)
def test_assert_can_run_throws_exception_when_runtime_exceeded(
report_maintainer_mock,
backfill_tool_mock,
secretary,
record_ready_for_processing,
sheriff_settings,
notify_client_mock,
):
no_time_left = timedelta(seconds=0)
sheriff_bot = PerfSheriffBot(report_maintainer_mock, backfill_tool_mock,
secretary, notify_client_mock, no_time_left)
with pytest.raises(MaxRuntimeExceeded):
sheriff_bot.assert_can_run()
def perf_sheriff_bot_factory(days_to_lookup: timedelta) -> PerfSheriffBot:
report_maintainer = __report_maintainer_factory(days_to_lookup)
backfill_tool = __backfill_tool_factory()
secretary_tool = SecretaryTool()
notify_client = notify_client_factory()
return PerfSheriffBot(report_maintainer, backfill_tool, secretary_tool, notify_client)
def test_assert_can_run_doesnt_throw_exception_when_enough_time_left(
report_maintainer_mock,
backfill_tool_mock,
secretary,
notify_client_mock,
record_ready_for_processing,
sheriff_settings,
):
enough_time_left = timedelta(minutes=10)
sheriff_bot = PerfSheriffBot(report_maintainer_mock, backfill_tool_mock,
secretary, notify_client_mock,
enough_time_left)
try:
sheriff_bot.assert_can_run()
except MaxRuntimeExceeded:
pytest.fail()
def test_db_limits_update_if_backfills_left(
report_maintainer_mock,
backfill_tool_mock,
secretary,
record_ready_for_processing,
sheriff_settings,
notify_client_mock,
):
initial_backfills = secretary.backfills_left(on_platform='linux')
sheriff_bot = PerfSheriffBot(report_maintainer_mock, backfill_tool_mock,
secretary, notify_client_mock)
sheriff_bot.sheriff(since=EPOCH,
frameworks=['raptor', 'talos'],
repositories=['autoland'])
record_ready_for_processing.refresh_from_db()
assert record_ready_for_processing.status == BackfillRecord.BACKFILLED
assert (initial_backfills -
5) == secretary.backfills_left(on_platform='linux')
def test_records_and_db_limits_remain_unchanged_if_runtime_exceeded(
report_maintainer_mock,
backfill_tool_mock,
secretary,
record_ready_for_processing,
sheriff_settings,
notify_client_mock,
):
no_time_left = timedelta(seconds=0)
sheriff_bot = PerfSheriffBot(report_maintainer_mock, backfill_tool_mock,
secretary, notify_client_mock, no_time_left)
try:
sheriff_bot.sheriff(since=EPOCH,
frameworks=['raptor', 'talos'],
repositories=['autoland'])
except MaxRuntimeExceeded:
pass
assert not has_changed(record_ready_for_processing)
assert not has_changed(sheriff_settings)
def test_no_email_is_sent_if_runtime_exceeded(
self,
report_maintainer_mock,
backfill_tool_mock,
secretary,
record_ready_for_processing,
sheriff_settings,
notify_client_mock,
):
no_time_left = timedelta(seconds=0)
sheriff_bot = PerfSheriffBot(report_maintainer_mock,
backfill_tool_mock, secretary,
notify_client_mock, no_time_left)
try:
sheriff_bot.sheriff(since=EPOCH,
frameworks=['raptor', 'talos'],
repositories=['autoland'])
except MaxRuntimeExceeded:
pass
assert notify_client_mock.email.call_count == 0
def test_email_is_sent_after_successful_backfills(
self,
report_maintainer_mock,
backfill_tool_mock,
secretary,
record_ready_for_processing,
sheriff_settings,
notify_client_mock,
):
sheriff_bot = PerfSheriffBot(
report_maintainer_mock,
backfill_tool_mock,
secretary,
notify_client_mock,
email_writer=self.email_writer_mock(),
)
sheriff_bot.sheriff(since=EPOCH,
frameworks=['raptor', 'talos'],
repositories=['autoland'])
record_ready_for_processing.refresh_from_db()
assert notify_client_mock.email.call_count == 1