@Test
public void testResolveGETWithNoHeaders() throws IOException {
HTTPRequest request = new HTTPRequest(URI.create("http://dummy/uri/123"), HTTPMethod.GET);
final HttpMethod method = mock(GetMethod.class);
HTTPClientResponseResolver resolver =
createResponseResolver(method, Status.valueOf(200), new Header[0]);
HTTPResponse response = resolver.resolve(request);
assertNotNull("Response was null", response);
assertEquals(200, response.getStatus().getCode());
assertEquals(0, response.getHeaders().size());
}
@Test
public void testResolvePUTWithNoHeaders() throws IOException {
HTTPRequest request = new HTTPRequest(URI.create("http://dummy/uri/123"), HTTPMethod.PUT);
final Payload payload = mock(Payload.class);
request = request.payload(payload);
when(payload.getMimeType()).thenReturn(new MIMEType("text/plain"));
final HttpMethod method = mock(PostMethod.class);
HTTPClientResponseResolver resolver =
createResponseResolver(method, Status.valueOf(200), new Header[0]);
HTTPResponse response = resolver.resolve(request);
assertNotNull(response);
assertEquals(200, response.getStatus().getCode());
assertEquals(0, response.getHeaders().size());
}
@Override
public HTTPResponse insert(HTTPRequest request, HTTPResponse response) {
Key key = Key.create(request, response);
Connection connection = getConnection();
String sql =
"insert into response(uri, vary, status, headers, payload, mimeType, cachetime) values (?, ?, ?, ?, ?, ?, ?)";
PreparedStatement statement = null;
try {
JdbcUtil.startTransaction(connection);
invalidate(key, connection);
statement = connection.prepareStatement(sql);
statement.setString(1, key.getURI().toString());
statement.setString(2, key.getVary().toJSON());
statement.setInt(3, response.getStatus().getCode());
statement.setString(4, response.getHeaders().toJSON());
InputStream inputStream = null;
if (response.hasPayload() && response.getPayload().isAvailable()) {
statement.setString(6, response.getPayload().getMimeType().toString());
inputStream = response.getPayload().getInputStream();
statement.setBinaryStream(5, inputStream);
} else {
statement.setNull(5, Types.BLOB);
statement.setNull(6, Types.VARCHAR);
}
statement.setTimestamp(7, new Timestamp(DateTimeUtils.currentTimeMillis()));
try {
statement.executeUpdate();
} finally {
IOUtils.closeQuietly(inputStream);
}
connection.commit();
return getImpl(connection, key);
} catch (SQLException e) {
JdbcUtil.rollback(connection);
JdbcUtil.close(connection);
throw new DataAccessException(e);
} finally {
JdbcUtil.endTransaction(connection);
JdbcUtil.close(statement);
}
}