How to Mock a Dependency for Readiness Probe in Python

import time
import unittest
from unittest.mock import Mock

class Service:
    def __init__(self, dependency):
        self.dependency = dependency

    def is_ready(self):
        try:
            result = self.dependency.check()
            return result == "ready"
        except Exception:
            return False

def main():
    # Real dependency that fails intermittently
    class FailingDependency:
        def check(self):
            time.sleep(0.1)
            raise ConnectionError("Dependency down")

    service = Service(FailingDependency())
    print("Real dependency ready:", service.is_ready())

    # Mocked dependency that always returns "ready"
    mock_dep = Mock()
    mock_dep.check.return_value = "ready"
    service_mocked = Service(mock_dep)
    print("Mocked dependency ready:", service_mocked.is_ready())

    # Mocked dependency that raises an exception
    mock_dep_fail = Mock()
    mock_dep_fail.check.side_effect = TimeoutError("Timeout")
    service_fail = Service(mock_dep_fail)
    print("Failing mock ready:", service_fail.is_ready())

if __name__ == "__main__":
    main()