def friends(request):
if request.user is None:
return HttpResponse(status=401)
if request.method == 'GET':
response = get_friends(user=request.user)
return HttpResponse(dumps(response))
elif request.method == 'POST':
friend_name = _get_post_param(param='username', request=request)
add_friend(user=request.user, friend_name=friend_name)
return HttpResponse(status=200)
def test_add_friend_calls_correct_methods(self, mock_friend_class, mock_user_class):
mock_friend_user = MagicMock()
mock_relationship = MagicMock()
mock_user = MagicMock
mock_user_class.objects.all().filter.return_value = [mock_friend_user]
mock_friend_class.objects.add_friend.return_value = mock_relationship
add_friend(mock_user, 'joe')
mock_user_class.objects.all().filter.assert_called_with(username='******')
mock_friend_class.objects.add_friend.assert_called_with(mock_user, mock_friend_user)
mock_relationship.accept.assert_called_with()
def test_add_friend_throws_exception_for_none(self):
self.assertRaises(ValueError, add_friend())