mirror of
https://github.com/moparisthebest/SickRage
synced 2024-12-14 20:12:19 -05:00
47 lines
1.7 KiB
Python
47 lines
1.7 KiB
Python
|
import json
|
||
|
|
from fanart.errors import ResponseFanartError
|
||
|
|
import os
|
||
|
|
import unittest
|
||
|
|
from httpretty import HTTPretty, httprettified
|
||
|
|
from fanart.tv import *
|
||
|
|
from fanart.tests import LOCALDIR
|
||
|
|
os.environ['FANART_APIKEY'] = 'e3c7f0d0beeaf45b3a0dd3b9dd8a3338'
|
||
|
|
|
||
|
|
|
||
|
|
class TvItemTestCase(unittest.TestCase):
|
||
|
|
@httprettified
|
||
|
|
def test_get_wilfred(self):
|
||
|
|
with open(os.path.join(LOCALDIR, 'response/tv_239761.json')) as fp:
|
||
|
|
body = fp.read()
|
||
|
|
HTTPretty.register_uri(
|
||
|
|
HTTPretty.GET,
|
||
|
|
'http://api.fanart.tv/webservice/series/e3c7f0d0beeaf45b3a0dd3b9dd8a3338/239761/JSON/all/1/2',
|
||
|
|
body=body
|
||
|
|
)
|
||
|
|
wilfred = TvShow.get(id=239761)
|
||
|
|
self.assertEqual(wilfred.tvdbid, '239761')
|
||
|
|
with open(os.path.join(LOCALDIR, 'json/wilfred.json')) as fp:
|
||
|
|
self.assertEqual(json.loads(wilfred.json()), json.load(fp))
|
||
|
|
|
||
|
|
@httprettified
|
||
|
|
def test_get_dexter(self):
|
||
|
|
with open(os.path.join(LOCALDIR, 'response/tv_79349.json')) as fp:
|
||
|
|
body = fp.read()
|
||
|
|
HTTPretty.register_uri(
|
||
|
|
HTTPretty.GET,
|
||
|
|
'http://api.fanart.tv/webservice/series/e3c7f0d0beeaf45b3a0dd3b9dd8a3338/79349/JSON/all/1/2',
|
||
|
|
body=body
|
||
|
|
)
|
||
|
|
dexter = TvShow.get(id=79349)
|
||
|
|
self.assertEqual(dexter.tvdbid, '79349')
|
||
|
|
self.assertEqual(dexter, eval(repr(dexter)))
|
||
|
|
|
||
|
|
@httprettified
|
||
|
|
def test_get_null(self):
|
||
|
|
HTTPretty.register_uri(
|
||
|
|
HTTPretty.GET,
|
||
|
|
'http://api.fanart.tv/webservice/series/e3c7f0d0beeaf45b3a0dd3b9dd8a3338/79349/JSON/all/1/2',
|
||
|
|
body='null'
|
||
|
|
)
|
||
|
|
self.assertRaises(ResponseFanartError, TvShow.get, id=79349)
|