def test_image_save_dir_not_exists(self):
image_file = './tests/fixtures/empty.txt'
save_dir = '/some/save/dir'
try:
CollectImages(image_file=image_file, save_dir=save_dir)
except OSError, e:
self.assertIn("Save Directory not found", e.message)
def test_image_file_not_exists(self):
image_file = '/path/not/found/file.txt'
save_dir = '/some/save/dir'
try:
CollectImages(image_file=image_file, save_dir=save_dir)
except OSError, e:
self.assertEqual(e.message, "File %s Not Found" % image_file)
def test_valid_but_not_found_urls(self):
image_file = './tests/fixtures/valid_not_found.txt'
save_dir = '/tmp/save_dir/'
try:
CollectImages(image_file=image_file,
save_dir=save_dir,
create=True)
except Exception, e:
self.fail('Not Expected Exception thrown: ' + str(e))
def test_one_urls_in_file(self):
image_file = './tests/fixtures/one_url.txt'
save_dir = '/tmp/save_dir/'
try:
CollectImages(image_file=image_file,
save_dir=save_dir,
create=True)
except Exception, e:
self.fail('Not Expected Exception thrown: ' + str(e))
def test_no_urls_in_file(self):
image_file = './tests/fixtures/invalid.txt'
save_dir = '/tmp/save_dir/'
try:
CollectImages(image_file=image_file,
save_dir=save_dir,
create=True)
except Exception, e:
self.assertEqual('No Valid URLs Found', e.message)
def test_image_empty_file(self):
image_file = './tests/fixtures/empty.txt'
save_dir = '/tmp/save_dir/'
try:
CollectImages(image_file=image_file,
save_dir=save_dir,
create=True)
except Exception, e:
self.assertEqual('Images File is Empty', e.message)
def test_image_save_dir_permission_denied(self):
image_file = './tests/fixtures/empty.txt'
save_dir = '/dir'
try:
CollectImages(image_file=image_file,
save_dir=save_dir,
create=True)
except OSError, e:
self.assertEqual(e.errno, errno.EACCES)