Skip to content

Commit 8dceadf

Browse files
committed
get rid of dead if branch check v2 and post function
1 parent 20b170d commit 8dceadf

1 file changed

Lines changed: 30 additions & 52 deletions

File tree

src/test/java/com/uid2/operator/UIDOperatorVerticleTest.java

Lines changed: 30 additions & 52 deletions
Original file line numberDiff line numberDiff line change
@@ -249,8 +249,8 @@ private void send(String apiVersion, Vertx vertx, String endpoint, JsonObject po
249249
}
250250

251251
private void send(String apiVersion, Vertx vertx, String endpoint, JsonObject postPayload, int expectedHttpCode, Handler<JsonObject> handler, Map<String, String> additionalHeaders) {
252-
if (apiVersion.equals("v2")) {
253-
ClientKey ck = (ClientKey) clientKeyProvider.get("");
252+
assertEquals("v2", apiVersion);
253+
ClientKey ck = (ClientKey) clientKeyProvider.get("");
254254

255255
long nonce = new BigInteger(Random.getBytes(8)).longValue();
256256

@@ -276,13 +276,6 @@ private void send(String apiVersion, Vertx vertx, String endpoint, JsonObject po
276276
handler.handle(tryParseResponse(ar.result()));
277277
}
278278
}, additionalHeaders);
279-
} else {
280-
post(vertx, endpoint, postPayload, ar -> {
281-
assertTrue(ar.succeeded());
282-
assertEquals(expectedHttpCode, ar.result().statusCode());
283-
handler.handle(tryParseResponse(ar.result()));
284-
}, additionalHeaders);
285-
}
286279
}
287280

288281
protected void sendTokenGenerate(String apiVersion, Vertx vertx, JsonObject v2PostPayload, int expectedHttpCode,
@@ -349,37 +342,36 @@ private void sendTokenRefresh(String apiVersion, Vertx vertx, VertxTestContext t
349342

350343
private void sendTokenRefresh(String apiVersion, Vertx vertx, VertxTestContext testContext, String refreshToken, String v2RefreshDecryptSecret, int expectedHttpCode,
351344
Handler<JsonObject> handler, Map<String, String> additionalHeaders) {
352-
if (apiVersion.equals("v2")) {
353-
WebClient client = WebClient.create(vertx);
354-
HttpRequest<Buffer> refreshHttpRequest = client.postAbs(getUrlForEndpoint("v2/token/refresh"));
355-
refreshHttpRequest.putHeader(HttpHeaders.CONTENT_TYPE.toString(), HttpMediaType.TEXT_PLAIN.getType());
356-
for (Map.Entry<String, String> entry : additionalHeaders.entrySet()) {
357-
refreshHttpRequest.putHeader(entry.getKey(), entry.getValue());
358-
}
345+
WebClient client = WebClient.create(vertx);
346+
HttpRequest<Buffer> refreshHttpRequest = client.postAbs(getUrlForEndpoint("v2/token/refresh"));
347+
refreshHttpRequest.putHeader(HttpHeaders.CONTENT_TYPE.toString(), HttpMediaType.TEXT_PLAIN.getType());
348+
for (Map.Entry<String, String> entry : additionalHeaders.entrySet()) {
349+
refreshHttpRequest.putHeader(entry.getKey(), entry.getValue());
350+
}
351+
352+
refreshHttpRequest
353+
.sendBuffer(Buffer.buffer(refreshToken.getBytes(StandardCharsets.UTF_8)), testContext.succeeding(response -> testContext.verify(() -> {
354+
assertEquals(expectedHttpCode, response.statusCode());
359355

360-
refreshHttpRequest
361-
.sendBuffer(Buffer.buffer(refreshToken.getBytes(StandardCharsets.UTF_8)), testContext.succeeding(response -> testContext.verify(() -> {
362-
assertEquals(expectedHttpCode, response.statusCode());
363-
364-
if (response.statusCode() == 200 && v2RefreshDecryptSecret != null) {
365-
byte[] byteResp = new byte[0];
366-
if (response.headers().contains(HttpHeaders.CONTENT_TYPE, HttpMediaType.APPLICATION_OCTET_STREAM.getType(), true)) {
367-
byteResp = response.bodyAsBuffer().getBytes();
368-
} else if (response.headers().contains(HttpHeaders.CONTENT_TYPE, HttpMediaType.TEXT_PLAIN.getType(), true)) {
369-
byteResp = Utils.decodeBase64String(response.bodyAsString());
370-
}
371-
byte[] decrypted = AesGcm.decrypt(byteResp, 0, Utils.decodeBase64String(v2RefreshDecryptSecret));
372-
JsonObject respJson = new JsonObject(new String(decrypted, StandardCharsets.UTF_8));
373-
374-
if (respJson.getString("status").equals("success"))
375-
decodeV2RefreshToken(respJson);
376-
377-
handler.handle(respJson);
378-
} else {
379-
handler.handle(tryParseResponse(response));
356+
if (response.statusCode() == 200 && v2RefreshDecryptSecret != null) {
357+
byte[] byteResp = new byte[0];
358+
if (response.headers().contains(HttpHeaders.CONTENT_TYPE, HttpMediaType.APPLICATION_OCTET_STREAM.getType(), true)) {
359+
byteResp = response.bodyAsBuffer().getBytes();
360+
} else if (response.headers().contains(HttpHeaders.CONTENT_TYPE, HttpMediaType.TEXT_PLAIN.getType(), true)) {
361+
byteResp = Utils.decodeBase64String(response.bodyAsString());
380362
}
381-
})));
382-
}
363+
byte[] decrypted = AesGcm.decrypt(byteResp, 0, Utils.decodeBase64String(v2RefreshDecryptSecret));
364+
JsonObject respJson = new JsonObject(new String(decrypted, StandardCharsets.UTF_8));
365+
366+
if (respJson.getString("status").equals("success"))
367+
decodeV2RefreshToken(respJson);
368+
369+
handler.handle(respJson);
370+
} else {
371+
handler.handle(tryParseResponse(response));
372+
}
373+
})));
374+
383375
}
384376

385377
private String decodeV2RefreshToken(JsonObject respJson) {
@@ -409,20 +401,6 @@ private JsonObject tryParseResponse(HttpResponse<Buffer> resp) {
409401
}
410402
}
411403

412-
private void post(Vertx vertx, String endpoint, JsonObject body, Handler<AsyncResult<HttpResponse<Buffer>>> handler, Map<String, String> additionalHeaders) {
413-
WebClient client = WebClient.create(vertx);
414-
ClientKey ck = clientKeyProvider.getClientKey("");
415-
HttpRequest<Buffer> req = client.postAbs(getUrlForEndpoint(endpoint));
416-
if (ck != null)
417-
req.putHeader("Authorization", "Bearer " + clientKey);
418-
419-
for (Map.Entry<String, String> entry : additionalHeaders.entrySet()) {
420-
req.putHeader(entry.getKey(), entry.getValue());
421-
}
422-
423-
req.sendJsonObject(body, handler);
424-
}
425-
426404
private void postV2(ClientKey ck, Vertx vertx, String endpoint, JsonObject body, long nonce, String referer, Handler<AsyncResult<HttpResponse<Buffer>>> handler) {
427405
postV2(ck, vertx, endpoint, body, nonce, referer, handler, Collections.emptyMap(), false);
428406
}

0 commit comments

Comments
 (0)