def test_command_update_1_comic(self):
response = xkcd.get_url(xkcd.api_url % "")
new_max_comic = json.loads(response.decode("utf-8"))["num"] - 1
xkcd.cur_max_comic = new_max_comic
output = xkcd.command_update()
expected_output = "1 new comic!\n"
self.assertEqual(output, expected_output)
def test_command_update_no_new_comics(self):
response = xkcd.get_url(xkcd.api_url % "")
max_comic = json.loads(response.decode("utf-8"))["num"]
xkcd.cur_max_comic = max_comic
output = xkcd.command_update()
expected_output = "No new comics.\n"
self.assertEqual(output, expected_output)
def test_update_search_db(self):
xkcd.cur_max_comic = 1
xkcd.command_update()
old_titles_loc = xkcd.titles_location
old_transcripts_loc = xkcd.transcripts_location
xkcd.titles_location = "test.txt"
xkcd.transcripts_location = "test2.txt"
fd = open("test.txt", "w")
fd.write(str(xkcd.cur_max_comic - 1))
fd.write(":'Test'\n")
fd.close()
fd = open("test2.txt", "w")
fd.write(str(xkcd.cur_max_comic - 1))
fd.write(":'Test'\n")
fd.close()
output = xkcd.command_update("search_db")
expected_output = "No new comics.\n1 comics not in title database " "found."
xkcd.titles_location = old_titles_loc
xkcd.transcripts_location = old_transcripts_loc
os.remove("test.txt")
os.remove("test2.txt")
self.assertEqual(output, expected_output)
def test_command_update(self):
xkcd.cur_max_comic = 1000
output = xkcd.command_update()
generic_output = " new comics!\n"
self.assertEqual(output[-13:], generic_output)