com.github.dreamhead.moco.model.DefaultHttpResponse Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of moco-core Show documentation

Moco is an easy setup stub framework, mainly focusing on testing and integration.

There is a newer version: 1.5.0

package com.github.dreamhead.moco.model;

import com.fasterxml.jackson.databind.annotation.JsonDeserialize;
import com.github.dreamhead.moco.HttpProtocolVersion;
import com.github.dreamhead.moco.HttpResponse;
import com.google.common.collect.ImmutableMap;
import io.netty.buffer.ByteBufInputStream;
import io.netty.handler.codec.http.FullHttpResponse;

import java.util.Map;

import static com.github.dreamhead.moco.model.MessageContent.content;
import static com.google.common.collect.ImmutableMap.copyOf;

@JsonDeserialize(builder = DefaultHttpResponse.Builder.class)
public class DefaultHttpResponse implements HttpResponse {
    private final HttpProtocolVersion version;
    private final MessageContent content;
    private final ImmutableMap headers;
    private final int status;

    public DefaultHttpResponse(HttpProtocolVersion version, int status, ImmutableMap headers, MessageContent content) {
        this.version = version;
        this.headers = headers;
        this.content = content;
        this.status = status;
    }

    public HttpProtocolVersion getVersion() {
        return version;
    }

    @Override
    public MessageContent getContent() {
        return content;
    }

    public ImmutableMap getHeaders() {
        return this.headers;
    }

    public int getStatus() {
        return status;
    }

    public static DefaultHttpResponse newResponse(FullHttpResponse response) {
        ImmutableMap.Builder headerBuilder = ImmutableMap.builder();
        for (Map.Entry entry : response.headers()) {
            headerBuilder.put(entry);
        }

        return builder()
                .withVersion(HttpProtocolVersion.versionOf(response.getProtocolVersion().text()))
                .withStatus(response.getStatus().code())
                .withHeaders(headerBuilder.build())
                .withContent(content()
                        .withContent(new ByteBufInputStream(response.content()))
                        .build())
                .build();
    }

    public static Builder builder() {
        return new Builder();
    }

    public static final class Builder {
        private HttpProtocolVersion version;
        private MessageContent content;
        private ImmutableMap headers;
        private int status;

        public Builder withVersion(HttpProtocolVersion version) {
            this.version = version;
            return this;
        }

        public Builder withContent(MessageContent content) {
            this.content = content;
            return this;
        }

        public Builder withHeaders(Map headers) {
            if (headers != null) {
                this.headers = copyOf(headers);
            }

            return this;
        }

        public Builder withStatus(int code) {
            this.status = code;
            return this;
        }

        public DefaultHttpResponse build() {
            return new DefaultHttpResponse(version, status, headers, content);
        }
    }
}