def test_results(self): feed = RandomChangeFeed(5) changes = feed.iter_changes(0) ids = set() for seq, change in enumerate(changes): self.assertTrue(isinstance(change, Change)) self.assertEqual(seq, change.sequence_id) self.assertEqual(None, change.document) self.assertEqual(False, change.deleted) self.assertTrue(change.id not in ids) ids.add(change.id)
def GetDocPillow(): return FakeConstructedPillow( name='GetDocPillow', checkpoint=PillowCheckpoint('get_doc_processor', 'text'), change_feed=RandomChangeFeed(10), processor=GetDocProcessor(), )
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 test_override_fn(self): val = object() feed = RandomChangeFeed(5, change_generator=lambda seq: val) changes = feed.iter_changes(0) for change in changes: self.assertEqual(val, change)
def test_len(self): feed = RandomChangeFeed(5) self.assertEqual(5, len(list(feed.iter_changes(0)))) self.assertEqual(3, len(list(feed.iter_changes(2)))) self.assertEqual(0, len(list(feed.iter_changes(5)))) self.assertEqual(0, len(list(feed.iter_changes(10))))
def __init__(self): super(FakePillow, self).__init__('fake pillow', PillowCheckpoint('test_pillow_import', 'text'), RandomChangeFeed(10), LoggingProcessor())