mirror of
https://github.com/moparisthebest/SickRage
synced 2024-12-16 04:52:19 -05:00
28 lines
899 B
Python
28 lines
899 B
Python
|
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)
|