public boolean delete(String token, String url, String resourcePath) {
WebTarget target = client.target("https://" + url).path(resourcePath);
Invocation.Builder invocationBuilder = target.request(MediaType.APPLICATION_JSON_TYPE);
setHeaders(invocationBuilder, token);
Response response = invocationBuilder.delete();
return response.getStatus() == Response.Status.OK.getStatusCode();
}
public <T> T putWor(
String url,
String resourcePath,
Object object,
Class<T> responseClass,
Map<String, Object> queryParams,
String worToken) {
WebTarget target = client.target("https://" + url).path(resourcePath);
if (queryParams != null) {
for (Map.Entry<String, Object> entry : queryParams.entrySet()) {
target = target.queryParam(entry.getKey(), entry.getValue());
}
}
Invocation.Builder invocationBuilder = target.request(MediaType.APPLICATION_JSON_TYPE);
setHeaders(invocationBuilder, worToken);
Response putResponse =
invocationBuilder.put(Entity.entity(object, MediaType.APPLICATION_JSON_TYPE));
if (putResponse.getStatus() != Response.Status.OK.getStatusCode()) {
Logger.error(
"PUT call to "
+ url
+ "/"
+ resourcePath
+ " returned status of "
+ putResponse.getStatus());
return null;
}
if (responseClass != null
&& putResponse.hasEntity()
&& putResponse.getStatus() == Response.Status.OK.getStatusCode())
return putResponse.readEntity(responseClass);
return null;
}
public Post fetch(Session session, NextRequest next) {
LOGGER.debug(
((next == null) ? "Initial request" : "next request: " + next)
+ "/n Session: "
+ session.toString());
WebTarget target = createInitialFetchTarget();
if (next != null) {
target = fillNextParameters(next, target);
}
Invocation.Builder request = target.request();
request = appendCookies(session, request);
Response response =
request.post(
Entity.entity(
new NextToken.NextTokenBuilder().createNextToken(),
MediaType.APPLICATION_JSON_TYPE));
response.bufferEntity();
try {
String responseJson = response.readEntity(String.class);
LOGGER.debug("Full response: " + responseJson);
} catch (Exception ex) {
ex.printStackTrace();
}
Post post = response.readEntity(Post.class);
response.close();
return post;
}
public PaymentNotification[] getNotifications(String userId) {
WebTarget notificationTarget = appTarget.path("notification/user/" + userId);
Invocation.Builder setupBuilder = notificationTarget.request(MediaType.APPLICATION_JSON);
Response response = setupBuilder.get();
Gson gson = new GsonBuilder().setDateFormat("yyyy-MM-dd HH:mm:ss.SSS").create();
String json = response.readEntity(String.class);
return gson.fromJson(json, PaymentNotification[].class);
}
public Response postSequenceFile(
String url, String recordId, EntryType entryType, String sequence) {
WebTarget target = client.target("https://" + url).path("/rest/file/sequence");
Invocation.Builder invocationBuilder = target.request(MediaType.APPLICATION_JSON_TYPE);
final FormDataMultiPart multiPart = new FormDataMultiPart();
multiPart.field("file", IOUtils.toInputStream(sequence), MediaType.TEXT_PLAIN_TYPE);
multiPart.field("entryRecordId", recordId);
multiPart.field("entryType", entryType.name());
return invocationBuilder.post(Entity.entity(multiPart, MediaType.MULTIPART_FORM_DATA_TYPE));
}
@Override
public <T> T get(String url, String path, Class<T> clazz, Map<String, Object> queryParams) {
WebTarget target = client.target("https://" + url).path(path);
if (queryParams != null) {
for (Map.Entry<String, Object> entry : queryParams.entrySet()) {
target = target.queryParam(entry.getKey(), entry.getValue());
}
}
return target.request(MediaType.APPLICATION_JSON_TYPE).buildGet().invoke(clazz);
}
public Response sendMoney(MoneySend moneySend) {
Gson gson = new Gson();
String payload = gson.toJson(moneySend);
WebTarget moneyTarget = appTarget.path("money");
WebTarget sendMoneyTarget = moneyTarget.path("send");
return sendMoneyTarget
.request()
.accept(MediaType.APPLICATION_JSON)
.post(Entity.entity(payload, MediaType.APPLICATION_JSON), Response.class);
}
public Response collectMoney(MoneyCollect moneyCollect) {
Gson gson = new Gson();
String payload = gson.toJson(moneyCollect);
WebTarget moneyTarget = appTarget.path("money");
WebTarget collectMoneyTarget = moneyTarget.path("collect");
return collectMoneyTarget
.request()
.accept(MediaType.APPLICATION_JSON)
.post(Entity.entity(payload, MediaType.APPLICATION_JSON), Response.class);
}
// WOR
public <T> T getWor(
String url, String path, Class<T> clazz, Map<String, Object> queryParams, String token) {
WebTarget target = client.target("https://" + url).path(path);
if (queryParams != null) {
for (Map.Entry<String, Object> entry : queryParams.entrySet()) {
target = target.queryParam(entry.getKey(), entry.getValue());
}
}
Invocation.Builder invocationBuilder = target.request(MediaType.APPLICATION_JSON_TYPE);
setHeaders(invocationBuilder, token);
return invocationBuilder.buildGet().invoke(clazz);
}
public static String checkLogin(Userinfo ui) {
Client client = ClientBuilder.newClient(config);
WebTarget target = client.target(getBaseUri());
return target
.path("users")
.path("role=" + ui.getUsertype())
.path("email=" + ui.getUsername())
.path("password=" + ui.getPassword())
.request()
.accept(MediaType.TEXT_HTML)
.get(String.class);
}
@Test
public void createTransactionReturnsOk() {
// create "cars" transaction
Response response =
client
.path("/transactionservice/transaction/10")
.request()
.put(Entity.json(new Transaction("cars", new BigDecimal(5000), 0l)));
assertEquals(200, response.getStatus());
assertEquals(
TransactionService.OperationResult.OK,
response.readEntity(TransactionService.OperationResult.class));
// create "shopping" transaction
response =
client
.path("/transactionservice/transaction/11")
.request()
.put(Entity.json(new Transaction("shopping", new BigDecimal(10000), 10l)));
assertEquals(200, response.getStatus());
assertEquals(
TransactionService.OperationResult.OK,
response.readEntity(TransactionService.OperationResult.class));
// get "cars" transactions
response = client.path("/transactionservice/type/cars").request().get();
assertEquals(200, response.getStatus());
@SuppressWarnings("unchecked")
List<Integer> ids = response.readEntity(List.class);
assertEquals(1, ids.size());
assertEquals(10, ids.get(0).intValue());
// get "sum" for transaction 10
response = client.path("/transactionservice/sum/10").request().get();
assertEquals(200, response.getStatus());
AggregatorService.Sum sum = response.readEntity(AggregatorService.Sum.class);
assertEquals(15000, sum.getSum().intValue());
// get "sum" for transaction 11
response = client.path("/transactionservice/sum/11").request().get();
assertEquals(200, response.getStatus());
sum = response.readEntity(AggregatorService.Sum.class);
assertEquals(10000, sum.getSum().intValue());
}
private WebTarget fillNextParameters(NextRequest next, WebTarget target) {
target =
target
.queryParam("first-article-idx", next.getQueryParams().getFirstArticleIdx())
.queryParam("last-article-idx", next.getQueryParams().getLastArticleIdx())
.queryParam("last-feed-id", next.getQueryParams().getLastFeedID())
.queryParam("subpage", next.getQueryParams().isSubpage());
return target;
}
public Session login(final UserCredential userCredential) {
WebTarget target =
client
.target("http://auth.getprismatic.com")
.path("auth")
.path("login")
.queryParam("api-version", "1.2")
.queryParam("ignore", "true")
.queryParam("whitelist_url", "http%3A%2F%2Fgetprismatic.com%2Fnews%2Fhome")
.queryParam("soon_url", "http%3A%2F%2Fgetprismatic.com%2Fwelcome")
.queryParam("create_url", "http%3A%2F%2Fgetprismatic.com%2Fcreateaccount")
.queryParam("resetpassword_url", "http%3A%2F%2Fgetprismatic.com%2Fresetpassword");
Response response =
target.request().post(Entity.entity(userCredential, MediaType.APPLICATION_JSON_TYPE));
checkResponseStatus(response);
return Session.parseFromHeaders(response.getHeaders().get("Set-Cookie"));
}
@Override
public <T> T post(
String url,
String resourcePath,
Object object,
Class<T> responseClass,
Map<String, Object> queryParams) {
WebTarget target = client.target("https://" + url).path(resourcePath);
if (queryParams != null) {
for (Map.Entry<String, Object> entry : queryParams.entrySet()) {
target = target.queryParam(entry.getKey(), entry.getValue());
}
}
Invocation.Builder invocationBuilder = target.request(MediaType.APPLICATION_JSON_TYPE);
Response postResponse =
invocationBuilder.post(Entity.entity(object, MediaType.APPLICATION_JSON_TYPE));
if (postResponse.hasEntity() && postResponse.getStatus() == Response.Status.OK.getStatusCode())
return postResponse.readEntity(responseClass);
return null;
}
public <T> T put(String url, String resourcePath, Object object, Class<T> responseClass) {
WebTarget target = client.target("https://" + url).path(resourcePath);
Invocation.Builder invocationBuilder = target.request(MediaType.APPLICATION_JSON_TYPE);
Response putResponse =
invocationBuilder.put(Entity.entity(object, MediaType.APPLICATION_JSON_TYPE));
if (putResponse.getStatus() != Response.Status.OK.getStatusCode()) {
Logger.error(
"PUT call to "
+ url
+ "/"
+ resourcePath
+ " returned status of "
+ putResponse.getStatus());
return null;
}
if (responseClass != null
&& putResponse.hasEntity()
&& putResponse.getStatus() == Response.Status.OK.getStatusCode())
return putResponse.readEntity(responseClass);
return null;
}
public Response backdoorTeardown() {
WebTarget backdoorTarget = appTarget.path("backdoor");
WebTarget setupTarget = backdoorTarget.path("teardown");
Invocation.Builder setupBuilder = setupTarget.request(MediaType.TEXT_PLAIN_TYPE);
return setupBuilder.get();
}
public Response put(String url, String resourcePath, Object object) {
WebTarget target = client.target("https://" + url).path(resourcePath);
Invocation.Builder invocationBuilder = target.request(MediaType.APPLICATION_JSON_TYPE);
return invocationBuilder.put(Entity.entity(object, MediaType.APPLICATION_JSON_TYPE));
}
