one working

Signed-off-by: Jess Frazelle <github@jessfraz.com>

generate/functions-ws.py.jinja2

@@ -93,7 +93,7 @@ def sync(
         client=client,
     )
 
-    return ws_connect(kwargs["url"].replace("http", "ws"), additional_headers=kwargs["headers"]) # type: ignore
+    return ws_connect(kwargs["url"].replace("http", "ws"), additional_headers=kwargs["headers"], close_timeout=120, compression=None, max_size=None) # type: ignore
 
 
 
@@ -126,7 +126,7 @@ async def asyncio(
         client=client,
     )
 
-    return await ws_connect_async(kwargs["url"].replace("http", "ws"), extra_headers=kwargs["headers"])
+    return await ws_connect_async(kwargs["url"].replace("http", "ws"), extra_headers=kwargs["headers"], close_timeout=120, compression=None, max_size=None)
 
 
 {% if has_request_body %}

kittycad/api/executor/create_executor_term.py

@@ -33,7 +33,7 @@ def sync(
         client=client,
     )
 
-    return ws_connect(kwargs["url"].replace("http", "ws"), additional_headers=kwargs["headers"])  # type: ignore
+    return ws_connect(kwargs["url"].replace("http", "ws"), additional_headers=kwargs["headers"], close_timeout=120, compression=None, max_size=None)  # type: ignore
 
 
 async def asyncio(
@@ -47,5 +47,9 @@ async def asyncio(
     )
 
     return await ws_connect_async(
-        kwargs["url"].replace("http", "ws"), extra_headers=kwargs["headers"]
+        kwargs["url"].replace("http", "ws"),
+        extra_headers=kwargs["headers"],
+        close_timeout=120,
+        compression=None,
+        max_size=None,
     )

kittycad/api/modeling/modeling_commands_ws.py

@@ -82,7 +82,7 @@ def sync(
         client=client,
     )
 
-    return ws_connect(kwargs["url"].replace("http", "ws"), additional_headers=kwargs["headers"])  # type: ignore
+    return ws_connect(kwargs["url"].replace("http", "ws"), additional_headers=kwargs["headers"], close_timeout=120, compression=None, max_size=None)  # type: ignore
 
 
 async def asyncio(
@@ -106,7 +106,11 @@ async def asyncio(
     )
 
     return await ws_connect_async(
-        kwargs["url"].replace("http", "ws"), extra_headers=kwargs["headers"]
+        kwargs["url"].replace("http", "ws"),
+        extra_headers=kwargs["headers"],
+        close_timeout=120,
+        compression=None,
+        max_size=None,
     )
 
 

kittycad/client_test.py

@@ -309,7 +309,7 @@ def test_list_users():
     print(f"ExtendedUserResultsPage: {response}")
 
 
-def test_ws():
+def test_ws_simple():
     # Create our client.
     client = ClientFromEnv()