class RepositoryIteratorTests(TestCase):
def setUp(self):
self.date_revs = {
datetime(2011, 1, 1): 0,
datetime(2011, 2, 2): 1
}
self.di = self.date_revs.keys()
self.di.sort()
self.repo = MockDateRepository(self.date_revs)
self.ri = RepositoryIterator(self.repo, date_iterator=self.di)
def test_repository_iterator_switches_to_first_revision(self):
date = self.ri.next()
eq_(self.repo.latest_rev_requested, 0)
eq_(datetime(2011, 1, 1), date)
def test_repository_iterator_switches_to_second_revision(self):
self.ri.next()
date = self.ri.next()
eq_(self.repo.latest_rev_requested, 1)
eq_(datetime(2011, 2, 2), date)
@raises(StopIteration)
def test_repository_iterator_throws_stop_iteration(self):
self.ri.next()
self.ri.next()
self.ri.next()
def setUp(self):
self.date_revs = {
datetime(2011, 1, 1): 0,
datetime(2011, 2, 2): 1
}
self.di = self.date_revs.keys()
self.di.sort()
self.repo = MockDateRepository(self.date_revs)
self.ri = RepositoryIterator(self.repo, date_iterator=self.di)
