from pyromancer.objects import Connection


class MockConnection(Connection):

    def __init__(self, *args, **kwargs):
        self.outbox = []

    def write(self, data):
        self.outbox.append(data)

Add mock Connection object for testing purposes.
Add mock Connection object for testing purposes.
