def make_fake_constructed_pillow(pillow_id, checkpoint_id): from pillowtop.feed.mock import RandomChangeFeed from pillowtop.processors import LoggingProcessor pillow = FakeConstructedPillow( name=pillow_id, checkpoint=PillowCheckpoint(checkpoint_id, 'text'), change_feed=RandomChangeFeed(10), processor=LoggingProcessor(), ) return pillow
def _make_couch_pillow(couch_db): from pillowtop.feed.couch import CouchChangeFeed from pillowtop.processors import LoggingProcessor from pillowtop.checkpoints.manager import PillowCheckpoint pillow = FakeConstructedPillow( name='fake-couch-pillow', checkpoint=PillowCheckpoint('fake-feed-test-checkpoint', 'text'), change_feed=CouchChangeFeed(couch_db=couch_db), processor=LoggingProcessor(), ) pillow.process_change = MagicMock(return_value=True) return pillow
def __init__(self): super(FakePillow, self).__init__('fake pillow', PillowCheckpoint('test_pillow_import', 'text'), RandomChangeFeed(10), LoggingProcessor())