def test_check_download(self):
tracker = LostFilmPlugin()
response = Response()
response.status_code = 200
self.assertEqual(tracker.check_download(response), Status.Ok)
response.status_code = 302
response.headers['Location'] = '/'
self.assertEqual(tracker.check_download(response), Status.NotFound)
response.status_code = 500
response.headers['Location'] = '/'
self.assertEqual(tracker.check_download(response), Status.Error)
def test_check_download(self):
tracker = LostFilmPlugin()
response = Response()
response.status_code = 200
assert tracker.check_download(response) == Status.Ok
response.status_code = 302
response.headers['Location'] = '/'
assert tracker.check_download(response) == Status.NotFound
response.status_code = 200
response._content = ('<!--\r\n'
'location.replace("/new/");\r\n'
'//-->').encode('utf-8')
assert tracker.check_download(response) == Status.NotFound
response.status_code = 500
response.headers['Location'] = '/'
# Should be error even with Location header
assert tracker.check_download(response) == Status.Error
def test_check_download(self):
tracker = LostFilmPlugin()
response = Response()
response.status_code = 200
assert tracker.check_download(response) == Status.Ok
response.status_code = 302
response.headers['Location'] = '/'
assert tracker.check_download(response) == Status.NotFound
response.status_code = 200
response._content = (
'<!--\r\n'
'<meta http-equiv="refresh" content="0; url=/">;\r\n'
'//-->').encode('utf-8')
assert tracker.check_download(response) == Status.NotFound
response.status_code = 500
response.headers['Location'] = '/'
# Should be error even with Location header
assert tracker.check_download(response) == Status.Error
def setUp(self):
super(TestLostFilmTrackerPlugin, self).setUp()
self.tracker_settings = TrackerSettings(10, None)
self.plugin = LostFilmPlugin()
self.plugin.init(self.tracker_settings)
