@@ -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