D-Net
/
AriadnePlus
4
0
Fork 0
Code Issues Pull Requests 1 Releases Wiki Activity

test for new indexing methods on rest module

This commit is contained in:
Enrico Ottonello 2020-10-20 16:42:44 +02:00
parent 240be91c5e
commit 1398c2495e
1 changed files with 177 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

177
dnet-ariadneplus/src/test/java/eu/dnetlib/ariadneplus/workflows/nodes/RestClientTest.java Normal file
View File

@ -0,0 +1,177 @@
package eu.dnetlib.ariadneplus.workflows.nodes;
import org.apache.commons.io.IOUtils;
import org.apache.http.HttpEntity;
import org.apache.http.HttpResponse;
import org.apache.http.client.HttpClient;
import org.apache.http.client.methods.HttpGet;
import org.apache.http.client.methods.HttpPost;
import org.apache.http.client.utils.URIBuilder;
import org.apache.http.impl.client.HttpClients;
import org.apache.http.util.EntityUtils;
import org.junit.Ignore;
import org.junit.Test;
import java.io.InputStream;
import java.net.URI;
import java.util.Arrays;
import java.util.List;
public class RestClientTest {
// @Test
@Ignore
public void getVersion() throws Exception {
int statusCode = -1;
String response = "";
String endpoint = "https://aggregator.ariadne.d4science.org/ariadneplus-graphdb/version";
System.out.println("getVersion endpoint: " + endpoint);
HttpClient client = null;
try {
URI getURI = new URIBuilder(endpoint)
.build();
client = HttpClients.createDefault();
HttpResponse res = client.execute(new HttpGet(getURI));
HttpEntity entity = res.getEntity();
String version = EntityUtils.toString(entity);
if (res.getStatusLine()!=null) {
statusCode = res.getStatusLine().getStatusCode();
}
System.out.println("statusCode: "+statusCode);
System.out.println("version: "+version);
}
catch (Throwable t) {
t.printStackTrace();
}
}
// @Test
@Ignore
public void selectIdentifiersTest() throws Exception {
String datasource = "ads";
String collectionId = "398";
String resourceType = "";
resourceType = "COLLECTION";
selectIdentifiersTest(datasource, collectionId, resourceType);
resourceType = "RECORD";
selectIdentifiersTest(datasource, collectionId, resourceType);
}
private List<String> selectIdentifiersTest(String datasource, String collectionId, String resourceType) throws Exception {
int statusCode = -1;
String response = "";
List<String> identifiers = null;
String endpoint = "https://aggregator.ariadne.d4science.org/ariadneplus-graphdb/selectIdentifiers";
System.out.println("selectCollectionIdentifiersTest endpoint: " + endpoint);
HttpClient client = null;
try {
URI getURI = new URIBuilder(endpoint)
.addParameter("datasource", datasource)
.addParameter("collectionId", collectionId)
.addParameter("resourceType", resourceType)
.build();
client = HttpClients.createDefault();
HttpResponse res = client.execute(new HttpGet(getURI));
if (res.getStatusLine()!=null) {
statusCode = res.getStatusLine().getStatusCode();
System.out.println("statusCode: "+statusCode);
}
HttpEntity entity = res.getEntity();
String content = EntityUtils.toString(entity);
String[] identifiersStr = content.split(",");
System.out.println("first decoded: "+identifiersStr[0]);
identifiers = Arrays.asList(identifiersStr);
System.out.println("retrieved "+identifiers.size()+ " identifiers");
if (!identifiers.isEmpty()) {
System.out.println("first: " + identifiers.get(0));
}
}
catch (Throwable t) {
t.printStackTrace();
}
return identifiers;
}
private String indexingTest(String datasource, String collectionId, String resourceType, String identifier) throws Exception {
int statusCode = -1;
String response = "";
String result = "";
String endpoint = "https://aggregator.ariadne.d4science.org/ariadneplus-graphdb/indexOnESByIdentifier";
System.out.println("indexingTest endpoint: " + endpoint + " datasource: "+datasource+ " collectionId: "+collectionId+ " identifier: "+identifier);
HttpClient client = null;
try {
URI postURI = new URIBuilder(endpoint)
.addParameter("datasource", datasource)
.addParameter("collectionId", collectionId)
.addParameter("resourceType", resourceType)
.addParameter("identifier", identifier)
.build();
client = HttpClients.createDefault();
HttpResponse res = client.execute(new HttpPost(postURI));
if (res.getStatusLine()!=null) {
statusCode = res.getStatusLine().getStatusCode();
System.out.println("statusCode: "+statusCode);
}
HttpEntity entity = res.getEntity();
result = EntityUtils.toString(entity);
System.out.println("result: " + result);
}
catch (Throwable t) {
t.printStackTrace();
}
return result;
}
// @Test
@Ignore
public void indexOnESByIdentifierTest() throws Exception {
String datasource = "ads";
String collectionId = "801";
String resourceType = "";
resourceType = "COLLECTION";
List<String> collectionIdentifiers = selectIdentifiersTest(datasource, collectionId, resourceType);
if (!collectionIdentifiers.isEmpty()) {
indexingTest(datasource, collectionId, resourceType, cleanIdentifier(collectionIdentifiers.get(0)));
}
resourceType = "RECORD";
List<String> recordIdentifiers = selectIdentifiersTest(datasource, collectionId, resourceType);
if (!recordIdentifiers.isEmpty()) {
indexingTest(datasource, collectionId, resourceType, cleanIdentifier(recordIdentifiers.get(0)));
}
}
// @Test
@Ignore
public void selectIdentifierTest() throws Exception {
String datasource = "ads";
String collectionId = "801";
String resourceType = "";
resourceType = "COLLECTION";
List<String> identifiers = selectIdentifiersTest(datasource, collectionId, resourceType);
System.out.println("identifiers: " + cleanIdentifier(identifiers.get(0)));
}
private String cleanIdentifier(String identifier) {
System.out.println("cleaning: "+identifier);
String cleaned = identifier;
try {
cleaned = identifier
.replace("[", "")
.replace("]", "")
.replace("\"", "");
}
catch (Exception e) {
}
return cleaned;
}
}