mirror of https://github.com/ae-utbm/sith3.git
146 lines
6.3 KiB
Python
146 lines
6.3 KiB
Python
# -*- coding:utf-8 -*
|
|
#
|
|
# Copyright 2023
|
|
# - Skia <skia@hya.sk>
|
|
#
|
|
# Ce fichier fait partie du site de l'Association des Étudiants de l'UTBM,
|
|
# http://ae.utbm.fr.
|
|
#
|
|
# This program is free software; you can redistribute it and/or modify it under
|
|
# the terms of the GNU General Public License a published by the Free Software
|
|
# Foundation; either version 3 of the License, or (at your option) any later
|
|
# version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful, but WITHOUT
|
|
# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
|
|
# FOR A PARTICULAR PURPOSE. See the GNU General Public License for more
|
|
# details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License along with
|
|
# this program; if not, write to the Free Sofware Foundation, Inc., 59 Temple
|
|
# Place - Suite 330, Boston, MA 02111-1307, USA.
|
|
#
|
|
#
|
|
|
|
from django.test import TestCase
|
|
from django.core.management import call_command
|
|
|
|
from core.models import User
|
|
from galaxy.models import Galaxy
|
|
|
|
|
|
class GalaxyTest(TestCase):
|
|
def setUp(self):
|
|
call_command("populate")
|
|
self.root = User.objects.get(username="root")
|
|
self.skia = User.objects.get(username="skia")
|
|
self.sli = User.objects.get(username="sli")
|
|
self.krophil = User.objects.get(username="krophil")
|
|
self.richard = User.objects.get(username="rbatsbak")
|
|
self.subscriber = User.objects.get(username="subscriber")
|
|
self.public = User.objects.get(username="public")
|
|
self.com = User.objects.get(username="comunity")
|
|
|
|
def test_user_self_score(self):
|
|
with self.assertNumQueries(8):
|
|
self.assertEqual(Galaxy.compute_user_score(self.root), 9)
|
|
self.assertEqual(Galaxy.compute_user_score(self.skia), 10)
|
|
self.assertEqual(Galaxy.compute_user_score(self.sli), 8)
|
|
self.assertEqual(Galaxy.compute_user_score(self.krophil), 2)
|
|
self.assertEqual(Galaxy.compute_user_score(self.richard), 10)
|
|
self.assertEqual(Galaxy.compute_user_score(self.subscriber), 8)
|
|
self.assertEqual(Galaxy.compute_user_score(self.public), 8)
|
|
self.assertEqual(Galaxy.compute_user_score(self.com), 1)
|
|
|
|
def test_users_score(self):
|
|
expected_scores = {
|
|
"krophil": {
|
|
"comunity": {"clubs": 0, "family": 0, "pictures": 0, "score": 0},
|
|
"public": {"clubs": 0, "family": 0, "pictures": 0, "score": 0},
|
|
"rbatsbak": {"clubs": 100, "family": 0, "pictures": 0, "score": 100},
|
|
"subscriber": {"clubs": 0, "family": 0, "pictures": 0, "score": 0},
|
|
},
|
|
"public": {
|
|
"comunity": {"clubs": 0, "family": 0, "pictures": 0, "score": 0}
|
|
},
|
|
"rbatsbak": {
|
|
"comunity": {"clubs": 0, "family": 0, "pictures": 0, "score": 0},
|
|
"public": {"clubs": 0, "family": 366, "pictures": 0, "score": 366},
|
|
"subscriber": {"clubs": 0, "family": 366, "pictures": 0, "score": 366},
|
|
},
|
|
"root": {
|
|
"comunity": {"clubs": 0, "family": 0, "pictures": 0, "score": 0},
|
|
"krophil": {"clubs": 0, "family": 0, "pictures": 0, "score": 0},
|
|
"public": {"clubs": 0, "family": 0, "pictures": 0, "score": 0},
|
|
"rbatsbak": {"clubs": 0, "family": 0, "pictures": 0, "score": 0},
|
|
"skia": {"clubs": 0, "family": 732, "pictures": 0, "score": 732},
|
|
"sli": {"clubs": 0, "family": 0, "pictures": 0, "score": 0},
|
|
"subscriber": {"clubs": 0, "family": 0, "pictures": 0, "score": 0},
|
|
},
|
|
"skia": {
|
|
"comunity": {"clubs": 0, "family": 0, "pictures": 0, "score": 0},
|
|
"krophil": {"clubs": 114, "family": 0, "pictures": 2, "score": 116},
|
|
"public": {"clubs": 0, "family": 0, "pictures": 0, "score": 0},
|
|
"rbatsbak": {"clubs": 100, "family": 0, "pictures": 0, "score": 100},
|
|
"sli": {"clubs": 0, "family": 366, "pictures": 4, "score": 370},
|
|
"subscriber": {"clubs": 0, "family": 0, "pictures": 0, "score": 0},
|
|
},
|
|
"sli": {
|
|
"comunity": {"clubs": 0, "family": 0, "pictures": 0, "score": 0},
|
|
"krophil": {"clubs": 17, "family": 0, "pictures": 2, "score": 19},
|
|
"public": {"clubs": 0, "family": 0, "pictures": 0, "score": 0},
|
|
"rbatsbak": {"clubs": 0, "family": 0, "pictures": 0, "score": 0},
|
|
"subscriber": {"clubs": 0, "family": 0, "pictures": 0, "score": 0},
|
|
},
|
|
"subscriber": {
|
|
"comunity": {"clubs": 0, "family": 0, "pictures": 0, "score": 0},
|
|
"public": {"clubs": 0, "family": 0, "pictures": 0, "score": 0},
|
|
},
|
|
}
|
|
computed_scores = {}
|
|
users = [
|
|
self.root,
|
|
self.skia,
|
|
self.sli,
|
|
self.krophil,
|
|
self.richard,
|
|
self.subscriber,
|
|
self.public,
|
|
self.com,
|
|
]
|
|
|
|
with self.assertNumQueries(100):
|
|
while len(users) > 0:
|
|
user1 = users.pop(0)
|
|
for user2 in users:
|
|
score, family, pictures, clubs = Galaxy.compute_users_score(
|
|
user1, user2
|
|
)
|
|
u1 = computed_scores.get(user1.username, {})
|
|
u1[user2.username] = {
|
|
"score": score,
|
|
"family": family,
|
|
"pictures": pictures,
|
|
"clubs": clubs,
|
|
}
|
|
computed_scores[user1.username] = u1
|
|
|
|
self.maxDiff = None # Yes, we want to see the diff if any
|
|
self.assertDictEqual(expected_scores, computed_scores)
|
|
|
|
def test_page_is_citizen(self):
|
|
Galaxy.rule()
|
|
self.client.login(username="root", password="plop")
|
|
response = self.client.get("/galaxy/1/")
|
|
self.assertContains(
|
|
response,
|
|
'<a onclick="focus_node(get_node_from_id(8))">Locate</a>',
|
|
status_code=200,
|
|
)
|
|
|
|
def test_page_not_citizen(self):
|
|
Galaxy.rule()
|
|
self.client.login(username="root", password="plop")
|
|
response = self.client.get("/galaxy/2/")
|
|
self.assertEquals(response.status_code, 404)
|