AccessTokenRefresherInitTest.java

/**
 * Copyright (C) 2015 Zalando SE (http://tech.zalando.com)
 *
 * 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
 *
 *         http://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.zalando.stups.tokens;

import java.net.URI;
import java.net.URISyntaxException;

import org.junit.Assert;
import org.junit.Test;

public class AccessTokenRefresherInitTest {

    @Test
    public void testInitializeFromEnvironment() {

        AccessTokenRefresher refresher = new TestAccessTokenRefresher(config(30, 10),
                "kio=1234567890,pierone=987654321");
        refresher.initializeFixedTokensFromEnvironment();
        Assert.assertEquals(refresher.get("kio"), "1234567890");
        Assert.assertEquals(refresher.get("pierone"), "987654321");
    }

    @Test(expected = AccessTokenUnavailableException.class)
    public void testInitializeFromEnvironmentFailure() {
        // System.setProperty("OAUTH2_ACCESS_TOKENS", "kio=,pierone=987654321");

        AccessTokenRefresher refresher = new TestAccessTokenRefresher(config(30, 10), "kio=,pierone=987654321");
        refresher.initializeFixedTokensFromEnvironment();

        // this works
        refresher.get("pierone");

        // this should fail
        refresher.get("kio");

    }

    private static AccessTokensBuilder config(final int refreshPercentLeft, final int warnPercentLeft) {
        try {
            return Tokens.createAccessTokensWithUri(new URI("http://localhost")).refreshPercentLeft(refreshPercentLeft)
                         .warnPercentLeft(warnPercentLeft);
        } catch (URISyntaxException e) {
            throw new IllegalStateException(e);
        }
    }

    private static class TestAccessTokenRefresher extends AccessTokenRefresher {

        private final String fixedToken;

        public TestAccessTokenRefresher(final AccessTokensBuilder configuration, final String fixedToken) {
            super(configuration);
            this.fixedToken = fixedToken;
        }

        @Override
        protected String getFixedToken() {
            return fixedToken;
        }

    }
}