def testScrapeMessages(self): """Test scrapeMessages function.""" countRequestedStart = bitcointalk.countRequested data = scrapeMessages(14, 1) countRequestedEnd = bitcointalk.countRequested self.assertEqual(countRequestedEnd - countRequestedStart, 1) self.assertEqual(len(data), 2) # Make sure we can pull in the associated messages without error pg.selectMessages([53, 56])
def testScrapeTopic(self): """Test scrapeTopic function.""" countRequestedStart = bitcointalk.countRequested datumFirst = scrapeTopic(14) datumSecond = scrapeTopic(14) countRequestedEnd = bitcointalk.countRequested self.assertEqual(countRequestedEnd - countRequestedStart, 1) datumExpected = { 'id': 14, 'name': 'Break on the supply\'s increase', 'board': 7, 'num_pages': 1 } self.assertEqual(datumFirst['count_read'], datumSecond['count_read']) self.assertEqual(datumFirst.pop('count_read') > 3057, True) self.assertEqual(datumSecond.pop('count_read') > 3057, True) self.assertEqual(datumExpected, datumFirst) self.assertEqual(datumExpected, datumSecond) self.assertEqual(datumFirst, datumSecond) # Make sure we can pull in the associated messages without error pg.selectMessages([53, 56])