def test_get_index_pages_raw(self): self.mox.StubOutWithMock(tal, 'enumerate_index_page_locs') self.mox.StubOutWithMock(requests, 'get') tal.enumerate_index_page_locs(2013, 2015).AndReturn( ['/2013', '/2014', '/2015'] ) requests.get('/2013', headers=common.DEFAULT_HEADERS).AndReturn( FakeResponse(200, 't1') ) requests.get('/2014', headers=common.DEFAULT_HEADERS).AndReturn( FakeResponse(200, 't2') ) requests.get('/2015', headers=common.DEFAULT_HEADERS).AndReturn( FakeResponse(404, 't3') ) self.mox.ReplayAll() results = tal.get_index_pages_raw(2013, 2015) self.assertEqual(len(results), 2) self.assertEqual(results[0], 't1') self.assertEqual(results[1], 't2')
def test_enumerate_index_page_locs_to_today(self): pages = tal.enumerate_index_page_locs(2013, 2015) self.assertEqual(pages[0], tal.INDEX_PAGE_TEMPLATE % 2013)
def test_enumerate_index_page_locs(self): pages = tal.enumerate_index_page_locs(2013, 2015) self.assertEqual(len(pages), 3) self.assertEqual(pages[0], tal.INDEX_PAGE_TEMPLATE % 2013) self.assertEqual(pages[1], tal.INDEX_PAGE_TEMPLATE % 2014) self.assertEqual(pages[2], tal.INDEX_PAGE_TEMPLATE % 2015)