Class MessageFactory

tests/tests/test_timeline.py:17–30

A factory for DAP messages that are not bound to a message channel.

class MessageFactory(object):
    """A factory for DAP messages that are not bound to a message channel."""

    def __init__(self):
        self._seq_iter = itertools.count(1)
        self.messages = collections.OrderedDict()
        self.event = functools.partial(self._make, messaging.Event)
        self.request = functools.partial(self._make, messaging.Request)
        self.response = functools.partial(self._make, messaging.Response)

    def _make(self, message_type, *args, **kwargs):
        message = message_type(None, next(self._seq_iter), *args, **kwargs)
        self.messages[message.seq] = message
        return message


@pytest.fixture