fix: Add Category to Connect Webview Create (#96)

Author: abimaelmartell

seamapi/connect_webviews.py

@@ -105,7 +105,8 @@ def get(
     @report_error
     def create(
         self,
-        accepted_providers: List[AcceptedProvider],
+        accepted_providers: Optional[List[AcceptedProvider]] = None,
+        category: Optional[str] = None,
         custom_redirect_url: Optional[str] = None,
         custom_redirect_failure_url: Optional[str] = None,
         device_selection_mode: Optional[str] = None,
@@ -115,7 +116,9 @@ def create(
 
         Parameters
         ----------
-        accepted_providers : list[AcceptedProvider]
+        category : str, optional
+            Provider category e.g. stable
+        accepted_providers : list[AcceptedProvider], optional
             A list of accepted providers e.g. august or noiseaware
         custom_redirect_url : str, optional
             Custom redirect url
@@ -135,7 +138,17 @@ def create(
             ConnectWebview
         """
 
-        create_payload = {"accepted_providers": accepted_providers}
+        create_payload = {}
+
+        if accepted_providers is None and category is None:
+            raise Exception(
+                "Must provide either accepted_providers or category"
+            )
+
+        if accepted_providers is not None:
+            create_payload["accepted_providers"] = accepted_providers
+        if category is not None:
+            create_payload["category"] = category
         if custom_redirect_url is not None:
             create_payload["custom_redirect_url"] = custom_redirect_url
         if custom_redirect_failure_url is not None:

seamapi/types.py

@@ -397,7 +397,9 @@ def get(self, connect_webview_id: str) -> ConnectWebview:
 
     @abc.abstractmethod
     def create(
-        self, accepted_providers: Optional[List[AcceptedProvider]] = None
+        self,
+        accepted_providers: Optional[List[AcceptedProvider]] = None,
+        category: Optional[str] = None,
     ) -> ConnectWebview:
         raise NotImplementedError
 

tests/connect_webviews/test_connect_webviews.py

@@ -10,3 +10,10 @@ def test_connect_webviews(seam: Seam):
 
     webviews = seam.connect_webviews.list()
     assert len(webviews) > 0
+
+    # Test with Category
+    new_webview = seam.connect_webviews.create(category=["stable"])
+    assert created_webview.url is not None
+
+    webview = seam.connect_webviews.get(new_webview.connect_webview_id)
+    assert len(webview.accepted_providers) > 0