class CrawlerTestCase(unittest.TestCase): def setUp(self): self.crawler = Crawler_quora() def tearDown(self): self.crawler = None def test_add_follower(self): self.assertFalse(self.crawler.add_follower(), 'add follower with empty information') self.assertFalse(self.crawler.add_follower(id='51248642'), 'add follower with wrong information') self.assertFalse(self.crawler.add_follower(user_name='wahahahah'), 'no such user') self.assertIsNotNone( self.crawler.add_follower(user_name='quan-zhang-27'), 'add follower with correct screen name') self.assertEqual(self.crawler.add_follower(user_name='quan-zhang-27'), 'quan-zhang-27', 'check return id if is equal to given correct id') def test_get_status(self): self.assertEqual(self.crawler.get_status(), [], 'get status with empty information') self.assertEqual(self.crawler.get_status(id='51248642'), [], 'get status with incorret information') self.assertEqual(self.crawler.get_status(user_name='wahahahah'), [], 'get status with wrong user name') self.assertNotEqual( len(self.crawler.get_status(user_name='quan-zhang-27')), 0, 'get status with correct screen name') self.assertNotEqual( len(self.crawler.get_status(user_name='Dan-holliday')), 0, 'get status with correct user name')
class CrawlerTestCase(unittest.TestCase): def setUp(self): self.crawler = Crawler_quora() def tearDown(self): self.crawler = None def test_add_follower(self): self.assertFalse(self.crawler.add_follower(), 'add follower with empty information') self.assertFalse(self.crawler.add_follower(id = '51248642'), 'add follower with wrong information') self.assertFalse(self.crawler.add_follower(user_name='wahahahah'), 'no such user') self.assertIsNotNone(self.crawler.add_follower(user_name='quan-zhang-27'), 'add follower with correct screen name') self.assertEqual(self.crawler.add_follower(user_name='quan-zhang-27'), 'quan-zhang-27', 'check return id if is equal to given correct id') def test_get_status(self): self.assertEqual(self.crawler.get_status(), [], 'get status with empty information') self.assertEqual(self.crawler.get_status(id = '51248642'), [], 'get status with incorret information') self.assertEqual(self.crawler.get_status(user_name='wahahahah'), [], 'get status with wrong user name') self.assertNotEqual(len(self.crawler.get_status(user_name='quan-zhang-27')), 0, 'get status with correct screen name') self.assertNotEqual(len(self.crawler.get_status(user_name='Dan-holliday')), 0, 'get status with correct user name')
def verify_account(**verify_kwargs): if verify_kwargs['flr_name'] is not None: if QrAccount.objects.filter(follower__name=verify_kwargs['flr_name']).exists(): return False tw_crawler = Crawler_quora() crawler_kwargs = {'user_name': verify_kwargs.get('user_name', None)} user_name = tw_crawler.add_follower(**crawler_kwargs) if not user_name: return '404' return {'user_name': user_name}