def test_download_by_text_file_with_broken_image_urls(self): data = self.URL_BROKEN_LINK + '\n' + self.URL_VALID_IMAGE_2 fake_file = io.StringIO(data) with patch('builtins.open', return_value=fake_file, create=True): url_list = read_in_file("foo.bar") self.assertEqual(download_by_url_list(url_list), 1)
def test_download_by_url_list_with_invalid_image_format(self): url_list = [self.URL_INVALID_IMAGE, self.URL_VALID_IMAGE_2] self.assertEqual(download_by_url_list(url_list), 1)
def test_download_by_url_list_with_valid_image_urls(self): url_list = [self.URL_VALID_IMAGE_1, self.URL_VALID_IMAGE_2] self.assertEqual(download_by_url_list(url_list), 2)
def test_download_by_url_list_with_broken_image_urls(self): url_list = [self.URL_BROKEN_LINK, self.URL_VALID_IMAGE_2] self.assertEqual(download_by_url_list(url_list), 1)
def test_download_by_url_list_with_empty_list(self): url_list = [] self.assertEqual(download_by_url_list(url_list), 0)