from health_check.backends import BaseHealthCheckBackend
from health_check.exceptions import (
    ServiceUnavailable,
    ServiceReturnedUnexpectedResult,
)

from driver.connector import Connector


class DriverHealthCheck(BaseHealthCheckBackend):
    critical_service = False

    def check_status(self):
        try:
            Connector().passthrough(b"<s>")
        except ConnectionRefusedError as e:
            self.add_error(ServiceUnavailable("IOError"), e)
        except Exception as e:
            self.add_error(ServiceReturnedUnexpectedResult("IOError"), e)

    def identifier(self):
        return "DriverDaemon"