def test_no_links_all(self): response = mock.Mock(status_code=404) self.requests.get.return_value = response repo = Repository('http://pypi.python.org') dists = repo.all_distributions('foo') with self.assertRaises(StopIteration): dists.next()
def test_all_distributions(self): package_page = mock.Mock(content=self.pkgsync_links) self.requests.get.return_value = package_page repo = Repository('http://pypi.python.org') dists = list(repo.all_distributions('pkgsync')) self.assertEqual(len(dists), 3) self.assertTrue('0.0.0' in [d.version for d in dists]) self.assertTrue('0.0.1' in [d.version for d in dists])