001/** 002 * Copyright (C) 2006-2018 Talend Inc. - www.talend.com 003 * 004 * Licensed under the Apache License, Version 2.0 (the "License"); 005 * you may not use this file except in compliance with the License. 006 * You may obtain a copy of the License at 007 * 008 * http://www.apache.org/licenses/LICENSE-2.0 009 * 010 * Unless required by applicable law or agreed to in writing, software 011 * distributed under the License is distributed on an "AS IS" BASIS, 012 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 013 * See the License for the specific language governing permissions and 014 * limitations under the License. 015 */ 016package org.talend.sdk.component.junit.http.internal.impl; 017 018import static java.util.Collections.emptyList; 019import static java.util.Comparator.comparing; 020import static java.util.stream.Collectors.toMap; 021 022import java.io.ByteArrayInputStream; 023import java.io.ByteArrayOutputStream; 024import java.io.IOException; 025import java.io.InputStream; 026import java.nio.charset.StandardCharsets; 027import java.util.List; 028import java.util.Map; 029import java.util.stream.StreamSupport; 030import java.util.zip.GZIPInputStream; 031 032import org.talend.sdk.component.junit.http.api.HttpApiHandler; 033import org.talend.sdk.component.junit.http.api.Response; 034 035import io.netty.handler.codec.http.FullHttpRequest; 036 037import lombok.extern.slf4j.Slf4j; 038 039@Slf4j 040public class DefaultResponseLocatorCapturingHandler extends PassthroughHandler { 041 042 public DefaultResponseLocatorCapturingHandler(final HttpApiHandler api) { 043 super(api); 044 } 045 046 @Override 047 protected void beforeResponse(final String requestUri, final FullHttpRequest request, final Response resp, 048 final Map<String, List<String>> responseHeaderFields) { 049 final DefaultResponseLocator.RequestModel requestModel = new DefaultResponseLocator.RequestModel(); 050 requestModel.setMethod(request.method().name()); 051 requestModel.setUri(requestUri); 052 requestModel.setHeaders(filterHeaders(request.headers())); 053 final DefaultResponseLocator.Model model = new DefaultResponseLocator.Model(); 054 model.setRequest(requestModel); 055 056 final DefaultResponseLocator.ResponseModel responseModel = new DefaultResponseLocator.ResponseModel(); 057 responseModel.setStatus(resp.status()); 058 responseModel.setHeaders(filterHeaders(resp.headers().entrySet())); 059 // todo: support as byte[] for not text responses 060 if (resp.payload() != null) { 061 if (responseHeaderFields 062 .getOrDefault("Content-Encoding", emptyList()) 063 .stream() 064 .anyMatch(it -> it.contains("gzip"))) { 065 responseModel.setPayload(new String(degzip(resp.payload()), StandardCharsets.UTF_8)); 066 } else { 067 responseModel.setPayload(new String(resp.payload(), StandardCharsets.UTF_8)); 068 } 069 } 070 model.setResponse(responseModel); 071 072 if (DefaultResponseLocator.class.isInstance(api.getResponseLocator())) { 073 DefaultResponseLocator.class.cast(api.getResponseLocator()).addModel(model); 074 } 075 } 076 077 private byte[] degzip(final byte[] payloadBytes) { 078 final ByteArrayOutputStream out = new ByteArrayOutputStream(); 079 final byte[] bytes = new byte[1024 * 8]; 080 int read; 081 try (final InputStream stream = new GZIPInputStream(new ByteArrayInputStream(payloadBytes))) { 082 while ((read = stream.read(bytes)) >= 0) { 083 if (read == 0) { 084 continue; 085 } 086 out.write(bytes, 0, read); 087 } 088 } catch (final IOException e) { 089 throw new IllegalStateException(e); 090 } 091 return out.toByteArray(); 092 } 093 094 private Map<String, String> filterHeaders(final Iterable<Map.Entry<String, String>> headers) { 095 return StreamSupport 096 .stream(headers.spliterator(), false) 097 .filter(h -> !api.getHeaderFilter().test(h.getKey())) 098 .sorted(comparing(Map.Entry::getKey)) 099 .collect(toMap(Map.Entry::getKey, Map.Entry::getValue)); 100 } 101}