def test_write_index_file(self):
"""
http://stackoverflow.com/questions/977491/comparing-2-txt-files-using-difflib-in-python
http://doughellmann.com/2007/10/pymotw-difflib.html
"""
img_urls = logpuzzle.read_urls('animal_code.google.com')
dest_dir = './puzzle_images'
logpuzzle.download_images(img_urls, dest_dir)
logpuzzle.write_index_file(dest_dir)
index_file = open('./index.html', 'r')
result_lines = index_file.readlines()
index_file.close()
index_expected_file = open('./index_expected.html', 'r')
expected_result_lines = index_expected_file.readlines()
index_expected_file.close()
# put expected_result_lines first, to make diff +- clearer
diff_unified = difflib.unified_diff(expected_result_lines, result_lines, lineterm='')
result = ('\n'.join(list(diff_unified)))
expected_result = ''
self.assertEqual(expected_result, result,
'write_index_file() expected {} but got {}'.format(expected_result, result))
self.clean_up_image_dir()
def test_z_download_images(self):
""" Choose test name alphabetically after test_write_index_file to run after it.
"""
#img_urls = logpuzzle.read_urls('place_code.google.com')
img_urls = logpuzzle.read_urls('animal_code.google.com')
dest_dir = './puzzle_images'
logpuzzle.download_images(img_urls, dest_dir)
result = os.listdir(dest_dir)
expected_result = ['img0.jpg', 'img1.jpg', 'img10.jpg', 'img11.jpg', 'img12.jpg', 'img13.jpg', 'img14.jpg', 'img15.jpg', 'img16.jpg', 'img17.jpg', 'img18.jpg', 'img19.jpg', 'img2.jpg', 'img3.jpg', 'img4.jpg', 'img5.jpg', 'img6.jpg', 'img7.jpg', 'img8.jpg', 'img9.jpg']
self.assertEqual(expected_result, result,
'write_index_file() expected {} but got {}'.format(expected_result, result))
