mirror of
https://github.com/ae-utbm/sith.git
synced 2026-03-13 15:15:03 +00:00
225 lines
7.8 KiB
Python
225 lines
7.8 KiB
Python
from io import BytesIO
|
|
from typing import Callable
|
|
from uuid import uuid4
|
|
|
|
import pytest
|
|
from django.conf import settings
|
|
from django.core.cache import cache
|
|
from django.core.files.uploadedfile import SimpleUploadedFile
|
|
from django.test import Client, TestCase
|
|
from django.urls import reverse
|
|
from model_bakery import baker
|
|
from model_bakery.recipe import Recipe
|
|
from PIL import Image
|
|
from pytest_django.asserts import assertNumQueries, assertRedirects
|
|
|
|
from club.models import Club
|
|
from core.baker_recipes import board_user, subscriber_user
|
|
from core.models import Group, User
|
|
from counter.baker_recipes import product_recipe
|
|
from counter.forms import ProductForm, ProductPriceFormSet
|
|
from counter.models import Price, Product, ProductType
|
|
|
|
|
|
@pytest.mark.django_db
|
|
@pytest.mark.parametrize("model", [Product, ProductType])
|
|
def test_resize_product_icon(model):
|
|
"""Test that the product icon is resized when saved."""
|
|
# Product and ProductType icons have a height of 70px
|
|
# so this image should be resized to 50x70
|
|
img = Image.new("RGB", (100, 140))
|
|
content = BytesIO()
|
|
img.save(content, format="JPEG")
|
|
name = str(uuid4())
|
|
|
|
product = baker.make(
|
|
model,
|
|
icon=SimpleUploadedFile(
|
|
f"{name}.jpg", content.getvalue(), content_type="image/jpeg"
|
|
),
|
|
)
|
|
|
|
assert product.icon.width == 50
|
|
assert product.icon.height == 70
|
|
assert product.icon.name == f"products/{name}.webp"
|
|
assert Image.open(product.icon).format == "WEBP"
|
|
|
|
|
|
@pytest.mark.django_db
|
|
@pytest.mark.parametrize(
|
|
("user_factory", "status_code"),
|
|
[
|
|
(lambda: baker.make(User, is_superuser=True), 200),
|
|
(board_user.make, 403),
|
|
(subscriber_user.make, 403),
|
|
(
|
|
lambda: baker.make(
|
|
User,
|
|
groups=[Group.objects.get(pk=settings.SITH_GROUP_COUNTER_ADMIN_ID)],
|
|
),
|
|
200,
|
|
),
|
|
(
|
|
lambda: baker.make(
|
|
User,
|
|
groups=[Group.objects.get(pk=settings.SITH_GROUP_ACCOUNTING_ADMIN_ID)],
|
|
),
|
|
200,
|
|
),
|
|
],
|
|
)
|
|
def test_fetch_product_access(
|
|
client: Client, user_factory: Callable[[], User], status_code: int
|
|
):
|
|
"""Test that only authorized users can use the `GET /product` route."""
|
|
client.force_login(user_factory())
|
|
assert (
|
|
client.get(reverse("api:search_products_detailed")).status_code == status_code
|
|
)
|
|
|
|
|
|
@pytest.mark.django_db
|
|
def test_fetch_product_nb_queries(client: Client):
|
|
client.force_login(baker.make(User, is_superuser=True))
|
|
cache.clear()
|
|
with assertNumQueries(6):
|
|
# - 2 for authentication
|
|
# - 1 for pagination
|
|
# - 1 for the actual request
|
|
# - 2 to prefetch the related prices and groups
|
|
client.get(reverse("api:search_products_detailed"))
|
|
|
|
|
|
class TestCreateProduct(TestCase):
|
|
@classmethod
|
|
def setUpTestData(cls):
|
|
cls.product_type = baker.make(ProductType)
|
|
cls.club = baker.make(Club)
|
|
cls.counter_admin = baker.make(
|
|
User, groups=[Group.objects.get(id=settings.SITH_GROUP_COUNTER_ADMIN_ID)]
|
|
)
|
|
cls.data = {
|
|
"name": "foo",
|
|
"description": "bar",
|
|
"product_type": cls.product_type.id,
|
|
"club": cls.club.id,
|
|
"code": "FOO",
|
|
"purchase_price": 1.0,
|
|
"selling_price": 1.0,
|
|
"special_selling_price": 1.0,
|
|
"limit_age": 0,
|
|
"price-TOTAL_FORMS": 0,
|
|
"price-INITIAL_FORMS": 0,
|
|
"action-TOTAL_FORMS": 0,
|
|
"action-INITIAL_FORMS": 0,
|
|
}
|
|
|
|
def test_form_simple(self):
|
|
form = ProductForm(data=self.data)
|
|
assert form.is_valid()
|
|
instance = form.save()
|
|
assert instance.club == self.club
|
|
assert instance.product_type == self.product_type
|
|
assert instance.name == "foo"
|
|
|
|
def test_view_simple(self):
|
|
self.client.force_login(self.counter_admin)
|
|
url = reverse("counter:new_product")
|
|
response = self.client.get(url)
|
|
assert response.status_code == 200
|
|
response = self.client.post(url, data=self.data)
|
|
assertRedirects(response, reverse("counter:product_list"))
|
|
product = Product.objects.last()
|
|
assert product.name == "foo"
|
|
assert product.club == self.club
|
|
assert product.product_type == self.product_type
|
|
|
|
|
|
class TestPriceFormSet(TestCase):
|
|
@classmethod
|
|
def setUpTestData(cls):
|
|
cls.product = product_recipe.make()
|
|
cls.counter_admin = baker.make(
|
|
User, groups=[Group.objects.get(id=settings.SITH_GROUP_COUNTER_ADMIN_ID)]
|
|
)
|
|
cls.groups = baker.make(Group, _quantity=3)
|
|
|
|
def test_add_price(self):
|
|
data = {
|
|
"prices-0-amount": 2,
|
|
"prices-0-label": "foo",
|
|
"prices-0-groups": [self.groups[0].id, self.groups[1].id],
|
|
"prices-0-is_always_shown": True,
|
|
"prices-1-amount": 1.5,
|
|
"prices-1-label": "",
|
|
"prices-1-groups": [self.groups[1].id, self.groups[2].id],
|
|
"prices-1-is_always_shown": False,
|
|
"prices-TOTAL_FORMS": 2,
|
|
"prices-INITIAL_FORMS": 0,
|
|
}
|
|
form = ProductPriceFormSet(instance=self.product, data=data)
|
|
assert form.is_valid()
|
|
form.save()
|
|
prices = list(self.product.prices.order_by("amount"))
|
|
assert len(prices) == 2
|
|
assert prices[0].amount == 1.5
|
|
assert prices[0].label == ""
|
|
assert prices[0].is_always_shown is False
|
|
assert set(prices[0].groups.all()) == {self.groups[1], self.groups[2]}
|
|
assert prices[1].amount == 2
|
|
assert prices[1].label == "foo"
|
|
assert prices[1].is_always_shown is True
|
|
assert set(prices[1].groups.all()) == {self.groups[0], self.groups[1]}
|
|
|
|
def test_change_prices(self):
|
|
price_a = baker.make(
|
|
Price, product=self.product, amount=1.5, groups=self.groups[:1]
|
|
)
|
|
price_b = baker.make(
|
|
Price, product=self.product, amount=2, groups=self.groups[1:]
|
|
)
|
|
data = {
|
|
"prices-0-id": price_a.id,
|
|
"prices-0-DELETE": True,
|
|
"prices-1-id": price_b.id,
|
|
"prices-1-DELETE": False,
|
|
"prices-1-amount": 3,
|
|
"prices-1-label": "foo",
|
|
"prices-1-groups": [self.groups[1].id],
|
|
"prices-1-is_always_shown": True,
|
|
"prices-TOTAL_FORMS": 2,
|
|
"prices-INITIAL_FORMS": 2,
|
|
}
|
|
form = ProductPriceFormSet(instance=self.product, data=data)
|
|
assert form.is_valid()
|
|
form.save()
|
|
prices = list(self.product.prices.order_by("amount"))
|
|
assert len(prices) == 1
|
|
assert prices[0].amount == 3
|
|
assert prices[0].label == "foo"
|
|
assert prices[0].is_always_shown is True
|
|
assert set(prices[0].groups.all()) == {self.groups[1]}
|
|
assert not Price.objects.filter(id=price_a.id).exists()
|
|
|
|
|
|
@pytest.mark.django_db
|
|
def test_price_for_user():
|
|
groups = baker.make(Group, _quantity=4)
|
|
users = [
|
|
baker.make(User, groups=groups[:2]),
|
|
baker.make(User, groups=groups[1:3]),
|
|
baker.make(User, groups=[groups[3]]),
|
|
]
|
|
recipe = Recipe(Price, product=product_recipe.make())
|
|
prices = [
|
|
recipe.make(amount=5, groups=groups, is_always_shown=True),
|
|
recipe.make(amount=4, groups=[groups[0]], is_always_shown=True),
|
|
recipe.make(amount=3, groups=[groups[1]], is_always_shown=False),
|
|
recipe.make(amount=2, groups=[groups[3]], is_always_shown=False),
|
|
recipe.make(amount=1, groups=[groups[1]], is_always_shown=False),
|
|
]
|
|
qs = Price.objects.order_by("-amount")
|
|
assert set(qs.for_user(users[0])) == {prices[0], prices[1], prices[4]}
|
|
assert set(qs.for_user(users[1])) == {prices[0], prices[4]}
|
|
assert set(qs.for_user(users[2])) == {prices[0], prices[3]}
|