from hawkeye_monitor.checks.base import ElasticsearchCheck
from hawkeye_monitor.data import CheckResult


class CheckIsAlive(ElasticsearchCheck):
    def validate(self, response) -> CheckResult:
        """
        Validates Results

        params:
            - response(dict) response object from the ES

        returns:
            - CheckResult

        """
        check_name = self.__class__.__name__
        check_passed = False
        message = None

        try:
            assert response["hits"]["hits"] >= 0
            check_passed = True
            message = "Elasticsearch IsAliveCheck Passed!"

        except AssertionError:
            check_passed = False
            message = "Elasticsearch IsAliveCheck Failed :("

        return CheckResult(
            **{
                "check_name": check_name,
                "check_passed": check_passed,
                "message": message,
            }
        )