Skip to content

Commit 0382984

Browse files
committed
Switch non-x-pack to new style requests (#32106)
In #29623 we added `Request` object flavored requests to the low level REST client and in #30315 we deprecated the old `performRequest`s. This changes most of the calls not in X-Pack to their new versions.
1 parent b00de36 commit 0382984

File tree

4 files changed

+39
-33
lines changed

4 files changed

+39
-33
lines changed

modules/transport-netty4/src/test/java/org/elasticsearch/rest/Netty4BadRequestIT.java

+11-8
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,8 @@
1919

2020
package org.elasticsearch.rest;
2121

22-
import org.apache.http.message.BasicHeader;
22+
import org.elasticsearch.client.Request;
23+
import org.elasticsearch.client.RequestOptions;
2324
import org.elasticsearch.client.Response;
2425
import org.elasticsearch.client.ResponseException;
2526
import org.elasticsearch.common.settings.Setting;
@@ -43,7 +44,7 @@
4344
public class Netty4BadRequestIT extends ESRestTestCase {
4445

4546
public void testBadRequest() throws IOException {
46-
final Response response = client().performRequest("GET", "/_nodes/settings", Collections.emptyMap());
47+
final Response response = client().performRequest(new Request("GET", "/_nodes/settings"));
4748
final ObjectPath objectPath = ObjectPath.createFromResponse(response);
4849
final Map<String, Object> map = objectPath.evaluate("nodes");
4950
int maxMaxInitialLineLength = Integer.MIN_VALUE;
@@ -77,9 +78,9 @@ public void testBadRequest() throws IOException {
7778
}
7879

7980
public void testInvalidParameterValue() throws IOException {
80-
final ResponseException e = expectThrows(
81-
ResponseException.class,
82-
() -> client().performRequest("GET", "/_cluster/settings", Collections.singletonMap("pretty", "neither-true-nor-false")));
81+
final Request request = new Request("GET", "/_cluster/settings");
82+
request.addParameter("pretty", "neither-true-nor-false");
83+
final ResponseException e = expectThrows(ResponseException.class, () -> client().performRequest(request));
8384
final Response response = e.getResponse();
8485
assertThat(response.getStatusLine().getStatusCode(), equalTo(400));
8586
final ObjectPath objectPath = ObjectPath.createFromResponse(response);
@@ -89,9 +90,11 @@ public void testInvalidParameterValue() throws IOException {
8990
}
9091

9192
public void testInvalidHeaderValue() throws IOException {
92-
final BasicHeader header = new BasicHeader("Content-Type", "\t");
93-
final ResponseException e =
94-
expectThrows(ResponseException.class, () -> client().performRequest("GET", "/_cluster/settings", header));
93+
final Request request = new Request("GET", "/_cluster/settings");
94+
final RequestOptions.Builder options = request.getOptions().toBuilder();
95+
options.addHeader("Content-Type", "\t");
96+
request.setOptions(options);
97+
final ResponseException e = expectThrows(ResponseException.class, () -> client().performRequest(request));
9598
final Response response = e.getResponse();
9699
assertThat(response.getStatusLine().getStatusCode(), equalTo(400));
97100
final ObjectPath objectPath = ObjectPath.createFromResponse(response);

modules/transport-netty4/src/test/java/org/elasticsearch/rest/Netty4HeadBodyIsEmptyIT.java

+19-11
Original file line numberDiff line numberDiff line change
@@ -19,8 +19,7 @@
1919

2020
package org.elasticsearch.rest;
2121

22-
import org.apache.http.entity.ContentType;
23-
import org.apache.http.entity.StringEntity;
22+
import org.elasticsearch.client.Request;
2423
import org.elasticsearch.client.Response;
2524
import org.elasticsearch.common.Strings;
2625
import org.elasticsearch.common.xcontent.XContentBuilder;
@@ -57,8 +56,9 @@ private void createTestDoc(final String indexName, final String typeName) throws
5756
builder.field("test", "test");
5857
}
5958
builder.endObject();
60-
client().performRequest("PUT", "/" + indexName + "/" + typeName + "/" + "1", emptyMap(),
61-
new StringEntity(Strings.toString(builder), ContentType.APPLICATION_JSON));
59+
Request request = new Request("PUT", "/" + indexName + "/" + typeName + "/" + "1");
60+
request.setJsonEntity(Strings.toString(builder));
61+
client().performRequest(request);
6262
}
6363
}
6464

@@ -109,8 +109,9 @@ public void testAliasExists() throws IOException {
109109
}
110110
builder.endObject();
111111

112-
client().performRequest("POST", "_aliases", emptyMap(), new StringEntity(Strings.toString(builder),
113-
ContentType.APPLICATION_JSON));
112+
Request request = new Request("POST", "/_aliases");
113+
request.setJsonEntity(Strings.toString(builder));
114+
client().performRequest(request);
114115
headTestCase("/_alias/test_alias", emptyMap(), greaterThan(0));
115116
headTestCase("/test/_alias/test_alias", emptyMap(), greaterThan(0));
116117
}
@@ -135,8 +136,9 @@ public void testTemplateExists() throws IOException {
135136
}
136137
builder.endObject();
137138

138-
client().performRequest("PUT", "/_template/template", emptyMap(),
139-
new StringEntity(Strings.toString(builder), ContentType.APPLICATION_JSON));
139+
Request request = new Request("PUT", "/_template/template");
140+
request.setJsonEntity(Strings.toString(builder));
141+
client().performRequest(request);
140142
headTestCase("/_template/template", emptyMap(), greaterThan(0));
141143
}
142144
}
@@ -164,8 +166,10 @@ public void testGetSourceAction() throws IOException {
164166
builder.endObject();
165167
}
166168
builder.endObject();
167-
client().performRequest("PUT", "/test-no-source", emptyMap(), new StringEntity(Strings.toString(builder),
168-
ContentType.APPLICATION_JSON));
169+
170+
Request request = new Request("PUT", "/test-no-source");
171+
request.setJsonEntity(Strings.toString(builder));
172+
client().performRequest(request);
169173
createTestDoc("test-no-source", "test-no-source");
170174
headTestCase("/test-no-source/test-no-source/1/_source", emptyMap(), NOT_FOUND.getStatus(), equalTo(0));
171175
}
@@ -190,7 +194,11 @@ private void headTestCase(
190194
final Map<String, String> params,
191195
final int expectedStatusCode,
192196
final Matcher<Integer> matcher) throws IOException {
193-
Response response = client().performRequest("HEAD", url, params);
197+
Request request = new Request("HEAD", url);
198+
for (Map.Entry<String, String> param : params.entrySet()) {
199+
request.addParameter(param.getKey(), param.getValue());
200+
}
201+
Response response = client().performRequest(request);
194202
assertEquals(expectedStatusCode, response.getStatusLine().getStatusCode());
195203
assertThat(Integer.valueOf(response.getHeader("Content-Length")), matcher);
196204
assertNull("HEAD requests shouldn't have a response body but " + url + " did", response.getEntity());

plugins/repository-hdfs/src/test/java/org/elasticsearch/repositories/hdfs/HaHdfsFailoverTestSuiteIT.java

+7-13
Original file line numberDiff line numberDiff line change
@@ -21,16 +21,13 @@
2121

2222
import java.io.IOException;
2323
import java.net.InetSocketAddress;
24-
import java.nio.charset.Charset;
2524
import java.nio.file.Files;
2625
import java.nio.file.Path;
2726
import java.security.AccessController;
2827
import java.security.PrivilegedActionException;
2928
import java.security.PrivilegedExceptionAction;
3029
import java.util.ArrayList;
31-
import java.util.Collections;
3230
import java.util.List;
33-
import java.util.Map;
3431

3532
import org.apache.hadoop.conf.Configuration;
3633
import org.apache.hadoop.ha.BadFencingConfigurationException;
@@ -42,9 +39,7 @@
4239
import org.apache.hadoop.hdfs.tools.DFSHAAdmin;
4340
import org.apache.hadoop.security.SecurityUtil;
4441
import org.apache.hadoop.security.UserGroupInformation;
45-
import org.apache.http.Header;
46-
import org.apache.http.message.BasicHeader;
47-
import org.apache.http.nio.entity.NStringEntity;
42+
import org.elasticsearch.client.Request;
4843
import org.elasticsearch.client.Response;
4944
import org.elasticsearch.client.RestClient;
5045
import org.elasticsearch.common.io.PathUtils;
@@ -58,8 +53,6 @@ public class HaHdfsFailoverTestSuiteIT extends ESRestTestCase {
5853

5954
public void testHAFailoverWithRepository() throws Exception {
6055
RestClient client = client();
61-
Map<String, String> emptyParams = Collections.emptyMap();
62-
Header contentHeader = new BasicHeader("Content-Type", "application/json");
6356

6457
String esKerberosPrincipal = System.getProperty("test.krb5.principal.es");
6558
String hdfsKerberosPrincipal = System.getProperty("test.krb5.principal.hdfs");
@@ -106,7 +99,8 @@ public void testHAFailoverWithRepository() throws Exception {
10699

107100
// Create repository
108101
{
109-
Response response = client.performRequest("PUT", "/_snapshot/hdfs_ha_repo_read", emptyParams, new NStringEntity(
102+
Request request = new Request("PUT", "/_snapshot/hdfs_ha_repo_read");
103+
request.setJsonEntity(
110104
"{" +
111105
"\"type\":\"hdfs\"," +
112106
"\"settings\":{" +
@@ -121,15 +115,15 @@ public void testHAFailoverWithRepository() throws Exception {
121115
"\"conf.dfs.client.failover.proxy.provider.ha-hdfs\": " +
122116
"\"org.apache.hadoop.hdfs.server.namenode.ha.ConfiguredFailoverProxyProvider\"" +
123117
"}" +
124-
"}",
125-
Charset.defaultCharset()), contentHeader);
118+
"}");
119+
Response response = client.performRequest(request);
126120

127121
Assert.assertEquals(200, response.getStatusLine().getStatusCode());
128122
}
129123

130124
// Get repository
131125
{
132-
Response response = client.performRequest("GET", "/_snapshot/hdfs_ha_repo_read/_all", emptyParams);
126+
Response response = client.performRequest(new Request("GET", "/_snapshot/hdfs_ha_repo_read/_all"));
133127
Assert.assertEquals(200, response.getStatusLine().getStatusCode());
134128
}
135129

@@ -138,7 +132,7 @@ public void testHAFailoverWithRepository() throws Exception {
138132

139133
// Get repository again
140134
{
141-
Response response = client.performRequest("GET", "/_snapshot/hdfs_ha_repo_read/_all", emptyParams);
135+
Response response = client.performRequest(new Request("GET", "/_snapshot/hdfs_ha_repo_read/_all"));
142136
Assert.assertEquals(200, response.getStatusLine().getStatusCode());
143137
}
144138
}

qa/verify-version-constants/src/test/java/org/elasticsearch/qa/verify_version_constants/VerifyVersionConstantsIT.java

+2-1
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,7 @@
2020
package org.elasticsearch.qa.verify_version_constants;
2121

2222
import org.elasticsearch.Version;
23+
import org.elasticsearch.client.Request;
2324
import org.elasticsearch.client.Response;
2425
import org.elasticsearch.test.rest.ESRestTestCase;
2526
import org.elasticsearch.test.rest.yaml.ObjectPath;
@@ -32,7 +33,7 @@
3233
public class VerifyVersionConstantsIT extends ESRestTestCase {
3334

3435
public void testLuceneVersionConstant() throws IOException, ParseException {
35-
final Response response = client().performRequest("GET", "/");
36+
final Response response = client().performRequest(new Request("GET", "/"));
3637
assertThat(response.getStatusLine().getStatusCode(), equalTo(200));
3738
final ObjectPath objectPath = ObjectPath.createFromResponse(response);
3839
final String elasticsearchVersionString = objectPath.evaluate("version.number").toString();

0 commit comments

Comments
 (0)