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))