# This file is auto-generated by the CDP protocol generator.
# Do not edit this file manually as your changes will be overwritten.
# Generated from Chrome DevTools Protocol specifications.

"""CDP Emulation Domain Event Registration"""

from typing import Callable, Optional

from typing import TYPE_CHECKING

if TYPE_CHECKING:
    from ..registry import EventRegistry
    from .events import VirtualTimeBudgetExpiredEvent


class EmulationRegistration:
    """Event registration interface for Emulation domain."""

    def __init__(self, registry: "EventRegistry"):
        self._registry = registry
        self._domain = "Emulation"

    def virtualTimeBudgetExpired(
        self,
        callback: Callable[["VirtualTimeBudgetExpiredEvent", Optional[str]], None],
    ) -> None:
        """
        Register a callback for virtualTimeBudgetExpired events.

        Notification sent after the virtual time budget for the current VirtualTimePolicy has run out.

        Args:
            callback: Function to call when event occurs.
                     Receives (event_data, session_id) as parameters.
        """
        self._registry.register("Emulation.virtualTimeBudgetExpired", callback)