def test_fetch(self): target_folder = '/tmp/variety/ImageFetcher' shutil.rmtree(target_folder, ignore_errors=True) os.makedirs(target_folder) for url in ["http://wallpapers.wallhaven.cc/wallpapers/full/wallhaven-207261.jpg", "http://unsplash.com/photos/7EqQ1s3wIAI/download", "http://az608707.vo.msecnd.net/files/GreaterFlamingos_EN-US13682107304_1366x768.jpg", "http://a.desktopprassets.com/wallpapers/07865fb0cb575e82fe43d3e1b634f6e2309e2114/foto_alese45.jpg", ]: f = ImageFetcher.fetch(url, target_folder, verbose=False) self.assertIsNotNone(f) self.assertTrue(os.path.isfile(f)) self.assertTrue(Util.is_image(f, check_contents=False)) self.assertTrue(Util.is_image(f, check_contents=True)) self.assertNotEqual('download', f)
def test_fetch(self): target_folder = '/tmp/variety/ImageFetcher' shutil.rmtree(target_folder, ignore_errors=True) os.makedirs(target_folder) for url in [ "http://wallpapers.wallhaven.cc/wallpapers/full/wallhaven-207261.jpg", "http://unsplash.com/photos/7EqQ1s3wIAI/download", "http://az608707.vo.msecnd.net/files/GreaterFlamingos_EN-US13682107304_1366x768.jpg", "http://a.desktopprassets.com/wallpapers/07865fb0cb575e82fe43d3e1b634f6e2309e2114/foto_alese45.jpg", ]: f = ImageFetcher.fetch(url, target_folder, verbose=False) self.assertIsNotNone(f) self.assertTrue(os.path.isfile(f)) self.assertTrue(Util.is_image(f, check_contents=False)) self.assertTrue(Util.is_image(f, check_contents=True)) self.assertNotEqual('download', f)
def test_download_one_for(test_case, dl): dl.target_folder = '/tmp/variety/%s' % dl.__class__.__name__ shutil.rmtree(dl.target_folder, ignore_errors=True) for _ in xrange(5): f = dl.download_one() if f and os.path.isfile(f) and Util.is_image(f, check_contents=True): return test_case.fail("Tried download_one 5 times, all failed")