class CrawlerTestCase(unittest.TestCase):
def setUp(self):
self.crawler = Crawler_twitter()
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.assertIsNotNone(self.crawler.add_follower(screen_name = 'matthewperryfan'),
'add follower with correct screen name')
self.assertEqual(self.crawler.add_follower(id = '30653573').get('id'), '30653573',
'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.assertNotEqual(len(self.crawler.get_status(screen_name = 'matthewperryfan')), 0,
'get status with correct screen name')
self.assertNotEqual(len(self.crawler.get_status(id = '30653573')), 0,
'get status with correct id')
class CrawlerTestCase(unittest.TestCase):
def setUp(self):
self.crawler = Crawler_twitter()
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.assertIsNotNone(
self.crawler.add_follower(screen_name='matthewperryfan'),
'add follower with correct screen name')
self.assertEqual(
self.crawler.add_follower(id='30653573').get('id'), '30653573',
'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.assertNotEqual(
len(self.crawler.get_status(screen_name='matthewperryfan')), 0,
'get status with correct screen name')
self.assertNotEqual(len(self.crawler.get_status(id='30653573')), 0,
'get status with correct id')
def verify_account(**verify_kwargs):
if verify_kwargs['flr_name'] is not None:
if TwAccount.objects.filter(follower__name=verify_kwargs['flr_name']).exists():
return False
tw_crawler = Crawler_twitter()
if verify_kwargs['act_id'] == u'None':
verify_kwargs['act_id'] = None
if verify_kwargs['screen_name'] == u'None':
verify_kwargs['screen_name'] = None
crawler_kwargs = {'id': verify_kwargs['act_id'],
'screen_name': verify_kwargs['screen_name']}
is_exist_follower = tw_crawler.add_follower(**crawler_kwargs)
if not is_exist_follower:
return '404'
# rename for consistency
is_exist_follower['act_id'] = is_exist_follower.pop('id')
return is_exist_follower
def verify_account(**verify_kwargs):
if verify_kwargs['flr_name'] is not None:
if TwAccount.objects.filter(
follower__name=verify_kwargs['flr_name']).exists():
return False
tw_crawler = Crawler_twitter()
if verify_kwargs['act_id'] == u'None':
verify_kwargs['act_id'] = None
if verify_kwargs['screen_name'] == u'None':
verify_kwargs['screen_name'] = None
crawler_kwargs = {
'id': verify_kwargs['act_id'],
'screen_name': verify_kwargs['screen_name']
}
is_exist_follower = tw_crawler.add_follower(**crawler_kwargs)
if not is_exist_follower:
return '404'
# rename for consistency
is_exist_follower['act_id'] = is_exist_follower.pop('id')
return is_exist_follower
def add_follower(request):
if request.method == 'POST':
tw_crawler = Crawler_twitter()
follower_id = request.POST['follower_id']
screen_name = request.POST['screen_name']
kwargs = {'screen_name': screen_name, 'id': follower_id}
is_exist_follower = tw_crawler.add_follower(**kwargs)
if not is_exist_follower:
return HttpResponse(False)
else:
if TwFollower.objects.filter(user_id=request.user.id).filter(
follower_id=is_exist_follower['id']).exists():
return HttpResponse(False)
new_follower = TwFollower()
new_follower.follower_id = is_exist_follower['id']
new_follower.screen_name = is_exist_follower['screen_name']
new_follower.user = request.user
new_follower.save()
return HttpResponse(True)
else:
raise Http404
def add_follower(request):
if request.method == 'POST':
tw_crawler = Crawler_twitter()
follower_id = request.POST['follower_id']
screen_name = request.POST['screen_name']
kwargs = {'screen_name': screen_name,
'id': follower_id}
is_exist_follower = tw_crawler.add_follower(**kwargs)
if not is_exist_follower:
return HttpResponse(False)
else:
if TwFollower.objects.filter(user_id=request.user.id).filter(follower_id=is_exist_follower['id']).exists():
return HttpResponse(False)
new_follower = TwFollower()
new_follower.follower_id = is_exist_follower['id']
new_follower.screen_name = is_exist_follower['screen_name']
new_follower.user = request.user
new_follower.save()
return HttpResponse(True)
else:
raise Http404
