def test_get_friends_returns(self, mock_friend_class): mock_friend_class.objects.friends.return_value = [MagicMock(username='******'), MagicMock(username='******'), MagicMock(username='******')] mock_user = MagicMock() actual = get_friends(user=mock_user) expected = ['john', 'james', 'ben'] self.assertListEqual(actual, expected) mock_friend_class.objects.friends.assert_called_with(mock_user)
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_get_friends_returns_empty_list_for_no_friends(self, mock_friend_class): mock_friend_class.objects.friends.return_value = [] actual = get_friends(user=MagicMock()) expected = [] self.assertListEqual(actual, expected)