com.github.dreamhead.moco.extractor.CookieRequestExtractor Maven / Gradle / Ivy

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.extractor;

import com.github.dreamhead.moco.HttpRequest;
import com.github.dreamhead.moco.HttpRequestExtractor;
import com.google.common.base.Optional;
import com.google.common.collect.ImmutableMap;

import static com.google.common.base.Optional.fromNullable;

public class CookieRequestExtractor extends HttpRequestExtractor {
    private final CookiesRequestExtractor extractor = new CookiesRequestExtractor();

    private final String key;

    public CookieRequestExtractor(final String key) {
        this.key = key;
    }

    @Override
    protected Optional doExtract(final HttpRequest request) {
        Optional> cookies = extractor.extract(request);
        return cookies.isPresent() ? fromNullable(cookies.get().get(this.key)) : Optional.absent();
    }
}