mirror of
https://github.com/ae-utbm/sith.git
synced 2024-11-22 14:13:21 +00:00
437 lines
15 KiB
Python
437 lines
15 KiB
Python
#
|
|
# Copyright 2023 © AE UTBM
|
|
# ae@utbm.fr / ae.info@utbm.fr
|
|
#
|
|
# This file is part of the website of the UTBM Student Association (AE UTBM),
|
|
# https://ae.utbm.fr.
|
|
#
|
|
# You can find the source code of the website at https://github.com/ae-utbm/sith
|
|
#
|
|
# LICENSED UNDER THE GNU GENERAL PUBLIC LICENSE VERSION 3 (GPLv3)
|
|
# SEE : https://raw.githubusercontent.com/ae-utbm/sith/master/LICENSE
|
|
# OR WITHIN THE LOCAL FILE "LICENSE"
|
|
#
|
|
#
|
|
import re
|
|
from datetime import timedelta
|
|
|
|
import pytest
|
|
from django.conf import settings
|
|
from django.core.cache import cache
|
|
from django.test import TestCase
|
|
from django.urls import reverse
|
|
from django.utils import timezone
|
|
from django.utils.timezone import now
|
|
from freezegun import freeze_time
|
|
from model_bakery import baker
|
|
|
|
from club.models import Club, Membership
|
|
from core.baker_recipes import subscriber_user
|
|
from core.models import User
|
|
from counter.models import (
|
|
Counter,
|
|
Customer,
|
|
Permanency,
|
|
Product,
|
|
Selling,
|
|
)
|
|
|
|
|
|
class TestCounter(TestCase):
|
|
@classmethod
|
|
def setUpTestData(cls):
|
|
cls.skia = User.objects.filter(username="skia").first()
|
|
cls.sli = User.objects.filter(username="sli").first()
|
|
cls.krophil = User.objects.filter(username="krophil").first()
|
|
cls.richard = User.objects.filter(username="rbatsbak").first()
|
|
cls.mde = Counter.objects.filter(name="MDE").first()
|
|
cls.foyer = Counter.objects.get(id=2)
|
|
|
|
def test_full_click(self):
|
|
self.client.post(
|
|
reverse("counter:login", kwargs={"counter_id": self.mde.id}),
|
|
{"username": self.skia.username, "password": "plop"},
|
|
)
|
|
response = self.client.get(
|
|
reverse("counter:details", kwargs={"counter_id": self.mde.id})
|
|
)
|
|
|
|
assert 'class="link-button">S' Kia</button>' in str(response.content)
|
|
|
|
counter_token = re.search(
|
|
r'name="counter_token" value="([^"]*)"', str(response.content)
|
|
).group(1)
|
|
|
|
response = self.client.post(
|
|
reverse("counter:details", kwargs={"counter_id": self.mde.id}),
|
|
{"code": self.richard.customer.account_id, "counter_token": counter_token},
|
|
)
|
|
counter_url = response.get("location")
|
|
refill_url = reverse(
|
|
"counter:refilling_create_fragment",
|
|
kwargs={"counter_id": self.mde.id, "customer_id": self.richard.customer.pk},
|
|
)
|
|
response = self.client.get(response.get("location"))
|
|
assert ">Richard Batsbak</" in str(response.content)
|
|
|
|
self.client.post(
|
|
refill_url,
|
|
{
|
|
"amount": "5",
|
|
"payment_method": "CASH",
|
|
"bank": "OTHER",
|
|
},
|
|
)
|
|
self.client.post(counter_url, "action=code&code=BARB", content_type="text/xml")
|
|
self.client.post(
|
|
counter_url, "action=add_product&product_id=4", content_type="text/xml"
|
|
)
|
|
self.client.post(
|
|
counter_url, "action=del_product&product_id=4", content_type="text/xml"
|
|
)
|
|
self.client.post(
|
|
counter_url, "action=code&code=2xdeco", content_type="text/xml"
|
|
)
|
|
self.client.post(
|
|
counter_url, "action=code&code=1xbarb", content_type="text/xml"
|
|
)
|
|
response = self.client.post(
|
|
counter_url, "action=code&code=fin", content_type="text/xml"
|
|
)
|
|
|
|
response_get = self.client.get(response.get("location"))
|
|
response_content = response_get.content.decode("utf-8")
|
|
assert "2 x Barbar" in str(response_content)
|
|
assert "2 x Déconsigne Eco-cup" in str(response_content)
|
|
assert "<p>Client : Richard Batsbak - Nouveau montant : 3.60" in str(
|
|
response_content
|
|
)
|
|
|
|
self.client.post(
|
|
reverse("counter:login", kwargs={"counter_id": self.mde.id}),
|
|
{"username": self.sli.username, "password": "plop"},
|
|
)
|
|
|
|
response = self.client.post(
|
|
refill_url,
|
|
{
|
|
"amount": "5",
|
|
"payment_method": "CASH",
|
|
"bank": "OTHER",
|
|
},
|
|
)
|
|
assert response.status_code == 302
|
|
|
|
self.client.post(
|
|
reverse("counter:login", kwargs={"counter_id": self.foyer.id}),
|
|
{"username": self.krophil.username, "password": "plop"},
|
|
)
|
|
|
|
response = self.client.get(
|
|
reverse("counter:details", kwargs={"counter_id": self.foyer.id})
|
|
)
|
|
|
|
counter_token = re.search(
|
|
r'name="counter_token" value="([^"]*)"', str(response.content)
|
|
).group(1)
|
|
|
|
response = self.client.post(
|
|
reverse("counter:details", kwargs={"counter_id": self.foyer.id}),
|
|
{"code": self.richard.customer.account_id, "counter_token": counter_token},
|
|
)
|
|
counter_url = response.get("location")
|
|
refill_url = reverse(
|
|
"counter:refilling_create_fragment",
|
|
kwargs={
|
|
"counter_id": self.foyer.id,
|
|
"customer_id": self.richard.customer.pk,
|
|
},
|
|
)
|
|
|
|
response = self.client.post(
|
|
refill_url,
|
|
{
|
|
"amount": "5",
|
|
"payment_method": "CASH",
|
|
"bank": "OTHER",
|
|
},
|
|
)
|
|
assert response.status_code == 302
|
|
|
|
def test_annotate_has_barman_queryset(self):
|
|
"""Test if the custom queryset method `annotate_has_barman` works as intended."""
|
|
self.sli.counters.set([self.foyer, self.mde])
|
|
counters = Counter.objects.annotate_has_barman(self.sli)
|
|
for counter in counters:
|
|
if counter.name in ("Foyer", "MDE"):
|
|
assert counter.has_annotated_barman
|
|
else:
|
|
assert not counter.has_annotated_barman
|
|
|
|
|
|
class TestCounterStats(TestCase):
|
|
@classmethod
|
|
def setUpTestData(cls):
|
|
cls.counter = Counter.objects.get(id=2)
|
|
cls.krophil = User.objects.get(username="krophil")
|
|
cls.skia = User.objects.get(username="skia")
|
|
cls.sli = User.objects.get(username="sli")
|
|
cls.root = User.objects.get(username="root")
|
|
cls.subscriber = User.objects.get(username="subscriber")
|
|
cls.old_subscriber = User.objects.get(username="old_subscriber")
|
|
cls.counter.sellers.add(cls.sli, cls.root, cls.skia, cls.krophil)
|
|
|
|
barbar = Product.objects.get(code="BARB")
|
|
|
|
# remove everything to make sure the fixtures bring no side effect
|
|
Permanency.objects.all().delete()
|
|
Selling.objects.all().delete()
|
|
|
|
now = timezone.now()
|
|
# total of sli : 5 hours
|
|
Permanency.objects.create(
|
|
user=cls.sli, start=now, end=now + timedelta(hours=1), counter=cls.counter
|
|
)
|
|
Permanency.objects.create(
|
|
user=cls.sli,
|
|
start=now + timedelta(hours=4),
|
|
end=now + timedelta(hours=6),
|
|
counter=cls.counter,
|
|
)
|
|
Permanency.objects.create(
|
|
user=cls.sli,
|
|
start=now + timedelta(hours=7),
|
|
end=now + timedelta(hours=9),
|
|
counter=cls.counter,
|
|
)
|
|
|
|
# total of skia : 16 days, 2 hours, 35 minutes and 54 seconds
|
|
Permanency.objects.create(
|
|
user=cls.skia, start=now, end=now + timedelta(hours=1), counter=cls.counter
|
|
)
|
|
Permanency.objects.create(
|
|
user=cls.skia,
|
|
start=now + timedelta(days=4, hours=1),
|
|
end=now + timedelta(days=20, hours=2, minutes=35, seconds=54),
|
|
counter=cls.counter,
|
|
)
|
|
|
|
# total of root : 1 hour + 20 hours (but the 20 hours were on last year)
|
|
Permanency.objects.create(
|
|
user=cls.root,
|
|
start=now + timedelta(days=5),
|
|
end=now + timedelta(days=5, hours=1),
|
|
counter=cls.counter,
|
|
)
|
|
Permanency.objects.create(
|
|
user=cls.root,
|
|
start=now - timedelta(days=300, hours=20),
|
|
end=now - timedelta(days=300),
|
|
counter=cls.counter,
|
|
)
|
|
|
|
# total of krophil : 0 hour
|
|
s = Selling(
|
|
label=barbar.name,
|
|
product=barbar,
|
|
club=Club.objects.get(name=settings.SITH_MAIN_CLUB["name"]),
|
|
counter=cls.counter,
|
|
unit_price=2,
|
|
seller=cls.skia,
|
|
)
|
|
|
|
krophil_customer = Customer.get_or_create(cls.krophil)[0]
|
|
sli_customer = Customer.get_or_create(cls.sli)[0]
|
|
skia_customer = Customer.get_or_create(cls.skia)[0]
|
|
root_customer = Customer.get_or_create(cls.root)[0]
|
|
|
|
# moderate drinker. Total : 100 €
|
|
s.quantity = 50
|
|
s.customer = krophil_customer
|
|
s.save(allow_negative=True)
|
|
|
|
# Sli is a drunkard. Total : 2000 €
|
|
s.quantity = 100
|
|
s.customer = sli_customer
|
|
for _ in range(10):
|
|
# little trick to make sure the instance is duplicated in db
|
|
s.pk = None
|
|
s.save(allow_negative=True) # save ten different sales
|
|
|
|
# Skia is a heavy drinker too. Total : 1000 €
|
|
s.customer = skia_customer
|
|
for _ in range(5):
|
|
s.pk = None
|
|
s.save(allow_negative=True)
|
|
|
|
# Root is quite an abstemious one. Total : 2 €
|
|
s.pk = None
|
|
s.quantity = 1
|
|
s.customer = root_customer
|
|
s.save(allow_negative=True)
|
|
|
|
def test_not_authenticated_user_fail(self):
|
|
# Test with not login user
|
|
response = self.client.get(reverse("counter:stats", args=[self.counter.id]))
|
|
assert response.status_code == 403
|
|
|
|
def test_unauthorized_user_fails(self):
|
|
self.client.force_login(User.objects.get(username="public"))
|
|
response = self.client.get(reverse("counter:stats", args=[self.counter.id]))
|
|
assert response.status_code == 403
|
|
|
|
def test_get_total_sales(self):
|
|
"""Test the result of the Counter.get_total_sales() method."""
|
|
assert self.counter.get_total_sales() == 3102
|
|
|
|
def test_top_barmen(self):
|
|
"""Test the result of Counter.get_top_barmen() is correct."""
|
|
users = [self.skia, self.root, self.sli]
|
|
perm_times = [
|
|
timedelta(days=16, hours=2, minutes=35, seconds=54),
|
|
timedelta(hours=21),
|
|
timedelta(hours=5),
|
|
]
|
|
assert list(self.counter.get_top_barmen()) == [
|
|
{
|
|
"user": user.id,
|
|
"name": f"{user.first_name} {user.last_name}",
|
|
"promo": user.promo,
|
|
"nickname": user.nick_name,
|
|
"perm_sum": perm_time,
|
|
}
|
|
for user, perm_time in zip(users, perm_times)
|
|
]
|
|
|
|
def test_top_customer(self):
|
|
"""Test the result of Counter.get_top_customers() is correct."""
|
|
users = [self.sli, self.skia, self.krophil, self.root]
|
|
sale_amounts = [2000, 1000, 100, 2]
|
|
assert list(self.counter.get_top_customers()) == [
|
|
{
|
|
"user": user.id,
|
|
"name": f"{user.first_name} {user.last_name}",
|
|
"promo": user.promo,
|
|
"nickname": user.nick_name,
|
|
"selling_sum": sale_amount,
|
|
}
|
|
for user, sale_amount in zip(users, sale_amounts)
|
|
]
|
|
|
|
|
|
class TestBarmanConnection(TestCase):
|
|
@classmethod
|
|
def setUpTestData(cls):
|
|
cls.krophil = User.objects.get(username="krophil")
|
|
cls.skia = User.objects.get(username="skia")
|
|
cls.skia.customer.account = 800
|
|
cls.krophil.customer.save()
|
|
cls.skia.customer.save()
|
|
|
|
cls.counter = Counter.objects.get(id=2)
|
|
|
|
def test_barman_granted(self):
|
|
self.client.post(
|
|
reverse("counter:login", args=[self.counter.id]),
|
|
{"username": "krophil", "password": "plop"},
|
|
)
|
|
response = self.client.get(reverse("counter:details", args=[self.counter.id]))
|
|
|
|
assert "<p>Entrez un code client : </p>" in str(response.content)
|
|
|
|
def test_counters_list_barmen(self):
|
|
self.client.post(
|
|
reverse("counter:login", args=[self.counter.id]),
|
|
{"username": "krophil", "password": "plop"},
|
|
)
|
|
response = self.client.get(reverse("counter:activity", args=[self.counter.id]))
|
|
|
|
assert '<li><a href="/user/10/">Kro Phil'</a></li>' in str(response.content)
|
|
|
|
def test_barman_denied(self):
|
|
self.client.post(
|
|
reverse("counter:login", args=[self.counter.id]),
|
|
{"username": "skia", "password": "plop"},
|
|
)
|
|
response_get = self.client.get(
|
|
reverse("counter:details", args=[self.counter.id])
|
|
)
|
|
|
|
assert "<p>Merci de vous identifier</p>" in str(response_get.content)
|
|
|
|
def test_counters_list_no_barmen(self):
|
|
self.client.post(
|
|
reverse("counter:login", args=[self.counter.id]),
|
|
{"username": "krophil", "password": "plop"},
|
|
)
|
|
response = self.client.get(reverse("counter:activity", args=[self.counter.id]))
|
|
|
|
assert '<li><a href="/user/1/">S' Kia</a></li>' not in str(response.content)
|
|
|
|
|
|
@pytest.mark.django_db
|
|
def test_barman_timeout():
|
|
"""Test that barmen timeout is well managed."""
|
|
bar = baker.make(Counter, type="BAR")
|
|
user = baker.make(User)
|
|
bar.sellers.add(user)
|
|
baker.make(Permanency, counter=bar, user=user, start=now())
|
|
|
|
qs = Counter.objects.annotate_is_open().filter(pk=bar.pk)
|
|
|
|
bar = qs[0]
|
|
assert bar.is_open
|
|
assert bar.barmen_list == [user]
|
|
qs.handle_timeout() # handling timeout before the actual timeout should be no-op
|
|
assert qs[0].is_open
|
|
with freeze_time() as frozen_time:
|
|
frozen_time.tick(timedelta(minutes=settings.SITH_BARMAN_TIMEOUT + 1))
|
|
qs.handle_timeout()
|
|
bar = qs[0]
|
|
assert not bar.is_open
|
|
assert bar.barmen_list == []
|
|
|
|
|
|
class TestClubCounterClickAccess(TestCase):
|
|
@classmethod
|
|
def setUpTestData(cls):
|
|
cls.counter = baker.make(Counter, type="OFFICE")
|
|
cls.customer = subscriber_user.make()
|
|
cls.counter_url = reverse(
|
|
"counter:details", kwargs={"counter_id": cls.counter.id}
|
|
)
|
|
cls.click_url = reverse(
|
|
"counter:click",
|
|
kwargs={"counter_id": cls.counter.id, "user_id": cls.customer.id},
|
|
)
|
|
|
|
cls.user = subscriber_user.make()
|
|
|
|
def setUp(self):
|
|
cache.clear()
|
|
|
|
def test_anonymous(self):
|
|
res = self.client.get(self.click_url)
|
|
assert res.status_code == 403
|
|
|
|
def test_logged_in_without_rights(self):
|
|
self.client.force_login(self.user)
|
|
res = self.client.get(self.click_url)
|
|
assert res.status_code == 403
|
|
# being a member of the club, without being in the board, isn't enough
|
|
baker.make(Membership, club=self.counter.club, user=self.user, role=1)
|
|
res = self.client.get(self.click_url)
|
|
assert res.status_code == 403
|
|
|
|
def test_board_member(self):
|
|
baker.make(Membership, club=self.counter.club, user=self.user, role=3)
|
|
self.client.force_login(self.user)
|
|
res = self.client.get(self.click_url)
|
|
assert res.status_code == 200
|
|
|
|
def test_barman(self):
|
|
self.counter.sellers.add(self.user)
|
|
self.client.force_login(self.user)
|
|
res = self.client.get(self.click_url)
|
|
assert res.status_code == 200
|