Skip to content

Commit

Permalink
merged with main
Browse files Browse the repository at this point in the history
  • Loading branch information
venkatajagannath committed Jul 21, 2024
1 parent cf5d3e8 commit c026144
Showing 1 changed file with 5 additions and 11 deletions.
16 changes: 5 additions & 11 deletions tests/hooks/test_ray_hooks.py
Original file line number Diff line number Diff line change
Expand Up @@ -54,24 +54,18 @@ def test_get_connection_form_widgets(self):
from wtforms import BooleanField, PasswordField, StringField

expected_widgets = {
"address": StringField(lazy_gettext("Ray address path"), widget=BS3TextFieldWidget()),
"address": StringField(lazy_gettext("Ray dashboard URL"), widget=BS3TextFieldWidget()),
"create_cluster_if_needed": BooleanField(lazy_gettext("Create cluster if needed")),
"cookies": StringField(lazy_gettext("Ray job cookies"), widget=BS3TextFieldWidget()),
"metadata": StringField(lazy_gettext("Ray job metadata"), widget=BS3TextFieldWidget()),
"headers": StringField(lazy_gettext("Ray job headers"), widget=BS3TextFieldWidget()),
"verify": BooleanField(lazy_gettext("Ray job verify")),
"cookies": StringField(lazy_gettext("Cookies"), widget=BS3TextFieldWidget()),
"metadata": StringField(lazy_gettext("Metadata"), widget=BS3TextFieldWidget()),
"headers": StringField(lazy_gettext("Headers"), widget=BS3TextFieldWidget()),
"verify": BooleanField(lazy_gettext("Verify")),
"kube_config_path": StringField(lazy_gettext("Kube config path"), widget=BS3TextFieldWidget()),
"kube_config": PasswordField(lazy_gettext("Kube config (JSON format)"), widget=BS3PasswordFieldWidget()),
"namespace": StringField(lazy_gettext("Namespace"), widget=BS3TextFieldWidget()),
"cluster_context": StringField(lazy_gettext("Cluster context"), widget=BS3TextFieldWidget()),
"disable_verify_ssl": BooleanField(lazy_gettext("Disable SSL")),
"disable_tcp_keepalive": BooleanField(lazy_gettext("Disable TCP keepalive")),
"xcom_sidecar_container_image": StringField(
lazy_gettext("XCom sidecar image"), widget=BS3TextFieldWidget()
),
"xcom_sidecar_container_resources": StringField(
lazy_gettext("XCom sidecar resources (JSON format)"), widget=BS3TextFieldWidget()
),
}

widgets = RayHook.get_connection_form_widgets()
Expand Down

0 comments on commit c026144

Please sign in to comment.