async def test_upload_fails_to_find_error_message(tmp_path, mocker,
httpserver):
html_dump_mock = mocker.patch('upsies.utils.html.dump')
response = 'unexpected html'
httpserver.expect_request(
uri='/upload.php',
method='POST',
).respond_with_data(response)
torrent_file = tmp_path / 'foo.torrent'
torrent_file.write_bytes(b'mocked torrent metainfo')
tracker = NblTracker(options={
'username': '******',
'password': '******',
'base_url': httpserver.url_for(''),
'announce': 'http://nbl.local/announce',
'exclude': 'some files',
}, )
tracker._logout_url = 'logout.php'
tracker._auth_key = 'mocked auth key'
tracker_jobs_mock = Mock(
create_torrent_job=Mock(output=(str(torrent_file), )),
mediainfo_job=Mock(output=('mocked mediainfo', )),
tvmaze_job=Mock(output=('12345', )),
category_job=Mock(output=('Season', )),
)
with pytest.raises(RuntimeError,
match=(r'^Failed to find error message. '
r'See upload.html for more information.$')):
await tracker.upload(tracker_jobs_mock)
assert html_dump_mock.call_args_list == [call(response, 'upload.html')]
async def test_upload_finds_error_message(message, exp_error, tmp_path, mocker,
httpserver):
html_dump_mock = mocker.patch('upsies.utils.html.dump')
httpserver.expect_request(
uri='/upload.php',
method='POST',
).respond_with_data(f'''
<html>
<div id="messagebar">{message}</div>
</html>
''')
torrent_file = tmp_path / 'foo.torrent'
torrent_file.write_bytes(b'mocked torrent metainfo')
tracker = NblTracker(options={
'username': '******',
'password': '******',
'base_url': httpserver.url_for(''),
'announce': 'http://nbl.local/announce',
'exclude': 'some files',
}, )
tracker._logout_url = 'logout.php'
tracker._auth_key = 'mocked auth key'
tracker_jobs_mock = Mock(
create_torrent_job=Mock(output=(str(torrent_file), )),
mediainfo_job=Mock(output=('mocked mediainfo', )),
tvmaze_job=Mock(output=('12345', )),
category_job=Mock(output=('Season', )),
)
with pytest.raises(errors.RequestError,
match=rf'^{re.escape(exp_error)}$'):
await tracker.upload(tracker_jobs_mock)
assert html_dump_mock.call_args_list == []
def test_is_logged_in():
tracker = NblTracker(options={
'username': '******',
'password': '******',
'base_url': 'http://nbl.local',
'announce': 'http://nbl.local/announce',
'exclude': 'some files',
}, )
# tracker.logged_in must be True if "_logout_url" and "_auth_key" are set
assert tracker.is_logged_in is False
tracker._logout_url = 'asdf'
assert tracker.is_logged_in is False
tracker._auth_key = 'asdf'
assert tracker.is_logged_in is True
delattr(tracker, '_logout_url')
assert tracker.is_logged_in is False
tracker._logout_url = 'asdf'
assert tracker.is_logged_in is True
delattr(tracker, '_auth_key')
assert tracker.is_logged_in is False
async def test_login_does_nothing_if_already_logged_in(mocker):
get_mock = mocker.patch('upsies.utils.http.get', AsyncMock())
post_mock = mocker.patch('upsies.utils.http.post', AsyncMock())
tracker = NblTracker(options={
'username': '******',
'password': '******',
'base_url': 'http://nbl.local',
'announce': 'http://nbl.local/announce',
'exclude': 'some files',
}, )
tracker._logout_url = 'anything'
tracker._auth_key = 'something'
assert tracker.is_logged_in
await tracker.login()
assert tracker.is_logged_in
assert get_mock.call_args_list == []
assert post_mock.call_args_list == []
assert tracker._logout_url == 'anything'
assert tracker._auth_key == 'something'
async def test_logout(logout_url, auth_key, mocker):
get_mock = mocker.patch('upsies.utils.http.get', AsyncMock())
post_mock = mocker.patch('upsies.utils.http.post', AsyncMock())
tracker = NblTracker(options={
'username': '******',
'password': '******',
'base_url': 'http://nbl.local',
'announce': 'http://nbl.local/announce',
'exclude': 'some files',
}, )
if logout_url is not None:
tracker._logout_url = logout_url
if auth_key is not None:
tracker._auth_key = auth_key
await tracker.logout()
if logout_url is not None:
assert get_mock.call_args_list == [
call(logout_url, user_agent=True),
]
else:
assert get_mock.call_args_list == []
assert post_mock.call_args_list == []
assert not hasattr(tracker, '_logout_url')
assert not hasattr(tracker, '_auth_key')
async def test_upload_succeeds(category, exp_category_code, ignore_dupes,
exp_data, tmp_path, mocker, httpserver):
class Handler(RequestHandler):
def handle(self, request):
request_seen = {
'method': request.method,
'User-Agent': request.headers.get('User-Agent', ''),
'multipart/form-data': dict(request.form),
}
# werkzeug.Request stores files in the `files` property
for field, filestorage in request.files.items():
request_seen['multipart/form-data'][field] = filestorage.read()
self.requests_seen.append(request_seen)
# Upload form redirects to torrent page
return Response(
status=307,
headers={'Location': '/torrents.php?id=123'},
)
handler = Handler()
httpserver.expect_request(
uri='/upload.php',
method='POST',
).respond_with_handler(handler, )
torrent_file = tmp_path / 'foo.torrent'
torrent_file.write_bytes(b'mocked torrent metainfo')
tracker = NblTracker(options={
'username': '******',
'password': '******',
'base_url': httpserver.url_for(''),
'announce': 'http://nbl.local/announce',
'exclude': 'some files',
'ignore_dupes': ignore_dupes,
}, )
tracker._logout_url = 'logout.php'
tracker._auth_key = 'mocked auth key'
tracker_jobs_mock = Mock(
create_torrent_job=Mock(output=(str(torrent_file), )),
mediainfo_job=Mock(output=('mocked mediainfo', )),
tvmaze_job=Mock(output=('12345', )),
category_job=Mock(output=(category, )),
)
torrent_page_url = await tracker.upload(tracker_jobs_mock)
assert torrent_page_url == httpserver.url_for('/torrents.php?id=123')
exp_form_data = {
'MAX_FILE_SIZE': '1048576',
'auth': 'mocked auth key',
'category': exp_category_code,
'desc': 'mocked mediainfo',
'file_input': b'mocked torrent metainfo',
'fontfont': '-1',
'fontsize': '-1',
'genre_tags': '',
'image': '',
'media': 'mocked mediainfo',
'mediaclean': '[mediainfo]mocked mediainfo[/mediainfo]',
'submit': 'true',
'tags': '',
'title': '',
'tvmazeid': '12345',
}
exp_form_data.update(exp_data)
assert handler.requests_seen == [{
'method': 'POST',
'User-Agent': f'{__project_name__}/{__version__}',
'multipart/form-data': exp_form_data,
}]