def invite(request, user_id):
# Cannot invite ourself
if request.user.pk == int(user_id):
raise Http404
# Find the friend to request
new_friend = get_object_or_404(Profile, user__pk=user_id)
# Create the friend request
friend_request = FriendRequest(user=request.user, to=new_friend.user)
friend_request.save()
# Alert the requester and the requested
send_localized_mail(
new_friend.user, _('[R.A.S.] Friend request'),
'RandoAmisSecours/friends/request_email.html', {
'from':
friend_request.user.get_full_name(),
'to':
friend_request.to.get_full_name(),
'accept':
request.build_absolute_uri(
reverse('friends.accept', args=[friend_request.pk])),
'refuse':
request.build_absolute_uri(
reverse('friends.refuse', args=[friend_request.pk]))
})
messages.success(
request,
_("Friend request sent to «%(name)s»") %
({
'name': new_friend.user.get_full_name()
}))
return HttpResponseRedirect(reverse('friends.search'))
def test_accept_simple(self):
# Simple accept session
FR = FriendRequest(user=self.user2, to=self.user1)
FR.save()
self.assertEqual(FriendRequest.objects.all().count(), 1)
self.assertEqual(self.user1.profile.friends.all().count(), 0)
self.assertEqual(self.user2.profile.friends.all().count(), 0)
self.assertEqual(self.user3.profile.friends.all().count(), 0)
response = self.client.get(reverse('friends.accept', args=[FR.pk]))
self.assertRedirects(response, reverse('accounts.profile'))
self.assertEqual(FriendRequest.objects.all().count(), 0)
self.assertEqual(self.user1.profile.friends.all().count(), 1)
self.assertEqual(self.user1.profile.friends.all()[0].user.pk, self.user2.pk)
self.assertEqual(self.user2.profile.friends.all().count(), 1)
self.assertEqual(self.user2.profile.friends.all()[0].user.pk, self.user1.pk)
self.assertEqual(self.user3.profile.friends.all().count(), 0)
def test_accept_simple(self):
# Simple accept session
FR = FriendRequest(user=self.user2, to=self.user1)
FR.save()
self.assertEqual(FriendRequest.objects.all().count(), 1)
self.assertEqual(self.user1.profile.friends.all().count(), 0)
self.assertEqual(self.user2.profile.friends.all().count(), 0)
self.assertEqual(self.user3.profile.friends.all().count(), 0)
response = self.client.get(reverse('friends.accept', args=[FR.pk]))
self.assertRedirects(response, reverse('accounts.profile'))
self.assertEqual(FriendRequest.objects.all().count(), 0)
self.assertEqual(self.user1.profile.friends.all().count(), 1)
self.assertEqual(self.user1.profile.friends.all()[0].user.pk,
self.user2.pk)
self.assertEqual(self.user2.profile.friends.all().count(), 1)
self.assertEqual(self.user2.profile.friends.all()[0].user.pk,
self.user1.pk)
self.assertEqual(self.user3.profile.friends.all().count(), 0)
def test_accept_advance(self):
# Two at a time
FR1 = FriendRequest(user=self.user3, to=self.user1)
FR1.save()
FR2 = FriendRequest(user=self.user2, to=self.user1)
FR2.save()
self.assertEqual(FriendRequest.objects.all().count(), 2)
self.assertEqual(self.user1.profile.friends.all().count(), 0)
self.assertEqual(self.user2.profile.friends.all().count(), 0)
self.assertEqual(self.user3.profile.friends.all().count(), 0)
response = self.client.get(reverse('friends.accept', args=[FR2.pk]))
self.assertRedirects(response, reverse('accounts.profile'))
self.assertEqual(FriendRequest.objects.all().count(), 1)
self.assertEqual(self.user1.profile.friends.all().count(), 1)
self.assertEqual(self.user1.profile.friends.all()[0].user.pk, self.user2.pk)
self.assertEqual(self.user2.profile.friends.all().count(), 1)
self.assertEqual(self.user2.profile.friends.all()[0].user.pk, self.user1.pk)
self.assertEqual(self.user3.profile.friends.all().count(), 0)
response = self.client.get(reverse('friends.accept', args=[FR2.pk]))
self.assertEqual(response.status_code, 404)
response = self.client.get(reverse('friends.accept', args=[FR1.pk]))
self.assertRedirects(response, reverse('accounts.profile'))
self.assertEqual(FriendRequest.objects.all().count(), 0)
self.assertEqual(self.user1.profile.friends.all().count(), 2)
self.assertEqual(self.user1.profile.friends.all()[0].user.pk, self.user2.pk)
self.assertEqual(self.user1.profile.friends.all()[1].user.pk, self.user3.pk)
self.assertEqual(self.user2.profile.friends.all().count(), 1)
self.assertEqual(self.user2.profile.friends.all()[0].user.pk, self.user1.pk)
self.assertEqual(self.user3.profile.friends.all().count(), 1)
self.assertEqual(self.user3.profile.friends.all()[0].user.pk, self.user1.pk)
response = self.client.get(reverse('friends.accept', args=[FR1.pk]))
self.assertEqual(response.status_code, 404)
# Only the 'to' can accept it
FR3 = FriendRequest(user=self.user3, to=self.user2)
FR3.save()
response = self.client.get(reverse('friends.accept', args=[FR3.pk]))
self.assertEqual(response.status_code, 404)
def invite(request, user_id):
# Cannot invite ourself
if request.user.pk == int(user_id):
raise Http404
# Find the friend to request
new_friend = get_object_or_404(Profile, user__pk=user_id)
# Create the friend request
friend_request = FriendRequest(user=request.user, to=new_friend.user)
friend_request.save()
# Alert the requester and the requested
send_localized_mail(new_friend.user, _('[R.A.S.] Friend request'),
'RandoAmisSecours/friends/request_email.html',
{'from': friend_request.user.get_full_name(),
'to': friend_request.to.get_full_name(),
'accept': request.build_absolute_uri(reverse('friends.accept',
args=[friend_request.pk])),
'refuse': request.build_absolute_uri(reverse('friends.refuse',
args=[friend_request.pk]))})
messages.success(request, _("Friend request sent to «%(name)s»") % ({'name': new_friend.user.get_full_name()}))
return HttpResponseRedirect(reverse('friends.search'))
def test_friend_request(self):
self.user2 = User.objects.create_user('tester',
'*****@*****.**',
'ertyfjnbfvfceqsryuj')
self.user3 = User.objects.create_user('Sophocle',
'*****@*****.**',
'gzgvaryurvyyjchrvyhubtr')
FR = FriendRequest(user=self.user1, to=self.user2)
FR.save()
FR2 = FriendRequest(user=self.user3, to=self.user1)
FR2.save()
FR3 = FriendRequest(user=self.user3, to=self.user2)
FR3.save()
response = self.client.get(reverse('accounts.profile'))
self.assertEqual(response.status_code, 200)
ctx = response.context
self.assertEqual(len(ctx['friend_requests']), 1)
self.assertEqual(ctx['friend_requests'][0].user, self.user3)
self.assertEqual(ctx['friend_requests'][0].to, self.user1)
self.assertEqual(len(ctx['friend_requests_sent']), 1)
self.assertEqual(ctx['friend_requests_sent'][0].user, self.user1)
self.assertEqual(ctx['friend_requests_sent'][0].to, self.user2)
def test_cancel(self):
FR1 = FriendRequest(user=self.user1, to=self.user3)
FR1.save()
FR2 = FriendRequest(user=self.user2, to=self.user1)
FR2.save()
self.assertEqual(FriendRequest.objects.all().count(), 2)
self.assertEqual(self.user1.profile.friends.all().count(), 0)
self.assertEqual(self.user2.profile.friends.all().count(), 0)
self.assertEqual(self.user3.profile.friends.all().count(), 0)
response = self.client.get(reverse('friends.cancel', args=[FR1.pk]))
self.assertRedirects(response, reverse('accounts.profile'))
self.assertEqual(FriendRequest.objects.all().count(), 1)
self.assertEqual(FriendRequest.objects.all()[0].pk, FR2.pk)
self.assertEqual(self.user1.profile.friends.all().count(), 0)
self.assertEqual(self.user2.profile.friends.all().count(), 0)
self.assertEqual(self.user3.profile.friends.all().count(), 0)
response = self.client.get(reverse('friends.cancel', args=[FR1.pk]))
self.assertEqual(response.status_code, 404)
def test_friends(self):
self.helper_test_login(reverse('friends.search'))
self.helper_test_login(reverse('friends.invite', args=[self.user2.pk]), redirect=reverse('friends.search'))
FriendRequest.objects.all().delete()
FR = FriendRequest(user=self.user2, to=self.user)
FR.save()
self.helper_test_login(reverse('friends.accept', args=[FR.pk]), redirect=reverse('accounts.profile'))
self.helper_test_login(reverse('friends.delete', args=[self.user2.pk]), redirect=reverse('accounts.profile'))
FR = FriendRequest(user=self.user2, to=self.user)
FR.save()
self.helper_test_login(reverse('friends.refuse', args=[FR.pk]), redirect=reverse('accounts.profile'))
self.helper_test_login(reverse('friends.invite', args=[self.user2.pk]), redirect=reverse('friends.search'))
self.helper_test_login(reverse('friends.cancel', args=[FriendRequest.objects.all()[0].pk]), redirect=reverse('accounts.profile'))
def test_friends(self):
self.helper_test_login(reverse('friends.search'))
self.helper_test_login(reverse('friends.invite', args=[self.user2.pk]),
redirect=reverse('friends.search'))
FriendRequest.objects.all().delete()
FR = FriendRequest(user=self.user2, to=self.user)
FR.save()
self.helper_test_login(reverse('friends.accept', args=[FR.pk]),
redirect=reverse('accounts.profile'))
self.helper_test_login(reverse('friends.delete', args=[self.user2.pk]),
redirect=reverse('accounts.profile'))
FR = FriendRequest(user=self.user2, to=self.user)
FR.save()
self.helper_test_login(reverse('friends.refuse', args=[FR.pk]),
redirect=reverse('accounts.profile'))
self.helper_test_login(reverse('friends.invite', args=[self.user2.pk]),
redirect=reverse('friends.search'))
self.helper_test_login(reverse(
'friends.cancel', args=[FriendRequest.objects.all()[0].pk]),
redirect=reverse('accounts.profile'))
def test_cancel(self):
FR1 = FriendRequest(user=self.user1, to=self.user3)
FR1.save()
FR2 = FriendRequest(user=self.user2, to=self.user1)
FR2.save()
self.assertEqual(FriendRequest.objects.all().count(), 2)
self.assertEqual(self.user1.profile.friends.all().count(), 0)
self.assertEqual(self.user2.profile.friends.all().count(), 0)
self.assertEqual(self.user3.profile.friends.all().count(), 0)
response = self.client.get(reverse('friends.cancel', args=[FR1.pk]))
self.assertRedirects(response, reverse('accounts.profile'))
self.assertEqual(FriendRequest.objects.all().count(), 1)
self.assertEqual(FriendRequest.objects.all()[0].pk, FR2.pk)
self.assertEqual(self.user1.profile.friends.all().count(), 0)
self.assertEqual(self.user2.profile.friends.all().count(), 0)
self.assertEqual(self.user3.profile.friends.all().count(), 0)
response = self.client.get(reverse('friends.cancel', args=[FR1.pk]))
self.assertEqual(response.status_code, 404)
def test_friend_request(self):
self.user2 = User.objects.create_user('tester', '*****@*****.**',
'ertyfjnbfvfceqsryuj')
self.user3 = User.objects.create_user('Sophocle',
'*****@*****.**',
'gzgvaryurvyyjchrvyhubtr')
FR = FriendRequest(user=self.user1, to=self.user2)
FR.save()
FR2 = FriendRequest(user=self.user3, to=self.user1)
FR2.save()
FR3 = FriendRequest(user=self.user3, to=self.user2)
FR3.save()
response = self.client.get(reverse('accounts.profile'))
self.assertEqual(response.status_code, 200)
ctx = response.context
self.assertEqual(len(ctx['friend_requests']), 1)
self.assertEqual(ctx['friend_requests'][0].user, self.user3)
self.assertEqual(ctx['friend_requests'][0].to, self.user1)
self.assertEqual(len(ctx['friend_requests_sent']), 1)
self.assertEqual(ctx['friend_requests_sent'][0].user, self.user1)
self.assertEqual(ctx['friend_requests_sent'][0].to, self.user2)
def test_accept_advance(self):
# Two at a time
FR1 = FriendRequest(user=self.user3, to=self.user1)
FR1.save()
FR2 = FriendRequest(user=self.user2, to=self.user1)
FR2.save()
self.assertEqual(FriendRequest.objects.all().count(), 2)
self.assertEqual(self.user1.profile.friends.all().count(), 0)
self.assertEqual(self.user2.profile.friends.all().count(), 0)
self.assertEqual(self.user3.profile.friends.all().count(), 0)
response = self.client.get(reverse('friends.accept', args=[FR2.pk]))
self.assertRedirects(response, reverse('accounts.profile'))
self.assertEqual(FriendRequest.objects.all().count(), 1)
self.assertEqual(self.user1.profile.friends.all().count(), 1)
self.assertEqual(self.user1.profile.friends.all()[0].user.pk,
self.user2.pk)
self.assertEqual(self.user2.profile.friends.all().count(), 1)
self.assertEqual(self.user2.profile.friends.all()[0].user.pk,
self.user1.pk)
self.assertEqual(self.user3.profile.friends.all().count(), 0)
response = self.client.get(reverse('friends.accept', args=[FR2.pk]))
self.assertEqual(response.status_code, 404)
response = self.client.get(reverse('friends.accept', args=[FR1.pk]))
self.assertRedirects(response, reverse('accounts.profile'))
self.assertEqual(FriendRequest.objects.all().count(), 0)
self.assertEqual(self.user1.profile.friends.all().count(), 2)
self.assertEqual(self.user1.profile.friends.all()[0].user.pk,
self.user2.pk)
self.assertEqual(self.user1.profile.friends.all()[1].user.pk,
self.user3.pk)
self.assertEqual(self.user2.profile.friends.all().count(), 1)
self.assertEqual(self.user2.profile.friends.all()[0].user.pk,
self.user1.pk)
self.assertEqual(self.user3.profile.friends.all().count(), 1)
self.assertEqual(self.user3.profile.friends.all()[0].user.pk,
self.user1.pk)
response = self.client.get(reverse('friends.accept', args=[FR1.pk]))
self.assertEqual(response.status_code, 404)
# Only the 'to' can accept it
FR3 = FriendRequest(user=self.user3, to=self.user2)
FR3.save()
response = self.client.get(reverse('friends.accept', args=[FR3.pk]))
self.assertEqual(response.status_code, 404)
