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)