cleaner - adds delete_gatekeeper_appointments

delete_gatekeeper_appointments takes care of deleting data from the Gatekeeper once it has expired / it has been completed

test/teos/unit/test_cleaner.py

@@ -1,8 +1,9 @@
 import random
 from uuid import uuid4
 
-from teos.responder import TransactionTracker
 from teos.cleaner import Cleaner
+from teos.gatekeeper import UserInfo
+from teos.responder import TransactionTracker
 from common.appointment import Appointment
 
 from test.teos.unit.conftest import get_random_value_hex
@@ -206,3 +207,36 @@ def test_delete_trackers_no_db_match(db_manager):
         # We should be able to delete the correct ones and not fail in the others
         Cleaner.delete_trackers(completed_trackers, height, trackers, tx_tracker_map, db_manager)
         assert not set(completed_trackers).issubset(trackers.keys())
+
+
+def test_delete_gatekeeper_appointments(gatekeeper):
+    # delete_gatekeeper_appointments should delete the appointments from user as long as both exist
+
+    appointments_not_to_delete = {}
+    appointments_to_delete = {}
+    # Let's add some users and appointments to the Gatekeeper
+    for _ in range(10):
+        user_id = get_random_value_hex(16)
+        # The UserInfo params do not matter much here
+        gatekeeper.registered_users[user_id] = UserInfo(available_slots=100, subscription_expiry=0)
+        for _ in range(random.randint(0, 10)):
+            # Add some appointments
+            uuid = get_random_value_hex(16)
+            gatekeeper.registered_users[user_id].appointments[uuid] = 1
+
+            if random.randint(0, 1) % 2:
+                appointments_to_delete[uuid] = user_id
+            else:
+                appointments_not_to_delete[uuid] = user_id
+
+    # Now let's delete half of them
+    Cleaner.delete_gatekeeper_appointments(gatekeeper, appointments_to_delete)
+
+    all_appointments_gatekeeper = []
+    # Let's get all the appointments in the Gatekeeper
+    for user_id, user in gatekeeper.registered_users.items():
+        all_appointments_gatekeeper.extend(user.appointments)
+
+    # Check that the first half of the appointments are not in the Gatekeeper, but the second half is
+    assert not set(appointments_to_delete).issubset(all_appointments_gatekeeper)
+    assert set(appointments_not_to_delete).issubset(all_appointments_gatekeeper)