mirror of
https://github.com/vale981/ray
synced 2025-03-06 10:31:39 -05:00
42 lines
1.1 KiB
Python
42 lines
1.1 KiB
Python
|
from ray_release.config import Test
|
||
|
|
from ray_release.exception import ReleaseTestConfigError, ResultsAlert
|
||
|
|
from ray_release.logger import logger
|
||
|
|
from ray_release.result import Result
|
||
|
|
|
||
|
|
from ray_release.alerts import (
|
||
|
|
default,
|
||
|
|
long_running_tests,
|
||
|
|
rllib_tests,
|
||
|
|
tune_tests,
|
||
|
|
xgboost_tests,
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
result_to_handle_map = {
|
||
|
|
"default": default.handle_result,
|
||
|
|
"long_running_tests": long_running_tests.handle_result,
|
||
|
|
"rllib_tests": rllib_tests.handle_result,
|
||
|
|
"tune_tests": tune_tests.handle_result,
|
||
|
|
"xgboost_tests": xgboost_tests.handle_result,
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
def handle_result(test: Test, result: Result):
|
||
|
|
alert_suite = test.get("alert", "default")
|
||
|
|
|
||
|
|
logger.info(
|
||
|
|
f"Checking results for test {test['name']} using alerting suite "
|
||
|
|
f"{alert_suite}"
|
||
|
|
)
|
||
|
|
|
||
|
|
if alert_suite not in result_to_handle_map:
|
||
|
|
raise ReleaseTestConfigError(f"Alert suite {alert_suite} not found.")
|
||
|
|
|
||
|
|
handler = result_to_handle_map[alert_suite]
|
||
|
|
error = handler(test, result)
|
||
|
|
|
||
|
|
if error:
|
||
|
|
raise ResultsAlert(error)
|
||
|
|
|
||
|
|
logger.info("No alerts have been raised - test passed successfully!")
|