def test_len(self):
wants = [sha.id for sha in self.objs if isinstance(sha, Commit)]
finder = MissingObjectFinder(self.store, (), wants)
iterator = GreenThreadsObjectStoreIterator(self.store,
iter(finder.next, None),
finder)
# One commit refers one tree and one blob
self.assertEqual(len(iterator), self.cmt_amount * 3)
haves = wants[0:self.cmt_amount - 1]
finder = MissingObjectFinder(self.store, haves, wants)
iterator = GreenThreadsObjectStoreIterator(self.store,
iter(finder.next, None),
finder)
self.assertEqual(len(iterator), 3)
def test_iter(self):
wants = [sha.id for sha in self.objs if isinstance(sha, Commit)]
finder = MissingObjectFinder(self.store, (), wants)
iterator = GreenThreadsObjectStoreIterator(self.store,
iter(finder.next, None),
finder)
objs = []
for sha, path in iterator:
self.assertIn(sha, self.objs)
objs.append(sha)
self.assertEqual(len(objs), len(self.objs))