def test_repopulate_queues_subgenres(): fetcher = MockFetcher(fetch_values) driver = Driver(test_url2, fetcher) driver.populate_state() before_subgenres = driver.subgenres driver.repopulate_queues(fetcher.fetch(test_url2)) assert_equal(before_subgenres, driver.subgenres) #move two subgenres into history driver.history = driver.history + driver.subgenres[0:2] after_subgenres = driver.subgenres[2::] #wipe driver.subgenres, so that it will actually refresh driver.subgenres = [] driver.repopulate_queues(fetcher.fetch(test_url2)) assert_equal(after_subgenres, driver.subgenres)
def test_next_url(): fetcher = MockFetcher(fetch_values) driver = Driver(test_url, fetcher, NullOutput()) driver.populate_state() next_url = driver.next_url() assert_equal(next_url, test_url2) urls = driver.process_page(fetcher.fetch(next_url), next_url) next_url = driver.next_url() expected_url = "https://itunes.apple.com/us/genre/podcasts-arts/id1301?mt=2&letter=A&page=2#page" assert_equal(next_url, expected_url)
def test_repopulate_queues_letters(): fetcher = MockFetcher(fetch_values) driver = Driver(test_url2, fetcher) driver.populate_state() before_letters = driver.letters driver.repopulate_queues(fetcher.fetch(test_url2)) assert_equal(before_letters, driver.letters) #Add the current url to history, since it would normally be there driver.history.append(driver.current_letter) #move two letters into history driver.history = driver.history + driver.letters[0:2] after_letters = driver.letters[2::] #wipe driver.letters, so that it will actually refresh driver.letters = [] driver.repopulate_queues(fetcher.fetch(test_url2)) assert_equal(after_letters, driver.letters)
def test_repopulate_queues(): fetcher = MockFetcher(fetch_values) driver = Driver(test_url2, fetcher) driver.populate_state() #Shouldn't change anything before_pages = driver.pages before_letters = driver.letters before_subgenres = driver.subgenres driver.repopulate_queues(fetcher.fetch(test_url2)) assert_equal(before_pages, driver.pages) assert_equal(before_letters, driver.letters) assert_equal(before_subgenres, driver.subgenres)