Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Adding test validating image download to default location #92

Merged
merged 1 commit into from
May 14, 2018
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Adding test validating image download to default location
  • Loading branch information
Sopher authored and Sopher committed May 4, 2018
commit 65324a6b80f5c2b429313ed486bd75a558472c80
53 changes: 53 additions & 0 deletions tests/test_google_images_download.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,53 @@
from google_images_download import google_images_download
import os, errno
import time


def silent_remove_of_file(file):
try:
os.remove(file)
except OSError as e:
if e.errno != errno.ENOENT:
raise e
return False
return True


def test_download_images_to_default_location():
start_time = time.time()
argumnets = {
"keywords": "Polar bears",
"limit": 5,
"print_urls": False
}
try:
temp = argumnets['output_folder']
except KeyError:
pass
else:
assert False, "This test checks download to default location yet an output folder was provided"

output_folder_path = os.path.join(os.path.realpath('.'), 'downloads', '{}'.format(argumnets['keywords']))
if os.path.exists(output_folder_path):
start_amount_of_files_in_output_folder = len([name for name in os.listdir(output_folder_path) if os.path.isfile(os.path.join(output_folder_path, name)) and os.path.getctime(os.path.join(output_folder_path, name)) < start_time])
else:
start_amount_of_files_in_output_folder = 0

response = google_images_download.googleimagesdownload()
response.download(argumnets)
files_modified_after_test_started = [name for name in os.listdir(output_folder_path) if os.path.isfile(os.path.join(output_folder_path, name)) and os.path.getmtime(os.path.join(output_folder_path, name)) > start_time]
end_amount_of_files_in_output_folder = len(files_modified_after_test_started)
print(f"Files downloaded by test {__name__}:")
for file in files_modified_after_test_started:
print(os.path.join(output_folder_path, file))


# assert end_amount_of_files_in_output_folder - start_amount_of_files_in_output_folder == argumnets['limit']
assert end_amount_of_files_in_output_folder == argumnets['limit']

print(f"Cleaning up all files downloaded by test {__name__}...")
for file in files_modified_after_test_started:
if silent_remove_of_file(os.path.join(output_folder_path, file)):
print(f"Deleted {os.path.join(output_folder_path, file)}")
else:
print(f"Failed to delete {os.path.join(output_folder_path, file)}")