def test_cancel_collect(self):
email = '*****@*****.**'
photo_url = 'test_url'
tag = u'tag1、tag2'
dbControl.save_photo_and_tag(photo_url=photo_url, description='no',
tag=tag, person_id_list=[], permission='public', owner=email)
photo_id= Photo.objects.get(photo_url=photo_url).id
dbControl.add_collect(email, photo_id)
collected_times = Photo.objects.get(id=photo_id).collected_times
self.assertEqual(collected_times, 1)
dbControl.cancel_collect(email=email, photo_id=photo_id)
collected_times = Photo.objects.get(id=photo_id).collected_times
self.assertEqual(collected_times, 0)
def test_add_collect(self):
import pdb
pdb.set_trace()
email = '*****@*****.**'
photo_url = 'test_url'
tag = u'tag1、tag2'
dbControl.save_photo_and_tag(photo_url=photo_url, description='no',
tag=tag, person_id_list=[], permission='public', owner=email)
photo_id= Photo.objects.get(photo_url=photo_url).id
dbControl.add_collect(email, photo_id)
collected_times = Photo.objects.get(id=photo_id).collected_times
self.assertTrue(Collect.objects.get(
email=email, photo_id=photo_id))
self.assertEqual(collected_times, 1)
self.assertTrue(Interest.objects.get(email=email, interested_tag='tag2'))
def collect_delete(request):
""" Cancel collect. """
return_dict = {}
if request.user.is_authenticated():
email = common.get_email(request)
if request.method == 'GET':
photo_id = int(request.GET['p_id'])
return_dict['SUC'] = dbControl.add_collect(email, photo_id)
return_dict['collected_times'] = dbControl.get_collected_times(photo_id)
return_dict['info'] = ''
else:
return_dict['SUC'] = False
return_dict['info'] = u'未登录'
return JsonResponse(return_dict)
