class IMDBPosterLoaderCase(TestCase):
"""IMDb poster loader case"""
def setUp(self):
self.loader = IMDBPosterLoader()
self._mock_fetch()
def _mock_fetch(self):
"""Mock fetching data"""
self._orig_fetch =self.loader._fetch_html
self.loader._fetch_html = MagicMock()
def test_get_poster(self):
"""Test get poster"""
self.loader._fetch_html.return_value = get_fixture('imdb.html')
self.loader.get_poster('family gut').should.be\
.equal('http://ia.media-imdb.com/images/M/MV5BMTY4MjIwMjcxMF5BMl5'
'BanBnXkFtZTYwMDA1NDg5._V1_SY44_CR0,0,32,44_.jpg')
def test_get_default_poster_if_not_exists(self):
"""Test get default poster if not exists"""
self.loader._fetch_html.return_value =\
get_fixture('imdb_none.html')
self.loader.get_poster('family gut').should.be.\
equal(self.loader.default_poster)
def setUp(self):
self.loader = IMDBPosterLoader()
self._mock_fetch()