1
0
mirror of https://github.com/moparisthebest/SickRage synced 2024-12-16 04:52:19 -05:00
SickRage/lib/fanart/tests/test_items.py

28 lines
899 B
Python
Raw Normal View History

from unittest import TestCase
import os
from fanart.items import LeafItem
from httpretty import httprettified, HTTPretty
from fanart.tests import LOCALDIR
class LeafItemTestCase(TestCase):
def setUp(self):
self.leaf = LeafItem(id=11977, likes=2, url='http://test.tv/50x50.txt')
def test_str(self):
self.assertEqual(str(self.leaf), 'http://test.tv/50x50.txt')
@httprettified
def test_content(self):
with open(os.path.join(LOCALDIR, 'response/50x50.png')) as fp:
body = fp.read()
HTTPretty.register_uri(
HTTPretty.GET,
'http://test.tv/50x50.txt',
body=body
)
self.assertEqual(self.leaf.content(), body)
self.assertEqual(len(HTTPretty.latest_requests), 1)
self.assertEqual(self.leaf.content(), body) # Cached
self.assertEqual(len(HTTPretty.latest_requests), 1)