org.constretto.spring.assembly.AssemblyWithMultipleEnvironmentsTest.java Source code

Introduction

Here is the source code for org.constretto.spring.assembly.AssemblyWithMultipleEnvironmentsTest.java

Source

/*
 * Copyright 2008 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
 * 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.constretto.spring.assembly;

import org.constretto.exception.ConstrettoException;
import static org.constretto.spring.annotation.Environment.*;
import org.constretto.spring.assembly.helper.ConfigurationService;
import static org.constretto.spring.internal.resolver.DefaultAssemblyContextResolver.ASSEMBLY_KEY;
import static org.junit.Assert.assertEquals;
import org.junit.Before;
import org.junit.Test;
import org.springframework.beans.factory.BeanCreationException;
import org.springframework.context.support.ClassPathXmlApplicationContext;

/**
 * @author <a href="mailto:kaare.nilsen@gmail.com">Kaare Nilsen</a>
 */
public class AssemblyWithMultipleEnvironmentsTest {
    ConfigurationService configurationService;

    @Before
    public void removeAssemblyKey() {
        System.clearProperty(ASSEMBLY_KEY);
    }

    @Test(expected = BeanCreationException.class)
    public void givenNoAssemblyContextAndNoDefaultPresentThenFail() {
        loadContextAndInjectConfigurationService();
    }

    @Test
    public void givenAssemblyContextWhenClassAnnotatedWithOneEnvironmentThenSelectCorrectImplementation() {
        System.setProperty(ASSEMBLY_KEY, PRODUCTION);
        loadContextAndInjectConfigurationService();
        assertEquals(PRODUCTION, configurationService.getRunningEnvironment());
    }

    @Test
    public void givenAssemblyContextWhenClassAnnotatedWithMultipleEnvironmentsThenSelectCorrectImplementation() {
        System.setProperty(ASSEMBLY_KEY, DEVELOPMENT);
        loadContextAndInjectConfigurationService();
        assertEquals("developmentAndTest", configurationService.getRunningEnvironment());

        System.setProperty(ASSEMBLY_KEY, TEST);
        loadContextAndInjectConfigurationService();
        assertEquals("developmentAndTest", configurationService.getRunningEnvironment());
    }

    @Test
    public void givenMultipleAssemblyContextsWhenClassAnnotatedWithMultipleEnvironmentsThenSelectCorrectImplementation() {
        System.setProperty(ASSEMBLY_KEY, DEVELOPMENT + "," + PRODUCTION);
        loadContextAndInjectConfigurationService();
        assertEquals("developmentAndTest", configurationService.getRunningEnvironment());

        System.setProperty(ASSEMBLY_KEY, TEST + "," + PRODUCTION);
        loadContextAndInjectConfigurationService();
        assertEquals("developmentAndTest", configurationService.getRunningEnvironment());

        System.setProperty(ASSEMBLY_KEY, PRODUCTION + "," + TEST);
        loadContextAndInjectConfigurationService();
        assertEquals(PRODUCTION, configurationService.getRunningEnvironment());

        System.setProperty(ASSEMBLY_KEY, PRODUCTION + "," + TEST + "," + DEVELOPMENT);
        loadContextAndInjectConfigurationService();
        assertEquals(PRODUCTION, configurationService.getRunningEnvironment());

        System.setProperty(ASSEMBLY_KEY, PRODUCTION + ",mock");
        loadContextAndInjectConfigurationService();
        assertEquals(PRODUCTION, configurationService.getRunningEnvironment());

        System.setProperty(ASSEMBLY_KEY, PRODUCTION + ",unknown");
        loadContextAndInjectConfigurationService();
        assertEquals(PRODUCTION, configurationService.getRunningEnvironment());

        System.setProperty(ASSEMBLY_KEY, "unknown," + PRODUCTION);
        loadContextAndInjectConfigurationService();
        assertEquals(PRODUCTION, configurationService.getRunningEnvironment());
    }

    @Test(expected = ConstrettoException.class)
    public void givenAssemblyContextsWithSamePriorityFoundInSeveralImplementationsThenBeanCreationExceptionIsThrown() {
        System.setProperty(ASSEMBLY_KEY, "mock");
        loadContextAndInjectConfigurationService();

    }

    @Test(expected = BeanCreationException.class)
    public void givenUnRecognizedAssemblyContextWithNoDefaultsPresentThenFail() {
        System.setProperty(ASSEMBLY_KEY, "noHit");
        loadContextAndInjectConfigurationService();
    }

    private void loadContextAndInjectConfigurationService() {
        ClassPathXmlApplicationContext ctx = new ClassPathXmlApplicationContext(
                "org/constretto/spring/assembly/AssemblyWithMultipleEnvironmentsTest-context.xml");
        configurationService = (ConfigurationService) ctx.getBean("configurationService");
    }
}