def test_crop_image(self, mock_image_open):
mock_crop_image = mock.Mock()
mock_base_image = mock.Mock(crop=mock.Mock(return_value=mock_crop_image))
mock_base_image.resize.return_value = mock_base_image
mock_image_open.return_value = mock_base_image
file_path = 'abc.jpeg'
crop_image(file_path, 1, 2, 3, 4)
mock_image_open.assert_called_once_with(file_path)
mock_base_image.crop.assert_called_once_with((1, 2, 4, 6))
mock_base_image.resize.assert_called_once_with(mock_crop_image.size)
mock_base_image.paste.assert_called_once_with(mock_crop_image)
mock_base_image.save.assert_called_once_with(file_path)
def capture_feed_to_file(self, file_path):
"""
Capture document info, include author info and document info
"""
self._fetch_url()
self._wait_feed_load_complete()
document_info_selector = '#plc_main'
info_location = self.find_element(document_info_selector).location
document_handle_selector = '.WB_feed_handle'
handler = self.find_element(document_handle_selector)
handler_location = handler.location
handler_size = handler.size
self.screen_shot(file_path)
utils.crop_image(file_path,
info_location['x'] - 3,
info_location['y'] - 3,
handler_size['width'] + 6,
handler_location['y'] + handler_size['height'] - info_location['y'] + 6)
