class TestFetchResizeSave(unittest.TestCase):
def setUp(self):
self._api_key = '__api_key__'
self._api_cx = '__api_cx__'
self._frs = FetchResizeSave(self._api_key, self._api_cx)
self._base_dir = os.path.join(
os.path.dirname(os.path.dirname(os.path.abspath(__file__))),
'tests')
self._file_paths = [
os.path.join(self._base_dir, '1.png'),
os.path.join(self._base_dir, '2.png'),
]
def tearDown(self):
self._frs = None
for path in self._file_paths:
try:
os.remove(path)
except FileNotFoundError:
pass
def test_init(self):
self.assertEqual(self._frs._search_resut, [])
def test_search_url(self):
self._frs.search({})
for i, item in enumerate(self._frs.results()):
self.assertEqual(item.url, items['items'][i]['link'])
def test_search_path(self):
self._frs.search({}, path_to_dir=self._base_dir, width=100, height=100)
for i, item in enumerate(self._frs.results()):
self.assertEqual(item.path, self._file_paths[i])
class TestFetchResizeSave(unittest.TestCase):
def setUp(self):
self._api_key = '__api_key__'
self._api_cx = '__api_cx__'
self._frs = FetchResizeSave(self._api_key, self._api_cx)
self._base_dir = os.path.join(
os.path.dirname(os.path.dirname(os.path.abspath(__file__))),
'tests')
self._file_paths = [
os.path.join(self._base_dir, '1.png'),
os.path.join(self._base_dir, '2.png'),
]
def tearDown(self):
self._frs = None
for path in self._file_paths:
try:
os.remove(path)
except OSError:
pass
def test_init(self):
self.assertTrue(
isinstance(self._frs._google_custom_search, GoogleCustomSearch))
self.assertEqual(self._frs._search_result, [])
self.assertEqual(self._frs._progress, False)
frs = FetchResizeSave(self._api_key,
self._api_cx,
progressbar_fn=lambda x, y: None)
self.assertEqual(frs._chunk_sizes, {})
self.assertEqual(frs._terminal_lines, {})
self.assertEqual(frs._download_progress, {})
self.assertNotEqual(frs._report_progress, None)
self.assertEqual(frs._width, None)
self.assertEqual(frs._height, None)
self.assertEqual(frs._path_to_dir, False)
self.assertEqual(frs._search_params, None)
self.assertEqual(frs._cache_discovery, True)
self.assertEqual(frs._page, 1)
self.assertEqual(frs._number_of_images, None)
def test_search_url(self):
self._frs.search({'num': 2})
for i, item in enumerate(self._frs.results()):
self.assertEqual(item.url, items['items'][i]['link'])
def test_search_path(self):
self._frs.search({}, path_to_dir=self._base_dir, width=100, height=100)
#for i, item in enumerate(self._frs.results()):
# self.assertEqual(item.path, self._file_paths[i])
def test_progressbar(self):
progress_data = []
def pbar(url, progress):
progress_data.append((url, progress))
frs = FetchResizeSave(self._api_key, self._api_cx, progressbar_fn=pbar)
frs.search({'num': 2}, path_to_dir=self._base_dir)
#self.assertEqual(
# progress_data,
# list(zip([items['items'][0]['link']] * 100, list(range(1, 101)))) +
# list(zip([items['items'][1]['link']] * 100, list(range(1, 101))))
#)
def test_bytes_io(self):
my_bytes_io = BytesIO()
self._frs.search({'num': 2})
for image in self._frs.results():
my_bytes_io.seek(0)
raw_image_data = image.get_raw_data()
image.copy_to(my_bytes_io, raw_image_data)
image.copy_to(my_bytes_io)
my_bytes_io.seek(0)
def test_paging(self):
self._frs.search({'num': 2})
self.assertEqual(self._frs._search_params, {'num': 2, 'start': 1})
self.assertEqual(self._frs._page, 1)
self.assertEqual(self._frs._number_of_images, 2)
self._frs.next_page()
self.assertEqual(self._frs._search_params, {'num': 2, 'start': 3})
self.assertEqual(self._frs._page, 2)
self.assertEqual(self._frs._number_of_images, 2)
