yaml tests

Author: Lingxi Chen

rest-api-spec/src/main/resources/rest-api-spec/api/wlm_stats_list.json

@@ -0,0 +1,35 @@
+{
+  "wlm_stats_list": {
+    "stability": "experimental",
+    "url": {
+      "paths": [
+        {
+          "path": "/_list/wlm_stats",
+          "methods": ["GET"]
+        }
+      ]
+    },
+    "params": {
+      "size": {
+        "type": "int",
+        "required": false,
+        "description": "Number of results per page"
+      },
+      "next_token": {
+        "type": "string",
+        "required": false,
+        "description": "Pagination token for next page"
+      },
+      "sort": {
+        "type": "string",
+        "required": false,
+        "description": "Sort field"
+      },
+      "order": {
+        "type": "string",
+        "required": false,
+        "description": "Sort order (asc or desc)"
+      }
+    }
+  }
+}

rest-api-spec/src/main/resources/rest-api-spec/test/wlm_stats/10_basic.yml

@@ -0,0 +1,55 @@
+---
+"Basic usage returns expected table header":
+  - do:
+      wlm_stats_list: {}
+  - is_true: $body
+  - match:
+      $body: /(?s).*NODE_ID\s+\|\s+QUERY_GROUP_ID\s+\|\s+TOTAL_COMPLETIONS\s+\|\s+TOTAL_REJECTIONS\s+\|\s+TOTAL_CANCELLATIONS\s+\|\s+CPU_USAGE\s+\|\s+MEMORY_USAGE.*/
+
+---
+"Custom size param still returns headers":
+  - do:
+      wlm_stats_list:
+        size: 1
+  - is_true: $body
+  - match:
+      $body: /TOTAL_REJECTIONS/
+
+---
+"Sort by node_id asc does not error":
+  - do:
+      wlm_stats_list:
+        sort: node_id
+        order: asc
+  - is_true: $body
+  - match:
+      $body: /NODE_ID/
+
+---
+"Sort by query_group_id desc does not error":
+  - do:
+      wlm_stats_list:
+        sort: query_group_id
+        order: desc
+  - is_true: $body
+  - match:
+      $body: /QUERY_GROUP_ID/
+
+---
+"Invalid sort field returns error":
+  - do:
+      catch: bad_request
+      wlm_stats_list:
+        sort: memory_usage
+        order: desc
+  - match:
+      error.reason: "Invalid value for 'sort'. Allowed: 'node_id', 'query_group_id'."
+
+---
+"Invalid sort order returns error":
+  - do:
+      catch: bad_request
+      wlm_stats_list:
+        order: upside_down
+  - match:
+      error.reason: "Invalid value for 'order'. Allowed: 'asc', 'desc'."

server/src/main/java/org/opensearch/rest/action/admin/cluster/RestWlmStatsAction.java

@@ -75,6 +75,10 @@ protected RestChannelConsumer prepareRequest(RestRequest request, NodeClient cli
             throw new IllegalArgumentException("Invalid value for 'order'. Allowed: 'asc', 'desc'.");
         }
 
+        if (!sortBy.equals("node_id") && !sortBy.equals("query_group")) {
+            throw new IllegalArgumentException("Invalid value for 'sort'. Allowed: 'node_id', 'query_group_id'.");
+        }
+
         boolean isTabular = request.rawPath().contains("_list/wlm_stats");
 
         if (isTabular) {