def test_invalid_type(self):
rf = RequestFactory()
user = User(id="uid", name="name", email="*****@*****.**")
user.save()
expected = HttpResponse("Wrong leaderboard type", status=400)
request = rf.post("get_leaderboard",
data=json.dumps({"object": {
"type": None
}}),
content_type="application/json")
response = get_leaderboard.handler(request)
self.assertEqual(str(response), str(expected))
request = rf.post("get_leaderboard",
data=json.dumps({"object": {
"type": ""
}}),
content_type="application/json")
response = get_leaderboard.handler(request)
self.assertEqual(str(response), str(expected))
request = rf.post("get_leaderboard",
data=json.dumps({"object": {
"type": "INVALID"
}}),
content_type="application/json")
response = get_leaderboard.handler(request)
self.assertEqual(str(response), str(expected))
def test_missing_object(self):
rf = RequestFactory()
expected = HttpResponse("Missing object", status=404)
request = rf.post("get_leaderboard",
data="",
content_type="application/json")
response = get_leaderboard.handler(request)
self.assertEqual(str(response), str(expected))
request = rf.post("get_leaderboard",
data=json.dumps({}),
content_type="application/json")
response = get_leaderboard.handler(request)
self.assertEqual(str(response), str(expected))
def test_friends_leaderboard(self):
rf = RequestFactory()
expected = JsonResponse(
{
"user_position":
2,
"leaderboard": [{
"name": "user2",
"score": 12
}, {
"name": "user1",
"score": 8
}, {
"name": "user3",
"score": 0
}]
},
status=200)
user1 = User(id="uid",
name="user1",
email="*****@*****.**",
n_reports=4,
weight=2.0)
user1.save()
user2 = User(id="uid2",
name="user2",
email="*****@*****.**",
n_reports=4,
weight=2.3)
user2.save()
user3 = User(id="uid3",
name="user3",
email="*****@*****.**",
n_reports=0,
weight=1.0)
user3.save()
friendship1 = Friendship(user=user1, friend=user2)
friendship1.save()
friendship2 = Friendship(user=user1, friend=user3)
friendship2.save()
request = rf.post("get_leaderboard",
data=json.dumps({"object": {
"type": "FRIENDS"
}}),
content_type="application/json")
response = get_leaderboard.handler(request)
self.assertEqual(str(response), str(expected))
def test_invalid_token(self):
utils.check_google_token = Mock(return_value=(False, "Error message"))
expected = HttpResponse("Error message", status=403)
response = get_leaderboard.handler(HttpRequest())
self.assertEqual(str(response), str(expected))
def test_invalid_user(self):
expected = HttpResponse("Missing user", status=404)
utils.check_google_token = Mock(return_value=(True, "uid"))
response = get_leaderboard.handler(HttpRequest())
self.assertEqual(str(response), str(expected))