Pull out torrent validation into validator class

This commit is contained in:
Joseph Schorr 2017-02-15 12:12:19 -05:00
parent 2944a4e13d
commit 09b3cfd549
3 changed files with 83 additions and 45 deletions

View file

@ -0,0 +1,28 @@
import pytest
from httmock import urlmatch, HTTMock
from util.config.validators import ConfigValidationException
from util.config.validators.validate_torrent import BittorrentValidator
@pytest.mark.parametrize('unvalidated_config,expected', [
({}, ConfigValidationException),
({'BITTORRENT_ANNOUNCE_URL': 'http://faketorrent/announce'}, None),
])
def test_validate_torrent(unvalidated_config,expected):
announcer_hit = [False]
@urlmatch(netloc=r'faketorrent', path='/announce')
def handler(url, request):
announcer_hit[0] = True
return {'status_code': 200, 'content': ''}
with HTTMock(handler):
validator = BittorrentValidator()
if expected is not None:
with pytest.raises(expected):
validator.validate(unvalidated_config, None, None)
assert not announcer_hit[0]
else:
validator.validate(unvalidated_config, None, None)
assert announcer_hit[0]