def add_friend(request): if request.method == 'POST': username = request.POST['username'] try: friend = User.objects.get(username=username) except User.DoesNotExist: return redirect("/friends/?error=no_friend") Friendship.create_friendship(request.user, friend) return redirect("/friends/")
def setUp(self): user1 = User.objects.create_user('johndoe', '*****@*****.**', 'johndoe') self.client.login(username='******', password='******') user2 = User.objects.create_user('janedoe', '*****@*****.**', 'janedoe') Friendship.create_friendship(user1, user2)