Skip to content

Commit

Permalink
Adding test validating image download to default location (#92)
Browse files Browse the repository at this point in the history
  • Loading branch information
amitaisopher authored and hardikvasa committed May 14, 2018
1 parent 68f8bf7 commit 0f1b04c
Showing 1 changed file with 53 additions and 0 deletions.
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)}")

0 comments on commit 0f1b04c

Please sign in to comment.