class ReleasesTest(BaseTest):
def setUp(self):
super(ReleasesTest,self).setUp()
self.scrapper = Scrapper(self.config)
self.opener = Mock()
self.opener.open.return_value = open('test_data/2011.html')
self.scrapper._opener = self.opener
def test_month_as_member(self):
self.assertEqual(month_as_number('August'), 8)
def _get_soup(self):
return self.scrapper._soupify('')
def _get_existing(self):
self.scrapper._cfg['existing_file'] = 'test_data/existing'
self.scrapper._existing = self.scrapper._load_existing()
return self.scrapper._existing
def test_load_existing_empty(self):
self.assertEqual(self.scrapper._load_existing(), set())
def test_load_existing(self):
self.assertEqual(len(self._get_existing()), 173)
def test_soupify(self):
self.assertEqual(self._get_soup().name, u'[document]')
def test_albums_for_month(self):
albums = self.scrapper._albums_for_month(self._get_soup(), 'January')
self.assertEqual(len(albums), 25)
self.assertEqual(albums[13], (u'Sirenia', u'The Enigma of Life', 21))
def test_albums_for_month_with_existing(self):
self._get_existing()
albums = self.scrapper._albums_for_month(self._get_soup(), 'March')
self.assertEqual(len(albums), 4)
self.assertEqual(albums[0], (u'Children of Bodom', u'Relentless Reckless Forever', 2))
def test_albumgs_for_missing_month(self):
self._get_existing()
albums = self.scrapper._albums_for_month(self._get_soup(), 'August')
self.assertEqual(len(albums), 0)
def test_albums(self):
self._get_existing()
albums = self.scrapper.albums
total = 0
for _,lst in albums:
total += len(lst)
self.assertEqual(total, 15)
def setUp(self):
super(ReleasesTest,self).setUp()
self.scrapper = Scrapper(self.config)
self.opener = Mock()
self.opener.open.return_value = open('test_data/2011.html')
self.scrapper._opener = self.opener
