def test_Gallery_property_comments_enabled():
assert isinstance(_upload.Gallery().comments_enabled, bool)
g = _upload.Gallery(comments_enabled=True)
assert g.comments_enabled is True
g.comments_enabled = False
assert g.comments_enabled is False
g.comments_enabled = 1
assert g.comments_enabled is True
def test_Gallery_property_title():
assert _upload.Gallery().title is None
g = _upload.Gallery(title='Foo, Bar, Baz')
assert g.title == 'Foo, Bar, Baz'
g.title = (1, 2, 3)
assert g.title == '(1, 2, 3)'
g.title = None
assert g.title is None
def test_Gallery_property_adult():
assert isinstance(_upload.Gallery().adult, bool)
g = _upload.Gallery(adult=True)
assert g.adult is True
g.adult = False
assert g.adult is False
g.adult = 1
assert g.adult is True
def test_Gallery_add_raises_RuntimeError_if_create_was_not_called_first():
g = _upload.Gallery()
mock_submit_file = Mock()
with patch.multiple(g, _submit_file=mock_submit_file):
with pytest.raises(RuntimeError, match=r'^create\(\) must be called first$'):
tuple(g.add('foo.jpg'))
assert mock_submit_file.call_args_list == []
def test_Gallery_create_gets_invalid_html_from_landing_page(mock_get, mock_post_json):
mock_get.return_value = '~~> This is not html. <-'
g = _upload.Gallery()
with pytest.raises(RuntimeError, match=r"Couldn't find CSRF token in HTML head"):
g.create()
assert _const.CSRF_TOKEN_HEADER not in g._session.headers
assert g._token == {}
assert g.created is False
def test_Gallery_properties_thumb_width_and_square_thumbs():
assert isinstance(_upload.Gallery().thumb_width, int)
assert isinstance(_upload.Gallery().square_thumbs, bool)
g = _upload.Gallery(thumb_width=200, square_thumbs=False)
assert g.thumb_width == 200
assert g.square_thumbs is False
assert g._thumbnail_width == _const.THUMBNAIL_WIDTHS_KEEP_ASPECT[200]
g.square_thumbs = True
assert g.thumb_width == 200
assert g.square_thumbs is True
assert g._thumbnail_width == _const.THUMBNAIL_WIDTHS_SQUARE[200]
g.thumb_width = 321
assert g.thumb_width == 300
assert g.square_thumbs is True
assert g._thumbnail_width == _const.THUMBNAIL_WIDTHS_SQUARE[300]
g.square_thumbs = 0
assert g.thumb_width == 300
assert g.square_thumbs is False
assert g._thumbnail_width == _const.THUMBNAIL_WIDTHS_KEEP_ASPECT[300]
def test_Gallery_submit_file_gets_json_response_with_files_field(mock_post_json):
gallery = _upload.Gallery()
mock_token = {'token_id': '123', 'token_secret': '456',
'gallery_id': 'abc', 'gallery_secret': 'def'}
gallery._token = mock_token
mock_post_json.return_value = {'files': []}
with pytest.raises(RuntimeError) as cm:
gallery._submit_file('path/to/file.jpg',
content_type=_const.CONTENT_TYPES['adult'],
thumbnail_size=_const.THUMBNAIL_WIDTHS_SQUARE[300],
timeout=12)
assert str(cm.value) == "Unexpected response: 'files' is empty: {'files': []}"
def test_Gallery_create_ensures_token_is_dictionary(mock_get, mock_post_json):
mock_get.return_value = ('<html><head>'
'<meta foo="bar" name="something" />'
'<meta content="THE-CSRF-TOKEN" name="csrf-token" />'
'<meta name="yo" />'
'</head></html>')
mock_post_json.return_value = (1, 2, 3)
g = _upload.Gallery()
with pytest.raises(RuntimeError, match=r'^Not a dict: \(1, 2, 3\)$'):
g.create()
assert _const.CSRF_TOKEN_HEADER in g._session.headers
assert g._token == {}
assert g.created is False
def test_Gallery_submit_file_cannot_open_file(mock_post_json):
gallery = _upload.Gallery()
sub = gallery._submit_file('this/file/does/not/exist.jpg',
content_type=_const.CONTENT_TYPES['family'],
thumbnail_size=_const.THUMBNAIL_WIDTHS_SQUARE[200],
timeout=123)
assert sub == {'success': False,
'error': 'No such file or directory',
'filename': 'exist.jpg',
'filepath': 'this/file/does/not/exist.jpg',
'image_url': None,
'thumbnail_url': None,
'web_url': None,
'gallery_url': None,
'edit_url': None}
def test_Gallery_submit_file_gets_file_with_unsupported_mimetype(mock_post_json, mock_guess_type):
mock_guess_type.return_value = ('text/plain', None)
gallery = _upload.Gallery()
sub = gallery._submit_file('path/to/file.txt',
content_type=_const.CONTENT_TYPES['family'],
thumbnail_size=_const.THUMBNAIL_WIDTHS_SQUARE[200],
timeout=123)
assert sub == {'success': False,
'error': 'Unsupported file type: text/plain',
'filename': 'file.txt',
'filepath': 'path/to/file.txt',
'image_url': None,
'thumbnail_url': None,
'web_url': None,
'gallery_url': None,
'edit_url': None}
def test_Gallery_submit_yields_return_values_from_submit_file():
filepaths = ('path/to/foo.jpg', '/other/path/to/bar.jpg')
g = _upload.Gallery()
mock_create = Mock()
mock_submit_file = Mock()
# Mock g.created to be True
g._token = {'token_id': '123', 'token_secret': '456',
'gallery_id': 'abc', 'gallery_secret': 'def'}
g._session.headers[_const.CSRF_TOKEN_HEADER] = 'something'
with patch.multiple(g, create=mock_create, _submit_file=mock_submit_file):
for sub in g.add(*filepaths, timeout=123):
pass
assert mock_submit_file.call_args_list == [call('path/to/foo.jpg',
_const.CONTENT_TYPES['family'],
_const.THUMBNAIL_WIDTHS_KEEP_ASPECT[100],
timeout=123),
call('/other/path/to/bar.jpg',
_const.CONTENT_TYPES['family'],
_const.THUMBNAIL_WIDTHS_KEEP_ASPECT[100],
timeout=123)]
def test_Gallery_submit_file_succeeds(mock_post_json):
gallery = _upload.Gallery()
mock_token = {'token_id': '123', 'token_secret': '456',
'gallery_id': 'abc', 'gallery_secret': 'def'}
gallery._token = mock_token
mock_post_json.return_value = {'files': [{'original_url': 'https://foo.example.org/asdf.jpg',
'thumbnail_url': 'https://foo.example.org/asdf_t.jpg',
'url': 'https://foo.example.org/asdf'}]}
sub = gallery._submit_file('path/to/file.jpg',
content_type=_const.CONTENT_TYPES['adult'],
thumbnail_size=_const.THUMBNAIL_WIDTHS_SQUARE[300],
timeout=12)
assert sub == _upload.Submission(
success=True,
filename='file.jpg',
filepath='path/to/file.jpg',
image_url='https://foo.example.org/asdf.jpg',
thumbnail_url='https://foo.example.org/asdf_t.jpg',
web_url='https://foo.example.org/asdf',
gallery_url=_const.GALLERY_URL_FORMAT.format(**mock_token),
edit_url=_const.EDIT_URL_FORMAT.format(**mock_token),
)
def test_Gallery_submit_file_raises_ValueError(mock_post_json):
gallery = _upload.Gallery()
mock_token = {'token_id': '123', 'token_secret': '456',
'gallery_id': 'abc', 'gallery_secret': 'def'}
gallery._token = mock_token
mock_post_json.side_effect = ValueError('Error while parsing JSON')
sub = gallery._submit_file('path/to/file.jpg',
content_type=_const.CONTENT_TYPES['adult'],
thumbnail_size=_const.THUMBNAIL_WIDTHS_SQUARE[300],
timeout=12)
assert sub == {'success': False,
'error': 'Error while parsing JSON',
'filename': 'file.jpg',
'filepath': 'path/to/file.jpg',
'image_url': None,
'thumbnail_url': None,
'web_url': None,
'gallery_url': None,
'edit_url': None}
assert mock_post_json.call_args_list == [
call(
gallery._session, _const.PROCESS_URL,
data=[
('token_id', mock_token['token_id']),
('token_secret', mock_token['token_secret']),
('content_type', _const.CONTENT_TYPES['adult']),
('thumbnail_size', _const.THUMBNAIL_WIDTHS_SQUARE[300]),
('gallery_id', mock_token['gallery_id']),
('gallery_secret', mock_token['gallery_secret']),
('comments_enabled', '0'),
],
files=[('files[]', ('file.jpg',
open('path/to/file.jpg'),
'image/jpeg'))],
timeout=12,
)
]
def test_Gallery_create_sets_session_header_and_token(mock_get, mock_post_json):
mock_get.return_value = ('<html><head>'
'<meta foo="bar" name="something" />'
'<meta content="THE-CSRF-TOKEN" name="csrf-token" />'
'<meta name="yo" />'
'</head></html>')
mock_post_json.return_value = {'something': 'asdf', 'foo': 'bar'}
g = _upload.Gallery(title='The Title', comments_enabled=True)
assert g.created is False
assert g.create() is None
assert mock_get.call_args_list == [
call(g._session, f'https://{_const.SERVICE_DOMAIN}/', timeout=30)]
assert mock_post_json.call_args_list == [
call(g._session, _const.TOKEN_URL, timeout=30,
data=[('gallery', 'true'),
('gallery_title', 'The Title'),
('comments_enabled', '1')])
]
assert g._session.headers[_const.CSRF_TOKEN_HEADER] == 'THE-CSRF-TOKEN'
assert g._token == {'something': 'asdf', 'foo': 'bar'}
assert g.created is True
with pytest.raises(RuntimeError, match=r'^Gallery was already created$'):
g.create()
