def setUp(self):
self.__parser = WallPageParser(ignore_errors = False)
class TestWallParser(unittest.TestCase):
'''Tests vk.com wall parser.'''
def setUp(self):
self.__parser = WallPageParser(ignore_errors = False)
def test_invalid_page(self):
'''Testing parsing of invalid page.'''
self.assertRaises(ParseError, lambda:
self.__parser.parse(open('wall_parser/invalid_page.html').read().decode('cp1251')))
def test_group_wall(self):
'''Testing parsing of group wall'''
self.__test_parsing(
open('wall_parser/group_profile_page.html').read().decode('cp1251'), {
'user_name': u'Хабрахабр',
'user_photo': 'http://cs11159.vk.com/g20629724/a_ba3bb3dc.jpg',
'posts': 10,
}
)
def test_user_empty_wall(self):
'''Testing parsing of empty wall'''
self.__test_parsing(
open('wall_parser/user_profile_page_with_empty_wall.html').read().decode('cp1251'), {
'user_name': u'Григорий Бакунов',
'user_photo': 'http://cs4383.vk.com/u78983895/a_912f563f.jpg',
'posts': 0,
}
)
def test_user_wall(self):
'''Testing parsing of user wall'''
self.__test_parsing(
open('wall_parser/user_profile_page.html').read().decode('cp1251'), {
'user_name': u'Павел Дуров',
'user_photo': 'http://vk.com/u00001/a_a964f9a2.jpg',
'posts': 10,
}
)
def __test_parsing(self, html, etalon):
'''Runs the test with the specified data.'''
clear_run_data = self.__normalize_data(self.__parser.parse(html))
self.assertEqual(etalon, clear_run_data)
dirty_run_data = self.__normalize_data(self.__parser.parse(html))
self.assertEqual(etalon, clear_run_data)
self.assertEqual(etalon, dirty_run_data)
def __normalize_data(self, data):
'''Removes fields that don't exist in the etalon.'''
if 'posts' in data:
for post in data['posts']:
self.assertNotEqual(post['text'].strip(), '')
data['posts'] = len(data['posts'])
return data
