org.springframework.cloud.dataflow.server.single.security.OAuth2ServerResource.java Source code

Introduction

Here is the source code for org.springframework.cloud.dataflow.server.single.security.OAuth2ServerResource.java

Source

/*
 * Copyright 2017-2019 the original author or authors.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      https://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package org.springframework.cloud.dataflow.server.single.security;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.junit.rules.ExternalResource;

import org.springframework.boot.builder.SpringApplicationBuilder;
import org.springframework.cloud.dataflow.server.single.security.support.OAuth2TestServer;
import org.springframework.context.ConfigurableApplicationContext;
import org.springframework.core.io.Resource;
import org.springframework.core.io.support.PathMatchingResourcePatternResolver;
import org.springframework.util.SocketUtils;

/**
 * Bootstraps an embedded OAuth2 server using {@link OAuth2TestServer}.
 *
 * @author Gunnar Hillert
 */
public class OAuth2ServerResource extends ExternalResource {

    private static final String OAUTH2_PORT_PROPERTY = "oauth2.port";

    private final Log LOGGER = LogFactory.getLog(OAuth2ServerResource.class);

    private String originalOAuth2Port;

    private int oauth2ServerPort;

    private ConfigurableApplicationContext application;

    public OAuth2ServerResource() {
        super();
    }

    @Override
    protected void before() throws Throwable {

        originalOAuth2Port = System.getProperty(OAUTH2_PORT_PROPERTY);

        this.oauth2ServerPort = SocketUtils.findAvailableTcpPort();

        LOGGER.info("Setting OAuth2 Server port to " + this.oauth2ServerPort);

        System.setProperty(OAUTH2_PORT_PROPERTY, String.valueOf(this.oauth2ServerPort));

        final String configurationLocation = "classpath:/org/springframework/cloud/dataflow/server/single/security/support/oauth2TestServerConfig.yml";

        final Resource resource = new PathMatchingResourcePatternResolver().getResource(configurationLocation);
        if (!resource.exists()) {
            throw new IllegalArgumentException(String
                    .format("Resource 'configurationLocation' ('%s') does not exist.", configurationLocation));
        }

        this.application = new SpringApplicationBuilder(OAuth2TestServer.class).build().run(
                "--spring.cloud.common.security.enabled=false", "--spring.cloud.kubernetes.enabled=false",
                "--spring.config.additional-location=" + configurationLocation);
    }

    @Override
    protected void after() {
        try {
            application.stop();
        } finally {
            if (originalOAuth2Port != null) {
                System.setProperty(OAUTH2_PORT_PROPERTY, originalOAuth2Port);
            } else {
                System.clearProperty(OAUTH2_PORT_PROPERTY);
            }
        }
    }

    public int getOauth2ServerPort() {
        return oauth2ServerPort;
    }

}