2727import java .util .Map ;
2828
2929import io .netty .handler .codec .http .HttpResponseStatus ;
30+ import mockwebserver3 .RecordedRequest ;
3031import okhttp3 .HttpUrl ;
31- import okhttp3 .mockwebserver .RecordedRequest ;
3232import org .assertj .core .api .Assertions ;
3333import org .jetbrains .annotations .NotNull ;
3434import org .junit .jupiter .api .AfterEach ;
@@ -103,8 +103,8 @@ void databaseParameter() throws InterruptedException {
103103 assertThat (mockServer .getRequestCount ()).isEqualTo (1 );
104104 RecordedRequest request = mockServer .takeRequest ();
105105 assertThat (request ).isNotNull ();
106- assertThat (request .getRequestUrl ()).isNotNull ();
107- assertThat (request .getRequestUrl ().queryParameter ("bucket" )).isEqualTo ("my-database" );
106+ assertThat (request .getUrl ()).isNotNull ();
107+ assertThat (request .getUrl ().queryParameter ("bucket" )).isEqualTo ("my-database" );
108108 }
109109
110110 @ Test
@@ -116,8 +116,8 @@ void databaseParameterSpecified() throws InterruptedException {
116116 assertThat (mockServer .getRequestCount ()).isEqualTo (1 );
117117 RecordedRequest request = mockServer .takeRequest ();
118118 assertThat (request ).isNotNull ();
119- assertThat (request .getRequestUrl ()).isNotNull ();
120- assertThat (request .getRequestUrl ().queryParameter ("bucket" )).isEqualTo ("my-database-2" );
119+ assertThat (request .getUrl ()).isNotNull ();
120+ assertThat (request .getUrl ().queryParameter ("bucket" )).isEqualTo ("my-database-2" );
121121 }
122122
123123 @ Test
@@ -144,8 +144,8 @@ void precisionParameter() throws InterruptedException {
144144 assertThat (mockServer .getRequestCount ()).isEqualTo (1 );
145145 RecordedRequest request = mockServer .takeRequest ();
146146 assertThat (request ).isNotNull ();
147- assertThat (request .getRequestUrl ()).isNotNull ();
148- assertThat (request .getRequestUrl ().queryParameter ("precision" )).isEqualTo ("ns" );
147+ assertThat (request .getUrl ()).isNotNull ();
148+ assertThat (request .getUrl ().queryParameter ("precision" )).isEqualTo ("ns" );
149149 }
150150
151151 @ Test
@@ -157,8 +157,8 @@ void precisionParameterSpecified() throws InterruptedException {
157157 assertThat (mockServer .getRequestCount ()).isEqualTo (1 );
158158 RecordedRequest request = mockServer .takeRequest ();
159159 assertThat (request ).isNotNull ();
160- assertThat (request .getRequestUrl ()).isNotNull ();
161- assertThat (request .getRequestUrl ().queryParameter ("precision" )).isEqualTo ("s" );
160+ assertThat (request .getUrl ()).isNotNull ();
161+ assertThat (request .getUrl ().queryParameter ("precision" )).isEqualTo ("s" );
162162 }
163163
164164 @ Test
@@ -170,9 +170,9 @@ void gzipParameter() throws InterruptedException {
170170 assertThat (mockServer .getRequestCount ()).isEqualTo (1 );
171171 RecordedRequest request = mockServer .takeRequest ();
172172 assertThat (request ).isNotNull ();
173- assertThat (request .getRequestUrl ()).isNotNull ();
174- assertThat (request .getHeader ("Content-Type" )).isEqualTo ("text/plain; charset=utf-8" );
175- assertThat (request .getHeader ("Content-Encoding" )).isNotEqualTo ("gzip" );
173+ assertThat (request .getUrl ()).isNotNull ();
174+ assertThat (request .getHeaders (). get ("Content-Type" )).isEqualTo ("text/plain; charset=utf-8" );
175+ assertThat (request .getHeaders (). get ("Content-Encoding" )).isNotEqualTo ("gzip" );
176176 }
177177
178178 @ Test
@@ -184,9 +184,9 @@ void gzipParameterSpecified() throws InterruptedException {
184184 assertThat (mockServer .getRequestCount ()).isEqualTo (1 );
185185 RecordedRequest request = mockServer .takeRequest ();
186186 assertThat (request ).isNotNull ();
187- assertThat (request .getRequestUrl ()).isNotNull ();
188- assertThat (request .getHeader ("Content-Type" )).isEqualTo ("text/plain; charset=utf-8" );
189- assertThat (request .getHeader ("Content-Encoding" )).isEqualTo ("gzip" );
187+ assertThat (request .getUrl ()).isNotNull ();
188+ assertThat (request .getHeaders (). get ("Content-Type" )).isEqualTo ("text/plain; charset=utf-8" );
189+ assertThat (request .getHeaders (). get ("Content-Encoding" )).isEqualTo ("gzip" );
190190 }
191191
192192 @ Test
@@ -199,10 +199,10 @@ void writeNoSyncFalseUsesV2API() throws InterruptedException {
199199 assertThat (mockServer .getRequestCount ()).isEqualTo (1 );
200200 RecordedRequest request = mockServer .takeRequest ();
201201 assertThat (request ).isNotNull ();
202- assertThat (request .getRequestUrl ()).isNotNull ();
203- assertThat (request .getRequestUrl ().encodedPath ()).isEqualTo ("/api/v2/write" );
204- assertThat (request .getRequestUrl ().queryParameter ("no_sync" )).isNull ();
205- assertThat (request .getRequestUrl ().queryParameter ("precision" )).isEqualTo ("ns" );
202+ assertThat (request .getUrl ()).isNotNull ();
203+ assertThat (request .getUrl ().encodedPath ()).isEqualTo ("/api/v2/write" );
204+ assertThat (request .getUrl ().queryParameter ("no_sync" )).isNull ();
205+ assertThat (request .getUrl ().queryParameter ("precision" )).isEqualTo ("ns" );
206206
207207 }
208208
@@ -216,10 +216,10 @@ void writeNoSyncTrueUsesV3API() throws InterruptedException {
216216 assertThat (mockServer .getRequestCount ()).isEqualTo (1 );
217217 RecordedRequest request = mockServer .takeRequest ();
218218 assertThat (request ).isNotNull ();
219- assertThat (request .getRequestUrl ()).isNotNull ();
220- assertThat (request .getRequestUrl ().encodedPath ()).isEqualTo ("/api/v3/write_lp" );
221- assertThat (request .getRequestUrl ().queryParameter ("no_sync" )).isEqualTo ("true" );
222- assertThat (request .getRequestUrl ().queryParameter ("precision" )).isEqualTo ("nanosecond" );
219+ assertThat (request .getUrl ()).isNotNull ();
220+ assertThat (request .getUrl ().encodedPath ()).isEqualTo ("/api/v3/write_lp" );
221+ assertThat (request .getUrl ().queryParameter ("no_sync" )).isEqualTo ("true" );
222+ assertThat (request .getUrl ().queryParameter ("precision" )).isEqualTo ("nanosecond" );
223223 }
224224
225225 @ Test
@@ -234,10 +234,10 @@ void writeNoSyncTrueOnV2ServerThrowsException() throws InterruptedException {
234234 assertThat (mockServer .getRequestCount ()).isEqualTo (1 );
235235 RecordedRequest request = mockServer .takeRequest ();
236236 assertThat (request ).isNotNull ();
237- assertThat (request .getRequestUrl ()).isNotNull ();
238- assertThat (request .getRequestUrl ().encodedPath ()).isEqualTo ("/api/v3/write_lp" );
239- assertThat (request .getRequestUrl ().queryParameter ("no_sync" )).isEqualTo ("true" );
240- assertThat (request .getRequestUrl ().queryParameter ("precision" )).isEqualTo ("millisecond" );
237+ assertThat (request .getUrl ()).isNotNull ();
238+ assertThat (request .getUrl ().encodedPath ()).isEqualTo ("/api/v3/write_lp" );
239+ assertThat (request .getUrl ().queryParameter ("no_sync" )).isEqualTo ("true" );
240+ assertThat (request .getUrl ().queryParameter ("precision" )).isEqualTo ("millisecond" );
241241
242242 assertThat (ae .statusCode ()).isEqualTo (HttpResponseStatus .METHOD_NOT_ALLOWED .code ());
243243 assertThat (ae .getMessage ()).contains ("Server doesn't support write with NoSync=true"
@@ -376,7 +376,7 @@ private void checkWriteCalled(final String expectedPath, final String expectedDB
376376 final String expectedPrecision , final boolean expectedNoSync ,
377377 final boolean expectedGzip ) throws InterruptedException {
378378 RecordedRequest request = assertThatServerRequested ();
379- HttpUrl requestUrl = request .getRequestUrl ();
379+ HttpUrl requestUrl = request .getUrl ();
380380 assertThat (requestUrl ).isNotNull ();
381381 assertThat (requestUrl .encodedPath ()).isEqualTo (expectedPath );
382382 if (expectedNoSync ) {
@@ -391,9 +391,9 @@ private void checkWriteCalled(final String expectedPath, final String expectedDB
391391 assertThat (requestUrl .queryParameter ("no_sync" )).isNull ();
392392 }
393393 if (expectedGzip ) {
394- assertThat (request .getHeader ("Content-Encoding" )).isEqualTo ("gzip" );
394+ assertThat (request .getHeaders (). get ("Content-Encoding" )).isEqualTo ("gzip" );
395395 } else {
396- assertThat (request .getHeader ("Content-Encoding" )).isNull ();
396+ assertThat (request .getHeaders (). get ("Content-Encoding" )).isNull ();
397397 }
398398 }
399399
@@ -415,11 +415,11 @@ void allParameterSpecified() throws InterruptedException {
415415 assertThat (mockServer .getRequestCount ()).isEqualTo (1 );
416416 RecordedRequest request = mockServer .takeRequest ();
417417 assertThat (request ).isNotNull ();
418- assertThat (request .getRequestUrl ()).isNotNull ();
419- assertThat (request .getHeader ("Content-Type" )).isEqualTo ("text/plain; charset=utf-8" );
420- assertThat (request .getHeader ("Content-Encoding" )).isEqualTo ("gzip" );
421- assertThat (request .getRequestUrl ().queryParameter ("precision" )).isEqualTo ("s" );
422- assertThat (request .getRequestUrl ().queryParameter ("bucket" )).isEqualTo ("your-database" );
418+ assertThat (request .getUrl ()).isNotNull ();
419+ assertThat (request .getHeaders (). get ("Content-Type" )).isEqualTo ("text/plain; charset=utf-8" );
420+ assertThat (request .getHeaders (). get ("Content-Encoding" )).isEqualTo ("gzip" );
421+ assertThat (request .getUrl ().queryParameter ("precision" )).isEqualTo ("s" );
422+ assertThat (request .getUrl ().queryParameter ("bucket" )).isEqualTo ("your-database" );
423423 }
424424
425425 @ Test
@@ -431,8 +431,8 @@ void contentHeaders() throws InterruptedException {
431431 assertThat (mockServer .getRequestCount ()).isEqualTo (1 );
432432 RecordedRequest request = mockServer .takeRequest ();
433433 assertThat (request ).isNotNull ();
434- assertThat (request .getHeader ("Content-Type" )).isEqualTo ("text/plain; charset=utf-8" );
435- assertThat (request .getHeader ("Content-Encoding" )).isNull ();
434+ assertThat (request .getHeaders (). get ("Content-Type" )).isEqualTo ("text/plain; charset=utf-8" );
435+ assertThat (request .getHeaders (). get ("Content-Encoding" )).isNull ();
436436 }
437437
438438 @ Test
@@ -444,7 +444,10 @@ void bodyRecord() throws InterruptedException {
444444 assertThat (mockServer .getRequestCount ()).isEqualTo (1 );
445445 RecordedRequest request = mockServer .takeRequest ();
446446 assertThat (request ).isNotNull ();
447- assertThat (request .getBody ().readUtf8 ()).isEqualTo ("mem,tag=one value=1.0" );
447+ // assertThat(request.getBody().readUtf8()).isEqualTo("mem,tag=one value=1.0");
448+ // assertThat(request.getBody().getUtf8$okio()).isEqualTo("mem,tag=one value=1.0");
449+ assertThat (request .getBody ().utf8 ()).isEqualTo ("mem,tag=one value=1.0" );
450+
448451 }
449452
450453 @ Test
@@ -460,7 +463,8 @@ void bodyPoint() throws InterruptedException {
460463 assertThat (mockServer .getRequestCount ()).isEqualTo (1 );
461464 RecordedRequest request = mockServer .takeRequest ();
462465 assertThat (request ).isNotNull ();
463- assertThat (request .getBody ().readUtf8 ()).isEqualTo ("mem,tag=one value=1.0" );
466+ // assertThat(request.getBody().readUtf8()).isEqualTo("mem,tag=one value=1.0");
467+ assertThat (request .getBody ().utf8 ()).isEqualTo ("mem,tag=one value=1.0" );
464468 }
465469
466470 @ Test
@@ -480,7 +484,8 @@ void bodyConcat() throws InterruptedException {
480484 assertThat (mockServer .getRequestCount ()).isEqualTo (1 );
481485 RecordedRequest request = mockServer .takeRequest ();
482486 assertThat (request ).isNotNull ();
483- assertThat (request .getBody ().readUtf8 ()).isEqualTo ("mem,tag=one value=1.0\n cpu,tag=two value=2.0" );
487+ // assertThat(request.getBody().readUtf8()).isEqualTo("mem,tag=one value=1.0\ncpu,tag=two value=2.0");
488+ assertThat (request .getBody ().utf8 ()).isEqualTo ("mem,tag=one value=1.0\n cpu,tag=two value=2.0" );
484489 }
485490
486491
@@ -502,17 +507,16 @@ void defaultTags() throws InterruptedException {
502507 RecordedRequest request = mockServer .takeRequest ();
503508
504509 assertThat (request ).isNotNull ();
505- assertThat (request .getBody ().readUtf8 ()).isEqualTo ("mem,model=M5,tag=one,unit=U2 value=1.0" );
510+ // assertThat(request.getBody().readUtf8()).isEqualTo("mem,model=M5,tag=one,unit=U2 value=1.0");
511+ assertThat (request .getBody ().utf8 ()).isEqualTo ("mem,model=M5,tag=one,unit=U2 value=1.0" );
506512
507513 }
508514
509515 @ Test
510516 public void retryHandled429Test () {
511- mockServer .enqueue (createResponse (429 )
512- .setBody ("{ \" message\" : \" Too Many Requests\" }" )
513- .setHeader ("retry-after" , "42" )
514- .setHeader ("content-type" , "application/json" )
515- );
517+ mockServer .enqueue (createResponse (429 ,
518+ Map .of ("retry-after" , "42" , "content-type" , "application/json" ),
519+ "{ \" message\" : \" Too Many Requests\" }" ));
516520
517521 Point point = Point .measurement ("mem" )
518522 .setTag ("tag" , "one" )
0 commit comments