fix spring config enable/disable commands (#7605)

Azure · May 17, 2024 · f4f31a2 · f4f31a2
1 parent 1f15ae8
commit f4f31a2
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 4 deletions.
diff --git a/src/spring/HISTORY.md b/src/spring/HISTORY.md
@@ -1,5 +1,9 @@
 Release History
 ===============
+1.24.2
+---
+* Fix commands `az spring config-server enable` and `az spring config-server disable`.
+
 1.24.1
 ---
 * Add command group `az spring private-dns-zone` to configure constumer private DNS zone with Azure Spring Apps.

diff --git a/src/spring/azext_spring/custom.py b/src/spring/azext_spring/custom.py
@@ -746,23 +746,23 @@ def eureka_disable(cmd, client, resource_group, name):
 
 
 def config_enable(cmd, client, resource_group, service):
-    config_server_resource = client.get(resource_group, service)
+    config_server_resource = client.config_servers.get(resource_group, service)
     if not config_server_resource.properties.enabled_state:
         raise CLIError("Only supported Standard consumption Tier.")
 
     config_server_properties = models.ConfigServerProperties(enabled_state="Enabled")
     config_server_resource = models.ConfigServerResource(properties=config_server_properties)
-    return cached_put(cmd, client.begin_update_patch, config_server_resource, resource_group, service).result()
+    return cached_put(cmd, client.config_servers.begin_update_patch, config_server_resource, resource_group, service).result()
 
 
 def config_disable(cmd, client, resource_group, service):
-    config_server_resource = client.get(resource_group, service)
+    config_server_resource = client.config_servers.get(resource_group, service)
     if not config_server_resource.properties.enabled_state:
         raise CLIError("Only supported Standard consumption Tier.")
 
     config_server_properties = models.ConfigServerProperties(enabled_state="Disabled")
     config_server_resource = models.ConfigServerResource(properties=config_server_properties)
-    return cached_put(cmd, client.begin_update_patch, config_server_resource, resource_group, service).result()
+    return cached_put(cmd, client.config_servers.begin_update_patch, config_server_resource, resource_group, service).result()
 
 
 def config_set(cmd, client, resource_group, service, config_file, no_wait=False):