From 0af72500c6903d6afe71885292de1c26867b505a Mon Sep 17 00:00:00 2001
From: Krophil <pierre.brunet@krophil.fr>
Date: Sun, 14 May 2017 12:54:26 +0200
Subject: [PATCH] Add some tests on counter

---
 core/management/commands/populate.py |  5 +++
 counter/tests.py                     | 46 ++++++++++++++++++++++++++++
 2 files changed, 51 insertions(+)

diff --git a/core/management/commands/populate.py b/core/management/commands/populate.py
index b0962ff3..e03b04b9 100644
--- a/core/management/commands/populate.py
+++ b/core/management/commands/populate.py
@@ -473,6 +473,11 @@ Welcome to the wiki page!
                     start=s.subscription_start)
             s.save()
 
+            # Add barman to counter
+            c = Counter.objects.get(id=2)
+            c.sellers.add(User.objects.get(pk=krophil.pk))
+            c.save()
+
             # Create an election
             public_group = Group.objects.get(id=settings.SITH_GROUP_PUBLIC_ID)
             subscriber_group = Group.objects.get(name=settings.SITH_MAIN_MEMBERS_GROUP)
diff --git a/counter/tests.py b/counter/tests.py
index 7a1ce787..c1a3a198 100644
--- a/counter/tests.py
+++ b/counter/tests.py
@@ -22,6 +22,7 @@
 #
 #
 
+<<<<<<< 52a643878a10b52d9d7a17cd6c2854112c0a65d0
 import re
 from pprint import pprint
 
@@ -74,3 +75,48 @@ class CounterTest(TestCase):
         # TODO finir le test en vérifiant que les produits ont bien été clickés
         # hint: pprint(response.__dict__)
 
+
+class BarmanConnectionTest(TestCase):
+    def setUp(self):
+        call_command("populate")
+        self.krophil = User.objects.get(username="krophil")
+        self.skia = User.objects.get(username="skia")
+        self.skia.customer.account = 800
+        self.krophil.customer.save()
+        self.skia.customer.save()
+
+        self.counter = Counter.objects.filter(id = 2).first()
+
+    def test_barman_granted(self):
+        response_post = self.client.post(reverse('counter:login', args=[self.counter.id]),
+                                        {'username': "krophil",
+                                        'password' : "plop"})
+        response_get = self.client.get(reverse("counter:details",
+                                                args=[self.counter.id]))
+
+        self.assertTrue('<p>Enter client code:</p>' in str(response_get.content))
+
+    def test_counters_list_barmen(self):
+        response_post = self.client.post(reverse('counter:login', args=[self.counter.id]),
+                                        {'username': "krophil",
+                                        'password' : "plop"})
+        response_get = self.client.get(reverse("counter:activity",
+                                                args=[self.counter.id]))
+        
+        self.assertTrue('<li><a href="/user/10/">Kro Phil&#39;</a></li>' in str(response_get.content))
+    
+    def test_barman_denied(self):
+        response_post = 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]))
+        
+        self.assertTrue('<p>Please, login</p>' in str(response_get.content))
+
+    def test_counters_list_no_barmen(self):
+        response_post = self.client.post(reverse('counter:login', args=[self.counter.id]),
+                                        {'username': "krophil",
+                                        'password' : "plop"})
+        response_get = self.client.get(reverse("counter:activity", args=[self.counter.id]))
+        
+        self.assertFalse('<li><a href="/user/1/">S&#39; Kia</a></li>' in str(response_get.content))
\ No newline at end of file