fix test

Signed-off-by: Jess Frazelle <github@jessfraz.com>
2024-04-05 13:30:47 -07:00
parent df9083c3e2
commit 79b977a55a
5 changed files with 630 additions and 618 deletions
--- a/generate/functions-ws.py.jinja2
+++ b/generate/functions-ws.py.jinja2
@ -154,10 +154,19 @@ class WebSocket:
        self.ws = sync(
        {% for arg in args %}
        {% if arg.in_query %}
+        {% if arg.is_optional == False %}
            {{arg.name}},
        {% endif %}
+        {% endif %}
        {% endfor %}
            client=client,
+        {% for arg in args %}
+        {% if arg.in_query %}
+        {% if arg.is_optional %}
+        {{arg.name}}={{arg.name}},
+        {% endif %}
+        {% endif %}
+        {% endfor %}
        )

    def __enter__(self,