forked from Almenon/reddit_episode_bot
-
Notifications
You must be signed in to change notification settings - Fork 0
/
unit_tester.py
72 lines (54 loc) · 2.49 KB
/
unit_tester.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
import unittest
from post_parser import parse,ParseError
from omdb_requester import get_info,OmdbError
from comment_formatter import format_post_reply
with open('info/badPosts.txt') as f:
badPosts = f.readlines()
with open('info/goodPosts.txt') as f:
goodPosts = f.readlines()
class TestRegex(unittest.TestCase):
def test_bad_posts(self):
for badPost in badPosts:
with self.assertRaises(ParseError, msg=badPost):
parse(badPost)
def test_good_posts(self):
for goodPost in goodPosts:
try:
parse(goodPost)
except ParseError:
self.fail(goodPost)
class TestOmdb(unittest.TestCase):
def test_bad_show(self):
with self.assertRaises(OmdbError):
get_info('seafoijosefijo',1,1)
def test_good_show(self):
try:
get_info() # default param is game of thrones, s1e1
except OmdbError as e:
self.fail(e)
class TestCommentFormatter(unittest.TestCase):
# def test_all_subreddits(self):
# try:
# [format_post_reply("s1e1",sub) for sub in subreddits]
# except Exception as e:
# self.fail(e)
def test_good_reply(self):
goodReply = "#####	 \n######	 \n####	 \n" \
"[**I Wasn't Ready**](http://www.imdb.com/title/tt2400770) [7.8 ★] | " \
"[Watch on Netflix](http://www.netflix.com/title/70242311) | [imdb](http://www.imdb.com/title/tt2400770)" \
"\n\n> [Piper Chapman is sent to jail as a result of her relationship with a drug smuggler.](/spoiler)"
reply = format_post_reply("s1e1","orangeisthenewblack")
self.assertEqual(reply,goodReply)
def test_no_netflix(self):
goodReply = "#####	 \n######	 \n####	 \n" \
"[**Tourist Trapped**](http://www.imdb.com/title/tt2152239) [8.5 ★] | [imdb](http://www.imdb.com/title/tt2152239)" \
"\n\n> [After finding a strange book in the forest, Dipper begins to learn about the dark side of Gravity Falls," \
" and suspects that Mabel's new boyfriend is a zombie.](#spoiler)"
reply = format_post_reply("s1e1","gravityfalls")
self.assertEqual(reply,goodReply)
# def test_missing_rating(self):
# this should be tested once you find another test case
# last example got updated
# def test_missing_plot(self):
# this should be tested once you find another test case
# last example got updated