def test_get_episode(self):
''' Tests getting episode from video's field. '''
configure_for_unittest()
from model import Video, Show, Season, Episode
show = Show(title='House of Cards').save()
season1 = Season(show=show, number=1).save()
season2 = Season(show=show, number=2).save()
episode1 = Episode(season=season1,
number=1,
release_date=dt(2010, 1, 1)).save()
episode2 = Episode(season=season1, number=2).save()
episode3 = Episode(season=season2, number=1).save()
video = Video(link='vk.com').save()
show.seasons.connect(season1)
show.seasons.connect(season2)
season1.episodes.connect(episode1)
season1.episodes.connect(episode2)
season2.episodes.connect(episode3)
episode1.videos.connect(video)
video.refresh()
self.assertEqual(video.episode.get().number, 1)
def test_get_episode(self):
''' Tests getting episode from video's field. '''
configure_for_unittest()
from model import Video, Show, Season, Episode
show = Show(title='House of Cards').save()
season1 = Season(show=show, number=1).save()
season2 = Season(show=show, number=2).save()
episode1 = Episode(season=season1, number=1, release_date=dt(2010,1,1)).save()
episode2 = Episode(season=season1, number=2).save()
episode3 = Episode(season=season2, number=1).save()
video = Video(link='vk.com').save()
show.seasons.connect(season1)
show.seasons.connect(season2)
season1.episodes.connect(episode1)
season1.episodes.connect(episode2)
season2.episodes.connect(episode3)
episode1.videos.connect(video)
video.refresh()
self.assertEqual(video.episode.get().number, 1)
