def test_remove_episodes_before(self): now = date.today() then = now - timedelta(3) show2 = Show(u"B", url=u"b") show2 = self.db.add_show(show2) episode1 = Episode(u"e", 1, 1, now, u"x", 1) episode2 = Episode(u"e", 1, 2, then, u"x", 1) episode3 = Episode(u"e", 1, 3, now, u"x", 1) episode4 = Episode(u"e", 1, 4, then, u"x", 1) self.db.add_episode(episode1, self.show) self.db.add_episode(episode2, self.show) self.db.add_episode(episode3, show2) self.db.add_episode(episode4, show2) episodes = self.db.get_episodes(basedate = then, n_days=10) self.assertEqual(4, len(episodes)) show2.remove_episodes_before(self.db, now) episodes = self.db.get_episodes(basedate = then, n_days=10) self.assertEqual(3, len(episodes)) self.show.remove_episodes_before(self.db, now) episodes = self.db.get_episodes(basedate = then, n_days=10) self.assertEqual(2, len(episodes))
def test_remove_episodes_before(self):
now = date.today()
then = now - timedelta(3)
show2 = Show(u"B", url=u"b")
show2 = self.db.add_show(show2)
episode1 = Episode(u"e", 1, 1, now, u"x", 1)
episode2 = Episode(u"e", 1, 2, then, u"x", 1)
episode3 = Episode(u"e", 1, 3, now, u"x", 1)
episode4 = Episode(u"e", 1, 4, then, u"x", 1)
self.db.add_episode(episode1, self.show)
self.db.add_episode(episode2, self.show)
self.db.add_episode(episode3, show2)
self.db.add_episode(episode4, show2)
episodes = self.db.get_episodes(basedate=then, n_days=10)
self.assertEqual(4, len(episodes))
show2.remove_episodes_before(self.db, now)
episodes = self.db.get_episodes(basedate=then, n_days=10)
self.assertEqual(3, len(episodes))
self.show.remove_episodes_before(self.db, now)
episodes = self.db.get_episodes(basedate=then, n_days=10)
self.assertEqual(2, len(episodes))
class ShowTest(TestCase): def setUp(self): self.path = mktemp() self.db = Database(self.path) self.show = Show(u"A", url=u"a") self.show = self.db.add_show(self.show) def tearDown(self): unlink(self.path) def test_str_and_repr(self): self.assertEqual(str(self.show), "Show: A") self.assertEqual(repr(self.show), 'Show(u"A", u"a")') def test_equality(self): show = Show(u"A", url=u"b") self.assertNotEqual(show, self.show) show = Show(u"B", url=u"a") self.assertNotEqual(show, self.show) show = Show(u"A", url=u"a") self.assertEqual(show, self.show) def test_remove_episodes_before(self): now = date.today() then = now - timedelta(3) show2 = Show(u"B", url=u"b") show2 = self.db.add_show(show2) episode1 = Episode(u"e", 1, 1, now, u"x", 1) episode2 = Episode(u"e", 1, 2, then, u"x", 1) episode3 = Episode(u"e", 1, 3, now, u"x", 1) episode4 = Episode(u"e", 1, 4, then, u"x", 1) self.db.add_episode(episode1, self.show) self.db.add_episode(episode2, self.show) self.db.add_episode(episode3, show2) self.db.add_episode(episode4, show2) episodes = self.db.get_episodes(basedate = then, n_days=10) self.assertEqual(4, len(episodes)) show2.remove_episodes_before(self.db, now) episodes = self.db.get_episodes(basedate = then, n_days=10) self.assertEqual(3, len(episodes)) self.show.remove_episodes_before(self.db, now) episodes = self.db.get_episodes(basedate = then, n_days=10) self.assertEqual(2, len(episodes))
class ShowTest(TestCase):
def setUp(self):
self.path = mktemp()
self.db = Database(self.path)
self.show = Show(u"A", url=u"a")
self.show = self.db.add_show(self.show)
def tearDown(self):
unlink(self.path)
def test_str_and_repr(self):
self.assertEqual(str(self.show), "Show: A")
self.assertEqual(repr(self.show), 'Show(u"A", u"a")')
def test_equality(self):
show = Show(u"A", url=u"b")
self.assertNotEqual(show, self.show)
show = Show(u"B", url=u"a")
self.assertNotEqual(show, self.show)
show = Show(u"A", url=u"a")
self.assertEqual(show, self.show)
def test_remove_episodes_before(self):
now = date.today()
then = now - timedelta(3)
show2 = Show(u"B", url=u"b")
show2 = self.db.add_show(show2)
episode1 = Episode(u"e", 1, 1, now, u"x", 1)
episode2 = Episode(u"e", 1, 2, then, u"x", 1)
episode3 = Episode(u"e", 1, 3, now, u"x", 1)
episode4 = Episode(u"e", 1, 4, then, u"x", 1)
self.db.add_episode(episode1, self.show)
self.db.add_episode(episode2, self.show)
self.db.add_episode(episode3, show2)
self.db.add_episode(episode4, show2)
episodes = self.db.get_episodes(basedate=then, n_days=10)
self.assertEqual(4, len(episodes))
show2.remove_episodes_before(self.db, now)
episodes = self.db.get_episodes(basedate=then, n_days=10)
self.assertEqual(3, len(episodes))
self.show.remove_episodes_before(self.db, now)
episodes = self.db.get_episodes(basedate=then, n_days=10)
self.assertEqual(2, len(episodes))
