From 9c93c004ecec4b12077cc0910bbe2f310b7444ff Mon Sep 17 00:00:00 2001
From: Sli <antoine@bartuccio.fr>
Date: Mon, 23 Dec 2024 01:18:01 +0100
Subject: [PATCH] Add more counter click tests

---
 counter/tests/test_counter.py | 76 ++++++++++++++++++++++++++++++++++-
 1 file changed, 75 insertions(+), 1 deletion(-)

diff --git a/counter/tests/test_counter.py b/counter/tests/test_counter.py
index 1649196d..99e9ac77 100644
--- a/counter/tests/test_counter.py
+++ b/counter/tests/test_counter.py
@@ -21,6 +21,7 @@ from django.conf import settings
 from django.contrib.auth.models import make_password
 from django.core.cache import cache
 from django.http import HttpResponse
+from django.shortcuts import resolve_url
 from django.test import Client, TestCase
 from django.urls import reverse
 from django.utils import timezone
@@ -30,7 +31,7 @@ from model_bakery import baker
 
 from club.models import Club, Membership
 from core.baker_recipes import board_user, old_subscriber_user, subscriber_user
-from core.models import User
+from core.models import Group, User
 from counter.baker_recipes import product_recipe
 from counter.models import (
     Counter,
@@ -219,12 +220,22 @@ class TestCounterClick(FullClickSetup, TestCase):
         super().setUpTestData()
 
         cls.underage_customer = subscriber_user.make()
+        cls.banned_counter_customer = subscriber_user.make()
+        cls.banned_alcohol_customer = subscriber_user.make()
 
         cls.set_age(cls.customer, 20)
         cls.set_age(cls.barmen, 20)
         cls.set_age(cls.club_admin, 20)
+        cls.set_age(cls.banned_alcohol_customer, 20)
         cls.set_age(cls.underage_customer, 17)
 
+        cls.banned_alcohol_customer.groups.add(
+            Group.objects.get(pk=settings.SITH_GROUP_BANNED_ALCOHOL_ID)
+        )
+        cls.banned_counter_customer.groups.add(
+            Group.objects.get(pk=settings.SITH_GROUP_BANNED_COUNTER_ID)
+        )
+
         cls.beer = product_recipe.make(
             limit_age=18, selling_price="1.5", special_selling_price="1"
         )
@@ -376,6 +387,69 @@ class TestCounterClick(FullClickSetup, TestCase):
 
         assert self.updated_amount(self.customer) == Decimal("8")
 
+    def test_click_alcool_unauthorized(self):
+        self.login_in_bar()
+
+        for user in [self.underage_customer, self.banned_alcohol_customer]:
+            self.refill_user(user, 10)
+
+            # Buy product without age limit
+            assert (
+                self.submit_basket(
+                    user,
+                    [
+                        BasketItem(self.snack.id, 2),
+                    ],
+                ).status_code
+                == 302
+            )
+
+            assert self.updated_amount(user) == Decimal("7")
+
+            # Buy product without age limit
+            assert (
+                self.submit_basket(
+                    user,
+                    [
+                        BasketItem(self.beer.id, 2),
+                    ],
+                ).status_code
+                == 200
+            )
+
+            assert self.updated_amount(user) == Decimal("7")
+
+    def test_click_unauthorized_customer(self):
+        self.login_in_bar()
+
+        for user in [
+            self.banned_counter_customer,
+            self.customer_old_can_not_buy,
+        ]:
+            self.refill_user(user, 10)
+            resp = self.submit_basket(
+                user,
+                [
+                    BasketItem(self.snack.id, 2),
+                ],
+            )
+            assert resp.status_code == 302
+            assert resp.url == resolve_url(self.counter)
+
+            assert self.updated_amount(self.banned_counter_customer) == Decimal("10")
+
+    def test_click_user_without_customer(self):
+        self.login_in_bar()
+        assert (
+            self.submit_basket(
+                self.customer_can_not_buy,
+                [
+                    BasketItem(self.snack.id, 2),
+                ],
+            ).status_code
+            == 404
+        )
+
     def test_annotate_has_barman_queryset(self):
         """Test if the custom queryset method `annotate_has_barman` works as intended."""
         counters = Counter.objects.annotate_has_barman(self.barmen)