def test_expire_no_check(self, execute_task):
persist['auth_tokens'] = {'default': None}
def test_run():
# Run the task and check tvdb data was populated.
task = execute_task('test_mark_expired')
entry = task.find_entry(title='House.S02E02.hdtv')
assert entry['tvdb_ep_name'] == 'Autopsy'
# Run the task once, this populates data from tvdb
test_run()
# Should not expire as it was checked less then an hour ago
persist['last_check'] = datetime.utcnow() - timedelta(hours=1)
with mock.patch(
'requests.sessions.Session.request',
side_effect=Exception(
'Tried to expire or lookup, less then an hour since last check'
),
):
# Ensure series is not marked as expired
with Session() as session:
mark_expired(session)
ep = (session.query(TVDBEpisode).filter(
TVDBEpisode.series_id == 73255).filter(
TVDBEpisode.episode_number == 2).filter(
TVDBEpisode.season_number == 2).first())
assert not ep.expired
assert not ep.series.expired
def test_expire_no_check(self, execute_task):
persist['auth_tokens'] = {'default': None}
def test_run():
# Run the task and check tvdb data was populated.
task = execute_task('test_mark_expired')
entry = task.find_entry(title='House.S02E02.hdtv')
assert entry['tvdb_ep_name'] == 'Autopsy'
# Run the task once, this populates data from tvdb
test_run()
# Should not expire as it was checked less then an hour ago
persist['last_check'] = datetime.utcnow() - timedelta(hours=1)
with mock.patch(
'requests.sessions.Session.request',
side_effect=Exception('Tried to expire or lookup, less then an hour since last check'),
):
# Ensure series is not marked as expired
with Session() as session:
mark_expired(session)
ep = (
session.query(TVDBEpisode)
.filter(TVDBEpisode.series_id == 73255)
.filter(TVDBEpisode.episode_number == 2)
.filter(TVDBEpisode.season_number == 2)
.first()
)
assert not ep.expired
assert not ep.series.expired
def test_expire_check(self, execute_task):
persist['auth_tokens'] = {'default': None}
def test_run():
# Run the task and check tvdb data was populated.
task = execute_task('test_mark_expired')
entry = task.find_entry(title='House.S02E02.hdtv')
assert entry['tvdb_ep_name'] == 'Autopsy'
# Run the task once, this populates data from tvdb
test_run()
# Should expire
persist['last_check'] = datetime.utcnow() - timedelta(hours=3)
expired_data = [
{
"id": 73255,
"lastUpdated": 1458186055
},
{
"id": 295743,
"lastUpdated": 1458186088
},
]
# Ensure series is marked as expired
with mock.patch.object(TVDBRequest, 'get', side_effect=[expired_data]):
with Session() as session:
mark_expired(session)
ep = (session.query(TVDBEpisode).filter(
TVDBEpisode.series_id == 73255).filter(
TVDBEpisode.episode_number == 2).filter(
TVDBEpisode.season_number == 2).first())
assert ep.expired
assert ep.series.expired
# Run the task again, should be re-populated from tvdb
test_run()
with Session() as session:
ep = (session.query(TVDBEpisode).filter(
TVDBEpisode.series_id == 73255).filter(
TVDBEpisode.episode_number == 2).filter(
TVDBEpisode.season_number == 2).first())
assert not ep.expired
assert not ep.series.expired
def test_expire_check(self, execute_task):
persist['auth_tokens'] = {'default': None}
def test_run():
# Run the task and check tvdb data was populated.
task = execute_task('test_mark_expired')
entry = task.find_entry(title='House.S02E02.hdtv')
assert entry['tvdb_ep_name'] == 'Autopsy'
# Run the task once, this populates data from tvdb
test_run()
# Should expire
persist['last_check'] = datetime.utcnow() - timedelta(hours=3)
expired_data = [
{"id": 73255, "lastUpdated": 1458186055},
{"id": 295743, "lastUpdated": 1458186088},
]
# Ensure series is marked as expired
with mock.patch.object(TVDBRequest, 'get', side_effect=[expired_data]):
with Session() as session:
mark_expired(session)
ep = (
session.query(TVDBEpisode)
.filter(TVDBEpisode.series_id == 73255)
.filter(TVDBEpisode.episode_number == 2)
.filter(TVDBEpisode.season_number == 2)
.first()
)
assert ep.expired
assert ep.series.expired
# Run the task again, should be re-populated from tvdb
test_run()
with Session() as session:
ep = (
session.query(TVDBEpisode)
.filter(TVDBEpisode.series_id == 73255)
.filter(TVDBEpisode.episode_number == 2)
.filter(TVDBEpisode.season_number == 2)
.first()
)
assert not ep.expired
assert not ep.series.expired
