Class FakeWalSegment

tests/test_wal_transfer.py:16–31

class FakeWalSegment(object):
    def __init__(self, seg_path, explicit=False,
                 upload_explosive=False,
                 mark_done_explosive=False):
        self.explicit = explicit
        self._upload_explosive = upload_explosive
        self._mark_done_explosive = mark_done_explosive

        self._marked = False
        self._uploaded = False

    def mark_done(self):
        if self._mark_done_explosive:
            raise self._mark_done_explosive

        self._marked = True


class FakeWalUploader(object):