package com.facebook.presto.jdbc.internal.airlift.http.client.testing;
import com.facebook.presto.jdbc.internal.guava.base.Charsets;
import com.facebook.presto.jdbc.internal.guava.collect.ImmutableListMultimap;
import com.facebook.presto.jdbc.internal.guava.collect.ListMultimap;
import com.facebook.presto.jdbc.internal.guava.io.CountingInputStream;
import com.facebook.presto.jdbc.internal.guava.net.HttpHeaders;
import com.facebook.presto.jdbc.internal.guava.net.MediaType;
import com.facebook.presto.jdbc.internal.airlift.http.client.HttpStatus;
import com.facebook.presto.jdbc.internal.airlift.http.client.Response;
import java.io.ByteArrayInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.util.List;
import static com.facebook.presto.jdbc.internal.guava.base.Preconditions.checkNotNull;
public class TestingResponse
implements Response
{
private final HttpStatus status;
private final ListMultimap<String, String> headers;
private final CountingInputStream countingInputStream;
public TestingResponse(HttpStatus status, ListMultimap<String, String> headers, byte[] bytes)
{
this(status, headers, new ByteArrayInputStream(checkNotNull(bytes, "bytes is null")));
}
public TestingResponse(HttpStatus status, ListMultimap<String, String> headers, InputStream input)
{
this.status = checkNotNull(status, "status is null");
this.headers = ImmutableListMultimap.copyOf(checkNotNull(headers, "headers is null"));
this.countingInputStream = new CountingInputStream(checkNotNull(input, "input is null"));
}
@Override
public int getStatusCode()
{
return status.code();
}
@Override
public String getStatusMessage()
{
return status.reason();
}
@Override
public String getHeader(String name)
{
List<String> list = getHeaders().get(name);
return list.isEmpty() ? null : list.get(0);
}
@Override
public ListMultimap<String, String> getHeaders()
{
return headers;
}
@Override
public long getBytesRead()
{
return countingInputStream.getCount();
}
@Override
public InputStream getInputStream()
throws IOException
{
return countingInputStream;
}
public static ListMultimap<String, String> contentType(MediaType type)
{
return ImmutableListMultimap.of(HttpHeaders.CONTENT_TYPE, type.toString());
}
public static Response mockResponse(HttpStatus status, MediaType type, String content)
{
return new TestingResponse(status, contentType(type), content.getBytes(Charsets.UTF_8));
}
}