[TSDK-298] Support collective and group events (nylas#217)

mrashed-dev · web-flow · commit 6acd150c62ba · 2022-05-10T16:31:37.000-04:00
This PR bring support to collective and group events.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -1,6 +1,10 @@
 nylas-python Changelog
 ======================
 
+Unreleased
+----------------
+* Add support for collective and group events
+
 v5.8.0
 ----------------
 * Add support for getting the number of queried objects (count view)
diff --git a/nylas/client/client.py b/nylas/client/client.py
@@ -316,6 +316,7 @@ def availability(
         interval,
         start_at,
         end_at,
+        event_collection_id=None,
         buffer=None,
         round_robin=None,
         free_busy=None,
@@ -356,6 +357,8 @@ def availability(
             data["buffer"] = buffer
         if round_robin is not None:
             data["round_robin"] = round_robin
+        if event_collection_id is not None:
+            data["event_collection_id"] = event_collection_id
 
         resp = self._request(HttpMethod.POST, url, json=data, cls=Calendar)
         _validate(resp)
diff --git a/nylas/client/restful_models.py b/nylas/client/restful_models.py
@@ -671,6 +671,9 @@ class Event(NylasAPIObject):
         "ical_uid",
         "metadata",
         "notifications",
+        "event_collection_id",
+        "capacity",
+        "round_robin_order",
     ]
     datetime_attrs = {"original_start_at": "original_start_time"}
     collection_name = "events"
@@ -772,7 +775,7 @@ def generate_ics(self, ical_uid=None, method=None, prodid=None):
             "Unexpected response from the API server. Returned 200 but no 'ics' string found."
         )
 
-    def save(self, **kwargs):
+    def validate(self):
         if (
             self.conferencing
             and "details" in self.conferencing
@@ -781,6 +784,18 @@ def save(self, **kwargs):
             raise ValueError(
                 "Cannot set both 'details' and 'autocreate' in conferencing object."
             )
+        if (
+            self.capacity
+            and self.capacity != -1
+            and self.participants
+            and len(self.participants) > self.capacity
+        ):
+            raise ValueError(
+                "The number of participants in the event exceeds the set capacity."
+            )
+
+    def save(self, **kwargs):
+        self.validate()
 
         super(Event, self).save(**kwargs)
 
diff --git a/tests/test_events.py b/tests/test_events.py
@@ -138,6 +138,51 @@ def test_event_conferencing_details_autocreate_error(mocked_responses, api_clien
     )
 
 
+@pytest.mark.usefixtures("mock_event_create_response")
+def test_event_error_if_participants_more_than_capacity(mocked_responses, api_client):
+    event = blank_event(api_client)
+    event.capacity = 1
+    event.participants = [
+        {"email": "person1@email.com"},
+        {"email": "person2@email.com"},
+    ]
+    with pytest.raises(ValueError) as excinfo:
+        event.save()
+    assert "The number of participants in the event exceeds the set capacity." in str(
+        excinfo
+    )
+
+
+@pytest.mark.usefixtures("mock_event_create_response")
+def test_event_no_error_if_capacity_negative_one(mocked_responses, api_client):
+    event = blank_event(api_client)
+    event.capacity = -1
+    event.participants = [
+        {"email": "person1@email.com"},
+        {"email": "person2@email.com"},
+    ]
+    event.save()
+
+
+@pytest.mark.usefixtures("mock_event_create_response")
+def test_event_no_error_if_participants_less_than_eql_capacity(
+    mocked_responses, api_client
+):
+    event = blank_event(api_client)
+    event.capacity = 2
+    event.participants = [
+        {"email": "person1@email.com"},
+        {"email": "person2@email.com"},
+    ]
+    event.save()
+    event.capacity = 3
+    event.participants = [
+        {"email": "person1@email.com"},
+        {"email": "person2@email.com"},
+    ]
+    event.save()
+
+
 @pytest.mark.usefixtures("mock_calendars", "mock_events")
 def test_calendar_events(api_client):
     calendar = api_client.calendars.first()
