com.jupiter.tools.mvc.requester.OAuthRequestPostProcessor Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of mvc-requester Show documentation

MvcRequester is a tool to make and assert REST-API communication

The newest version!

package com.jupiter.tools.mvc.requester;

import org.springframework.mock.web.MockHttpServletRequest;
import org.springframework.test.web.servlet.request.RequestPostProcessor;

import static org.apache.commons.lang3.StringUtils.isNotBlank;

/**
 * Created by Maxim Seredkin on 29.03.2018.
 * 
 * Post-processor which will add an OAuth token in the request.
 *
 * @author Maxim Seredkin
 */
public class OAuthRequestPostProcessor implements RequestPostProcessor {
    private final String token;

    public OAuthRequestPostProcessor(String token) {
        this.token = token;
    }

    @Override
    public MockHttpServletRequest postProcessRequest(MockHttpServletRequest mockHttpServletRequest) {
        if (isNotBlank(token)) {
            mockHttpServletRequest.addHeader("Authorization", String.format("Bearer %s", token));
        }

        return mockHttpServletRequest;
    }
}