honeymoose
/
OpenSearch
mirror of https://github.com/honeymoose/OpenSearch.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

[TEST] be more specific around http method used for sniffing

This commit is contained in:
javanna 2016-05-30 10:07:34 +02:00 committed by Luca Cavanna
parent 044a97c740
commit 3745305ffb
1 changed files with 13 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
client-sniffer/src/test/java/org/elasticsearch/client/sniff/HostsSnifferTests.java
View File

@ -29,6 +29,7 @@ import okhttp3.mockwebserver.MockResponse;
import okhttp3.mockwebserver.MockWebServer;
import okhttp3.mockwebserver.RecordedRequest;
import org.apache.http.HttpHost;
import org.apache.http.client.methods.HttpGet;
import org.apache.lucene.util.LuceneTestCase;
import org.elasticsearch.client.ElasticsearchResponse;
import org.elasticsearch.client.ElasticsearchResponseException;
@ -123,18 +124,19 @@ public class HostsSnifferTests extends LuceneTestCase {
final Dispatcher dispatcher = new Dispatcher() {
@Override
public MockResponse dispatch(RecordedRequest request) throws InterruptedException {
String decodedUrl;
try {
decodedUrl = URLDecoder.decode(request.getPath(), StandardCharsets.UTF_8.name());
} catch (UnsupportedEncodingException e) {
throw new RuntimeException(e);
}
String sniffUrl = "/_nodes/http?timeout=" + sniffTimeout + "ms";
if (sniffUrl.equals(decodedUrl)) {
return new MockResponse().setBody(sniffResponse.nodesInfoBody).setResponseCode(sniffResponse.nodesInfoResponseCode);
} else {
return new MockResponse().setResponseCode(404);
if (request.getMethod().equals(HttpGet.METHOD_NAME)) {
String decodedUrl;
try {
decodedUrl = URLDecoder.decode(request.getPath(), StandardCharsets.UTF_8.name());
} catch (UnsupportedEncodingException e) {
throw new RuntimeException(e);
}
String sniffUrl = "/_nodes/http?timeout=" + sniffTimeout + "ms";
if (sniffUrl.equals(decodedUrl)) {
return new MockResponse().setBody(sniffResponse.nodesInfoBody).setResponseCode(sniffResponse.nodesInfoResponseCode);
}
}
return new MockResponse().setResponseCode(404);
}
};
server.setDispatcher(dispatcher);