Skip to content

Commit

Permalink
add IT for workload management CRUD APIs
Browse files Browse the repository at this point in the history 
Signed-off-by: Ruirui Zhang <mariazrr@amazon.com>
  • Loading branch information
@ruai0511
ruai0511 committed Sep 16, 2024
1 parent 2e98df3 commit 7f88472
Show file tree
Hide file tree
Showing 3 changed files with 247 additions and 0 deletions.
1 change: 1 addition & 0 deletions CHANGELOG.md
View file
Original file line number Diff line number Diff line change
Expand Up @@ -6,6 +6,7 @@ The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),
## [Unreleased 2.x]
### Added
- MultiTermQueries in keyword fields now default to `indexed` approach and gated behind cluster setting ([#15637](https://github.com/opensearch-project/OpenSearch/pull/15637))
- [Workload Management] Add Integration Tests for Workload Management CRUD APIs ([#15955](https://github.com/opensearch-project/OpenSearch/pull/15955))
- [Workload Management] QueryGroup resource cancellation framework changes ([#15651](https://github.com/opensearch-project/OpenSearch/pull/15651))

### Dependencies
Expand Down
1 change: 1 addition & 0 deletions plugins/workload-management/build.gradle
View file
Original file line number Diff line number Diff line change
Expand Up @@ -10,6 +10,7 @@
*/

apply plugin: 'opensearch.yaml-rest-test'
apply plugin: 'opensearch.java-rest-test'
apply plugin: 'opensearch.internal-cluster-test'

opensearchplugin {
Expand Down
245 changes: 245 additions & 0 deletions ...rkload-management/src/javaRestTest/java/org/opensearch/rest/WorkloadManagementRestIT.java
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,245 @@
/*
* SPDX-License-Identifier: Apache-2.0
*
* The OpenSearch Contributors require contributions made to
* this file be licensed under the Apache-2.0 license or a
* compatible open source license.
*/

package org.opensearch.rest;

import org.apache.hc.core5.http.io.entity.EntityUtils;
import org.opensearch.client.Request;
import org.opensearch.client.Response;
import org.opensearch.client.ResponseException;
import org.opensearch.test.rest.OpenSearchRestTestCase;

public class WorkloadManagementRestIT extends OpenSearchRestTestCase {

public void testCreate() throws Exception {
Request request = new Request("PUT", "_wlm/query_group");
request.setJsonEntity("{\n" +
" \"name\": \"analytics\",\n" +
" \"resiliency_mode\": \"enforced\",\n" +
" \"resource_limits\": {\n" +
" \"cpu\" : 0.4,\n" +
" \"memory\" : 0.2\n" +
" }\n" +
"}'");
Response response = client().performRequest(request);
assertEquals(response.getStatusLine().getStatusCode(), 200);
client().performRequest(new Request("DELETE", "_wlm/query_group/analytics"));
}

public void testMultipleCreate() throws Exception {
Request request = new Request("PUT", "_wlm/query_group");
request.setJsonEntity("{\n" +
" \"name\": \"analytics2\",\n" +
" \"resiliency_mode\": \"enforced\",\n" +
" \"resource_limits\": {\n" +
" \"cpu\" : 0.4,\n" +
" \"memory\" : 0.2\n" +
" }\n" +
"}'");
Response response = client().performRequest(request);
assertEquals(response.getStatusLine().getStatusCode(), 200);

Request request2 = new Request("PUT", "_wlm/query_group");
request2.setJsonEntity("{\n" +
" \"name\": \"users\",\n" +
" \"resiliency_mode\": \"soft\",\n" +
" \"resource_limits\": {\n" +
" \"cpu\" : 0.2,\n" +
" \"memory\" : 0.1\n" +
" }\n" +
"}'");
Response response2 = client().performRequest(request2);
assertEquals(response2.getStatusLine().getStatusCode(), 200);

Request request3 = new Request("PUT", "_wlm/query_group");
request3.setJsonEntity("{\n" +
" \"name\": \"users2\",\n" +
" \"resiliency_mode\": \"soft\",\n" +
" \"resource_limits\": {\n" +
" \"cpu\" : 0.41,\n" +
" \"memory\" : 0.71\n" +
" }\n" +
"}'");
assertThrows(ResponseException.class, () -> client().performRequest(request3));
client().performRequest(new Request("DELETE", "_wlm/query_group/analytics2"));
client().performRequest(new Request("DELETE", "_wlm/query_group/users"));
}

public void testGet() throws Exception {
Request request = new Request("PUT", "_wlm/query_group");
request.setJsonEntity("{\n" +
" \"name\": \"analytics3\",\n" +
" \"resiliency_mode\": \"enforced\",\n" +
" \"resource_limits\": {\n" +
" \"cpu\" : 0.4,\n" +
" \"memory\" : 0.2\n" +
" }\n" +
"}'");
Response response = client().performRequest(request);
assertEquals(response.getStatusLine().getStatusCode(), 200);

Request request2 = new Request("GET", "_wlm/query_group/analytics3");
Response response2 = client().performRequest(request2);
assertEquals(response2.getStatusLine().getStatusCode(), 200);
String responseBody2 = EntityUtils.toString(response2.getEntity());
assertTrue(responseBody2.contains("\"name\":\"analytics3\""));
assertTrue(responseBody2.contains("\"resiliency_mode\":\"enforced\""));
assertTrue(responseBody2.contains("\"cpu\":0.4"));
assertTrue(responseBody2.contains("\"memory\":0.2"));

Request request3 = new Request("GET", "_wlm/query_group/analytics97");
assertThrows(ResponseException.class, () -> client().performRequest(request3));

client().performRequest(new Request("DELETE", "_wlm/query_group/analytics3"));
}

public void testDelete() throws Exception {
Request request = new Request("PUT", "_wlm/query_group");
request.setJsonEntity("{\n" +
" \"name\": \"analytics4\",\n" +
" \"resiliency_mode\": \"enforced\",\n" +
" \"resource_limits\": {\n" +
" \"cpu\" : 0.4,\n" +
" \"memory\" : 0.2\n" +
" }\n" +
"}'");
Response response = client().performRequest(request);
assertEquals(response.getStatusLine().getStatusCode(), 200);

Request request2 = new Request("DELETE", "_wlm/query_group/analytics4");
Response response2 = client().performRequest(request2);
assertEquals(response2.getStatusLine().getStatusCode(), 200);
String responseBody2 = EntityUtils.toString(response2.getEntity());
assertTrue(responseBody2.contains("\"acknowledged\":true"));

Request request3 = new Request("DELETE", "_wlm/query_group/analytics99");
assertThrows(ResponseException.class, () -> client().performRequest(request3));
}

public void testUpdate() throws Exception {
Request request = new Request("PUT", "_wlm/query_group");
request.setJsonEntity("{\n" +
" \"name\": \"analytics5\",\n" +
" \"resiliency_mode\": \"enforced\",\n" +
" \"resource_limits\": {\n" +
" \"cpu\" : 0.4,\n" +
" \"memory\" : 0.2\n" +
" }\n" +
"}'");
Response response = client().performRequest(request);
assertEquals(response.getStatusLine().getStatusCode(), 200);

Request request2 = new Request("PUT", "_wlm/query_group/analytics5");
request2.setJsonEntity("{\n" +
" \"resiliency_mode\": \"monitor\",\n" +
" \"resource_limits\": {\n" +
" \"cpu\" : 0.41,\n" +
" \"memory\" : 0.21\n" +
" }\n" +
"}'");
Response response2 = client().performRequest(request2);
assertEquals(response2.getStatusLine().getStatusCode(), 200);
String responseBody2 = EntityUtils.toString(response2.getEntity());
assertTrue(responseBody2.contains("\"name\":\"analytics5\""));
assertTrue(responseBody2.contains("\"resiliency_mode\":\"monitor\""));
assertTrue(responseBody2.contains("\"cpu\":0.41"));
assertTrue(responseBody2.contains("\"memory\":0.21"));

Request request3 = new Request("PUT", "_wlm/query_group/analytics5");
request3.setJsonEntity("{\n" +
" \"resource_limits\": {\n" +
" \"cpu\" : 1.1,\n" +
" \"memory\" : -0.1\n" +
" }\n" +
"}'");
assertThrows(ResponseException.class, () -> client().performRequest(request3));

Request request4 = new Request("PUT", "_wlm/query_group/analytics98");
request4.setJsonEntity("{\n" +
" \"resiliency_mode\": \"monitor\",\n" +
" \"resource_limits\": {\n" +
" \"cpu\" : 0.43,\n" +
" \"memory\" : 0.23\n" +
" }\n" +
"}'");
assertThrows(ResponseException.class, () -> client().performRequest(request4));
client().performRequest(new Request("DELETE", "_wlm/query_group/analytics5"));
}

public void testCRUD() throws Exception {
Request request = new Request("PUT", "_wlm/query_group");
request.setJsonEntity("{\n" +
" \"name\": \"analytics6\",\n" +
" \"resiliency_mode\": \"enforced\",\n" +
" \"resource_limits\": {\n" +
" \"cpu\" : 0.4,\n" +
" \"memory\" : 0.2\n" +
" }\n" +
"}'");
Response response = client().performRequest(request);
assertEquals(response.getStatusLine().getStatusCode(), 200);

Request request1 = new Request("PUT", "_wlm/query_group/analytics6");
request1.setJsonEntity("{\n" +
" \"resiliency_mode\": \"monitor\",\n" +
" \"resource_limits\": {\n" +
" \"cpu\" : 0.41,\n" +
" \"memory\" : 0.21\n" +
" }\n" +
"}'");
Response response1 = client().performRequest(request1);
assertEquals(response1.getStatusLine().getStatusCode(), 200);

Request request2 = new Request("GET", "_wlm/query_group/analytics6");
Response response2 = client().performRequest(request2);
assertEquals(response2.getStatusLine().getStatusCode(), 200);
String responseBody2 = EntityUtils.toString(response2.getEntity());
assertTrue(responseBody2.contains("\"name\":\"analytics6\""));
assertTrue(responseBody2.contains("\"resiliency_mode\":\"monitor\""));
assertTrue(responseBody2.contains("\"cpu\":0.41"));
assertTrue(responseBody2.contains("\"memory\":0.21"));

Request request3 = new Request("PUT", "_wlm/query_group");
request3.setJsonEntity("{\n" +
" \"name\": \"users3\",\n" +
" \"resiliency_mode\": \"monitor\",\n" +
" \"resource_limits\": {\n" +
" \"cpu\" : 0.6,\n" +
" \"memory\" : 0.8\n" +
" }\n" +
"}'");
assertThrows(ResponseException.class, () -> client().performRequest(request3));

Request request4 = new Request("PUT", "_wlm/query_group");
request4.setJsonEntity("{\n" +
" \"name\": \"users3\",\n" +
" \"resiliency_mode\": \"monitor\",\n" +
" \"resource_limits\": {\n" +
" \"cpu\" : 0.59,\n" +
" \"memory\" : 0.79\n" +
" }\n" +
"}'");
Response response4 = client().performRequest(request4);
assertEquals(response4.getStatusLine().getStatusCode(), 200);

Request request5 = new Request("DELETE", "_wlm/query_group/analytics6");
Response response5 = client().performRequest(request5);
assertEquals(response5.getStatusLine().getStatusCode(), 200);
String responseBody5 = EntityUtils.toString(response5.getEntity());
assertTrue(responseBody5.contains("\"acknowledged\":true"));

Request request6 = new Request("GET", "_wlm/query_group/");
Response response6 = client().performRequest(request6);
assertEquals(response6.getStatusLine().getStatusCode(), 200);
String responseBody6 = EntityUtils.toString(response6.getEntity());
assertTrue(responseBody6.contains("\"query_groups\""));
assertTrue(responseBody6.contains("\"users3\""));
assertFalse(responseBody6.contains("\"analytics6\""));
client().performRequest(new Request("DELETE", "_wlm/query_group/users3"));
}
}

0 comments on commit 7f88472

Please sign in to comment.