def test_remove_show(self): show1 = self.db.add_show(Show(u"random show", url=u"z")) show2 = self.db.add_show(Show(u"other show", url=u"x")) self.db.commit() now = date.today() episode1 = Episode(u"first", 1, 1, now, u"x", 1) episode2 = Episode(u"second",1, 2, now, u"x", 1) episode3 = Episode(u"first", 1, 1, now, u"x", 1) self.db.add_episode(episode1, show1) self.db.add_episode(episode2, show1) self.db.add_episode(episode3, show2) self.db.commit() episodes = self.db.get_episodes() self.assertEqual(3, len(episodes)) self.db.remove_show(show1.id) self.db.commit() episodes = self.db.get_episodes() self.assertEqual(1, len(episodes)) self.assertIn(episode3, episodes)
def test_remove_before_with_show(self): show1 = self.db.add_show(Show(u"some show", url=u"a")) show2 = self.db.add_show(Show(u"some other show", url=u"b")) today = date.today() yesterday = today - timedelta(1) episode1 = Episode(u"episode1", 1, 1, yesterday, u"x", 1) episode2 = Episode(u"episode1", 1, 2, yesterday, u"x", 1) episode3 = Episode(u"episode1", 1, 2, yesterday, u"x", 1) self.db.add_episode(episode1, show1) self.db.add_episode(episode2, show1) self.db.add_episode(episode3, show2) self.db.commit() episodes = self.db.get_episodes(basedate = yesterday, n_days=10) self.assertIn(episode1, episodes) self.assertIn(episode2, episodes) self.assertIn(episode3, episodes) self.db.remove_before(today, show=show1) episodes = self.db.get_episodes(basedate = yesterday, n_days=10) self.assertNotIn(episode1, episodes) self.assertNotIn(episode2, episodes) self.assertIn(episode3, episodes)
def test_get_enabled_shows(self): show1 = self.db.add_show(Show(u"1", url=u"1")) show1.enabled = False show2 = self.db.add_show(Show(u"2", url=u"2")) self.db.commit() shows = self.db.get_enabled_shows() self.assertNotIn(show1, shows) self.assertIn(show2, shows)
def test_clear(self): today = date.today() show = self.db.add_show(Show(u"some show", url=u"urlX")) self.assertEqual(0, len(self.db.get_episodes())) episode1 = Episode(u"e", 1, 1, today, u"x", 1) episode2 = Episode(u"e", 1, 2, today, u"x", 1) episode3 = Episode(u"e", 1, 3, today, u"x", 1) episode4 = Episode(u"e", 1, 3, today, u"x", 1) episode5 = Episode(u"e", 1, 4, today, u"x", 1) self.db.add_episode(episode1, show) self.db.add_episode(episode2, show) self.db.add_episode(episode3, show) self.db.commit() episodes = self.db.get_episodes() self.assertEqual(3, len(episodes)) self.assertIn(episode1, episodes) self.assertIn(episode2, episodes) self.assertIn(episode3, episodes) self.db.clear() self.assertEqual(0, len(self.db.get_episodes())) self.db.add_episode(episode4, show) self.db.add_episode(episode5, show) self.db.commit() episodes = self.db.get_episodes() self.assertEqual(2, len(episodes)) self.assertIn(episode4, episodes) self.assertIn(episode5, episodes)
def test_remove_before(self): show = self.db.add_show(Show(u"some show")) today = date.today() yesterday = today - timedelta(1) before = yesterday - timedelta(1) tomorrow = today + timedelta(1) episode1 = Episode(u"episode1", 1, 1, before, u"x", 1) episode2 = Episode(u"episode2", 1, 2, yesterday, u"x", 2) episode3 = Episode(u"episode3", 1, 3, today, u"x", 3) episode4 = Episode(u"episode4", 1, 4, tomorrow, u"x", 4) self.db.add_episode(episode1, show) self.db.add_episode(episode2, show) self.db.add_episode(episode3, show) self.db.add_episode(episode4, show) self.db.commit() episodes = self.db.get_episodes(basedate = before, n_days=10) self.assertIn(episode1, episodes) self.assertIn(episode2, episodes) self.assertIn(episode3, episodes) self.assertIn(episode4, episodes) self.db.remove_before(today) episodes = self.db.get_episodes(basedate = before, n_days=10) self.assertNotIn(episode1, episodes) self.assertNotIn(episode2, episodes) self.assertIn(episode3, episodes) self.assertIn(episode4, episodes)
def test_search(self): show = self.db.add_show(Show(u"some show")) ep1 = Episode(u"first episode", 3, 10, date.today(), u"FOO", 30) ep2 = Episode(u"Second episode", 3, 12, date.today(), u"FOO", 32) self.db.add_episode(ep1, show) self.db.add_episode(ep2, show) self.db.commit() episodes = self.db.search(u"first") self.assertIn(ep1, episodes) self.assertNotIn(ep2, episodes) episodes = self.db.search(u"second") self.assertNotIn(ep1, episodes) self.assertIn(ep2, episodes) episodes = self.db.search(u"episode") self.assertIn(ep1, episodes) self.assertIn(ep2, episodes) episodes = self.db.search(u"some show") self.assertIn(ep1, episodes) self.assertIn(ep2, episodes)
def test_send_mail_tls(self):
show1 = Show(u"Columbo", u"")
ep1 = Episode(u"Episode 1", 1, 1, date(2014, 8, 10), u"x", 1)
ep2 = Episode(u"Episode 2", 1, 2, date(2014, 12, 1), u"x", 2)
ep1.show = show1
ep2.show = show1
message = NewEpisodesNotification([ep1, ep2],
"[%airdate] %show %seasonx%epnum - %eptitle","%Y-%m-%d")
notifier = EmailNotifier("localhost", 97, MockSMTP)
notifier.use_tls = True
message.send(notifier, "*****@*****.**")
self.assertEqual(self.smtp.count, 1)
self.assertEqual(self.smtp.tls, 1)
self.assertEqual(self.smtp.user, None)
self.assertEqual(self.smtp.password, None)
self.assertEqual(self.smtp.from_addr, "*****@*****.**")
self.assertEqual(self.smtp.to_addr, "*****@*****.**")
self.assertEqual(self.smtp.msg,
"From: [email protected]\n"
"To: [email protected]\n"
"Subject: Upcoming TV episodes\n"
"Your upcoming episodes:\n\n"
"[2014-08-10] Columbo 1x01 - Episode 1\n"
"[2014-12-01] Columbo 1x02 - Episode 2\n")
def test_send_mail_pretend(self):
show1 = Show(u"Columbo", u"")
ep1 = Episode(u"Episode 1", 1, 1, date(2014, 8, 10), u"x", 1)
ep2 = Episode(u"Episode 2", 1, 2, date(2014, 12, 1), u"x", 2)
ep1.show = show1
ep2.show = show1
stdout = sys.stdout
sys.stdout = StringIO()
message = NewEpisodesNotification([ep1, ep2],
"[%airdate] %show %seasonx%epnum - %eptitle","%Y-%m-%d")
notifier = EmailNotifier("localhost", 97, MockSMTP)
message.send(notifier, "*****@*****.**", pretend=True)
self.assertEqual(self.smtp.count, 0)
output = sys.stdout.getvalue()
sys.stdout = stdout
# StringIO adds an extra \n
self.assertEqual(output,
"From: [email protected]\n"
"To: [email protected]\n"
"Subject: Upcoming TV episodes\n"
"Your upcoming episodes:\n\n"
"[2014-08-10] Columbo 1x01 - Episode 1\n"
"[2014-12-01] Columbo 1x02 - Episode 2\n\n")
def setUp(self): self.show = Show(u"Test show 36") self.show.show_id = 36 then = date(2008, 1, 1) self.episode = Episode(u"Episode 41", 2, 5, then, u"NX01", 3) self.episode.show = self.show self.smtp = MockSMTP(None, None) self.smtp.reset()
def test_add_episode(self): show = Show(u"some show", url=u"foo") show = self.db.add_show(show) self.db.commit() ep1 = Episode(u"Some episode", 3, 10, date.today(), u"FOO", 30) ep2 = Episode(u"No episode", 3, 12, date.today(), u"FOO", 32) self.db.add_episode(ep1, show) self.db.add_episode(ep2, show) self.db.commit() episodes = self.db.get_episodes() self.assertTrue(ep1 in episodes) self.assertTrue(ep2 in episodes)
def test_rollback(self): show = Show(u"some show") show = self.db.add_show(show) self.db.commit() ep1 = Episode(u"first", 3, 10, date.today(), u"FOO", 30) self.db.add_episode(ep1, show) self.db.rollback() ep2 = Episode(u"Second", 3, 12, date.today(), u"FOO", 32) self.db.add_episode(ep2, show) self.db.commit() episodes = self.db.get_episodes() self.assertFalse(ep1 in episodes) self.assertTrue(ep2 in episodes)