added some functional tests

Author: cmdelatorre

test/conftest.py

@@ -5,13 +5,13 @@
 from peewee import SqliteDatabase
 from telegram import Bot
 
-from pycamp_bot.models import Pycampista, Slot, Pycamp, WizardAtPycamp
+from pycamp_bot.models import Pycampista, Slot, Pycamp, WizardAtPycamp, PycampistaAtPycamp
 
 
 # use an in-memory SQLite for tests.
 test_db = SqliteDatabase(':memory:')
 
-MODELS = [Pycampista, Slot, Pycamp, WizardAtPycamp]
+MODELS = [Pycampista, Slot, Pycamp, WizardAtPycamp, PycampistaAtPycamp]
 
 
 def use_test_database(fn):

test/test_pycamp_model.py

@@ -42,7 +42,7 @@ def test_returns_correct_wizard_within_its_turno(self):
             init=datetime(2024,6,20),
             end=datetime(2024,6,23),
         )
-        pycamper = Pycampista.create(username="pepe", wizard=True)
+        pycamper = p.add_wizard("pepe", 123)
         wizard.persist_wizards_schedule_in_db(p)
 
         assert p.get_current_wizard() == pycamper

test/test_wizard.py

@@ -22,7 +22,7 @@ def teardown_module(module):
     # database here. But for tests this is probably not necessary.
 
 
-class TestWizardScheduleSlots:
+class BaseForOtherWizardsTests:
 
     def init_pycamp(self):
         self.pycamp = Pycamp.create(
@@ -31,6 +31,8 @@ def init_pycamp(self):
             end=datetime(2024,6,24),
         )
 
+
+class TestWizardScheduleSlots(BaseForOtherWizardsTests):
     @use_test_database
     def test_correct_number_of_slots_in_one_day(self):
         p = Pycamp.create(
@@ -95,14 +97,7 @@ def test_no_slot_after_last_day_lunch(self):
                 assert start >= lunch_time_end
 
 
-class TestDefineWizardsSchedule:
-
-    def init_pycamp(self):
-        self.pycamp = Pycamp.create(
-            headquarters="Narnia",
-            init=datetime(2024,6,20),
-            end=datetime(2024,6,24),
-        )
+class TestDefineWizardsSchedule(BaseForOtherWizardsTests):
 
     # If no wizards, returns {}
     @use_test_database
@@ -164,3 +159,41 @@ def test_all_slots_are_signed_a_wizard(self):
         assert all(
             (isinstance(s, Pycampista) and s.wizard) for s in sched.values()
         )
+
+class TestListWizards(BaseForOtherWizardsTests):
+
+    @use_test_database
+    def test_wizard_registration(self):
+        self.init_pycamp()
+        w = self.pycamp.add_wizard("Gandalf", 123)
+        wizards = self.pycamp.get_wizards()
+        assert len(wizards) == 1
+        assert w.username == wizards[0].username
+
+
+    @use_test_database
+    def test_wizard_registration_works_in_one_pycamp_only(self):
+        self.init_pycamp()
+        self.pycamp.add_wizard("Gandalf", 123)
+        
+        other_pycamp = Pycamp.create(
+            headquarters="Mordor",
+            init=datetime(2025,3,22),
+            end=datetime(2025,3,24),
+        )
+        w = other_pycamp.add_wizard("Merlin", 456)
+        #import ipdb; ipdb.set_trace()
+        results = other_pycamp.get_wizards()
+        assert len(results) == 1
+        assert w.username == results[0].username
+
+
+    # @use_test_database
+    # def test_no_active_pycamp_then_fail(self):
+    #     self.init_pycamp()
+    #     agregarle un mago
+        
+    #     creo OTRO Pycamp
+    #     agregarle otro mago
+
+    #     pedir listado: check está el OTRO mago solamente