Example 1 with Buffer
use of okio.Buffer in project buck by facebook.
the class HttpArtifactCacheTest method testStore.
@Test
public void testStore() throws Exception {
final RuleKey ruleKey = new RuleKey("00000000000000000000000000000000");
final String data = "data";
FakeProjectFilesystem filesystem = new FakeProjectFilesystem();
Path output = Paths.get("output/file");
filesystem.writeContentsToPath(data, output);
final AtomicBoolean hasCalled = new AtomicBoolean(false);
argsBuilder.setProjectFilesystem(filesystem);
argsBuilder.setStoreClient(withMakeRequest(((path, requestBuilder) -> {
Request request = requestBuilder.url(SERVER).build();
hasCalled.set(true);
Buffer buf = new Buffer();
request.body().writeTo(buf);
byte[] actualData = buf.readByteArray();
byte[] expectedData;
try (ByteArrayOutputStream out = new ByteArrayOutputStream();
DataOutputStream dataOut = new DataOutputStream(out)) {
dataOut.write(HttpArtifactCacheBinaryProtocol.createKeysHeader(ImmutableSet.of(ruleKey)));
byte[] metadata = HttpArtifactCacheBinaryProtocol.createMetadataHeader(ImmutableSet.of(ruleKey), ImmutableMap.of(), ByteSource.wrap(data.getBytes(Charsets.UTF_8)));
dataOut.writeInt(metadata.length);
dataOut.write(metadata);
dataOut.write(data.getBytes(Charsets.UTF_8));
expectedData = out.toByteArray();
}
assertArrayEquals(expectedData, actualData);
Response response = new Response.Builder().body(createDummyBody()).code(HttpURLConnection.HTTP_ACCEPTED).protocol(Protocol.HTTP_1_1).request(request).build();
return new OkHttpResponseWrapper(response);
})));
HttpArtifactCache cache = new HttpArtifactCache(argsBuilder.build());
cache.storeImpl(ArtifactInfo.builder().addRuleKeys(ruleKey).build(), output, createFinishedEventBuilder());
assertTrue(hasCalled.get());
cache.close();
}
Also used :
Path(java.nio.file.Path)
LazyPath(com.facebook.buck.io.LazyPath)
Buffer(okio.Buffer)
RuleKey(com.facebook.buck.rules.RuleKey)
FakeProjectFilesystem(com.facebook.buck.testutil.FakeProjectFilesystem)
DataOutputStream(java.io.DataOutputStream)
Request(okhttp3.Request)
ByteArrayOutputStream(java.io.ByteArrayOutputStream)
OkHttpResponseWrapper(com.facebook.buck.slb.OkHttpResponseWrapper)
Response(okhttp3.Response)
HttpResponse(com.facebook.buck.slb.HttpResponse)
AtomicBoolean(java.util.concurrent.atomic.AtomicBoolean)
Test(org.junit.Test)
Example 2 with Buffer
use of okio.Buffer in project okhttputils by hongyangAndroid.
the class LoggerInterceptor method bodyToString.
private String bodyToString(final Request request) {
try {
final Request copy = request.newBuilder().build();
final Buffer buffer = new Buffer();
copy.body().writeTo(buffer);
return buffer.readUtf8();
} catch (final IOException e) {
return "something error when show requestBody.";
}
}Example 3 with Buffer
use of okio.Buffer in project k-9 by k9mail.
the class ImapFolderTest method fetchPart_withTextSection_shouldProcessImapResponses.
@Test
public void fetchPart_withTextSection_shouldProcessImapResponses() throws Exception {
ImapFolder folder = createFolder("Folder");
prepareImapFolderForOpen(OPEN_MODE_RO);
folder.open(OPEN_MODE_RO);
ImapMessage message = createImapMessage("1");
Part part = createPlainTextPart("1.1");
setupSingleFetchResponseToCallback();
folder.fetchPart(message, part, null);
ArgumentCaptor<Body> bodyArgumentCaptor = ArgumentCaptor.forClass(Body.class);
verify(part).setBody(bodyArgumentCaptor.capture());
Body body = bodyArgumentCaptor.getValue();
Buffer buffer = new Buffer();
body.writeTo(buffer.outputStream());
assertEquals("text", buffer.readUtf8());
}
Also used :
Buffer(okio.Buffer)
Part(com.fsck.k9.mail.Part)
Body(com.fsck.k9.mail.Body)
BinaryTempFileBody(com.fsck.k9.mail.internet.BinaryTempFileBody)
Test(org.junit.Test)
Example 4 with Buffer
use of okio.Buffer in project okhttp-OkGo by jeasonlzy.
the class TypeUtils method buffer.
public static ResponseBody buffer(final ResponseBody body) throws IOException {
Buffer buffer = new Buffer();
body.source().readAll(buffer);
return ResponseBody.create(body.contentType(), body.contentLength(), buffer);
}
Also used :
Buffer(okio.Buffer)
Example 5 with Buffer
use of okio.Buffer in project Parse-SDK-Android by ParsePlatform.
the class ParseOkHttpClientTest method testParseOkHttpClientExecuteWithExternalInterceptorAndGZIPResponse.
@Test
public void testParseOkHttpClientExecuteWithExternalInterceptorAndGZIPResponse() throws Exception {
// Make mock response
Buffer buffer = new Buffer();
final ByteArrayOutputStream byteOut = new ByteArrayOutputStream();
GZIPOutputStream gzipOut = new GZIPOutputStream(byteOut);
gzipOut.write("content".getBytes());
gzipOut.close();
buffer.write(byteOut.toByteArray());
MockResponse mockResponse = new MockResponse().setStatus("HTTP/1.1 " + 201 + " " + "OK").setBody(buffer).setHeader("Content-Encoding", "gzip");
// Start mock server
server.enqueue(mockResponse);
server.start();
ParseHttpClient client = new ParseOkHttpClient(10000, null);
final Semaphore done = new Semaphore(0);
// Add plain interceptor to disable decompress response stream
client.addExternalInterceptor(new ParseNetworkInterceptor() {
@Override
public ParseHttpResponse intercept(Chain chain) throws IOException {
done.release();
ParseHttpResponse parseResponse = chain.proceed(chain.getRequest());
// Make sure the response we get from the interceptor is the raw gzip stream
byte[] content = ParseIOUtils.toByteArray(parseResponse.getContent());
assertArrayEquals(byteOut.toByteArray(), content);
// We need to set a new stream since we have read it
return new ParseHttpResponse.Builder().setContent(new ByteArrayInputStream(byteOut.toByteArray())).build();
}
});
// We do not need to add Accept-Encoding header manually, httpClient library should do that.
String requestUrl = server.url("/").toString();
ParseHttpRequest parseRequest = new ParseHttpRequest.Builder().setUrl(requestUrl).setMethod(ParseHttpRequest.Method.GET).build();
// Execute request
ParseHttpResponse parseResponse = client.execute(parseRequest);
// Make sure the response we get is ungziped by OkHttp library
byte[] content = ParseIOUtils.toByteArray(parseResponse.getContent());
assertArrayEquals("content".getBytes(), content);
// Make sure interceptor is called
assertTrue(done.tryAcquire(10, TimeUnit.SECONDS));
server.shutdown();
}
Also used :
Buffer(okio.Buffer)
MockResponse(okhttp3.mockwebserver.MockResponse)
ParseHttpRequest(com.parse.http.ParseHttpRequest)
ByteArrayOutputStream(java.io.ByteArrayOutputStream)
Semaphore(java.util.concurrent.Semaphore)
IOException(java.io.IOException)
GZIPOutputStream(java.util.zip.GZIPOutputStream)
ParseNetworkInterceptor(com.parse.http.ParseNetworkInterceptor)
ByteArrayInputStream(java.io.ByteArrayInputStream)
ParseHttpResponse(com.parse.http.ParseHttpResponse)
Test(org.junit.Test)
Aggregations
Buffer (okio.Buffer)576
Test (org.junit.Test)226
IOException (java.io.IOException)113
MockResponse (okhttp3.mockwebserver.MockResponse)70
BufferedSource (okio.BufferedSource)66
Request (okhttp3.Request)64
ByteString (okio.ByteString)48
TikXml (com.tickaroo.tikxml.TikXml)44
RequestBody (okhttp3.RequestBody)42
ResponseBody (okhttp3.ResponseBody)41
BufferedSink (okio.BufferedSink)35
Test (org.junit.jupiter.api.Test)35
Response (okhttp3.Response)34
EOFException (java.io.EOFException)32
Charset (java.nio.charset.Charset)26
Source (okio.Source)24
InputStream (java.io.InputStream)21
MediaType (okhttp3.MediaType)20
Date (java.util.Date)17
Sink (okio.Sink)17
