[API] Test Runner: Adds internal desired balance API

elastic · Dec 6, 2022 · eebcb23 · eebcb23
1 parent 106a629
commit eebcb23
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/elasticsearch-api/api-spec-testing/test_file/action.rb b/elasticsearch-api/api-spec-testing/test_file/action.rb
@@ -161,9 +161,14 @@ def perform_internal(method, args, client, es_version)
           http = 'DELETE'
           path = '/_internal/desired_nodes/'
           body = args.delete('body')
-        when '_internal.get_desired_nodes'
+        when /_internal\.get_([a-z_]+)/
           http = 'GET'
-          path = '/_internal/desired_nodes/_latest'
+          path = case Regexp.last_match(1)
+                 when 'desired_nodes'
+                   '/_internal/desired_nodes/_latest'
+                 when 'desired_balance'
+                   '/_internal/desired_balance'
+                 end
           body = nil
         when '_internal.health'
           path = if args['feature']