com.netflix.archaius.test.TestCompositeConfig Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of archaius2-test Show documentation
archaius2-test
The newest version!
package com.netflix.archaius.test;

import java.util.Iterator;
import java.util.Properties;

import com.netflix.archaius.api.Config;
import com.netflix.archaius.api.config.SettableConfig;
import com.netflix.archaius.config.DefaultCompositeConfig;
import com.netflix.archaius.config.DefaultSettableConfig;

/**
 * Implementation of {@link DefaultCompositeConfig} and {@link SettableConfig}
 * for use in testing utilities. 
 */
public class TestCompositeConfig extends DefaultCompositeConfig implements SettableConfig {

    private static final String CLASS_LEVEL_LAYER_NAME =    "CLASS_LEVEL_TEST_OVERRIDES";
    private static final String METHOD_LEVEL_LAYER_NAME =   "METHOD_LEVEL_TEST_OVERRIDES";
    private static final String RUNTIME_LAYER_NAME =        "RUNTIME";
    
    
    public TestCompositeConfig(SettableConfig runtimeOverrides, SettableConfig classLevelOverrides, SettableConfig methodLevelOverrides) {
        try {
            addConfig(RUNTIME_LAYER_NAME, runtimeOverrides);
            addConfig(METHOD_LEVEL_LAYER_NAME, methodLevelOverrides);
            addConfig(CLASS_LEVEL_LAYER_NAME, classLevelOverrides);
        } catch (Exception e) {
            throw new RuntimeException(e);
        }
    }
    
    @Deprecated
    public TestCompositeConfig(SettableConfig classLevelOverrides, SettableConfig methodLevelOverrides) {
        this(new DefaultSettableConfig(), classLevelOverrides, methodLevelOverrides);
    }
    
    public void resetForTest() {
        clear(getSettableConfig(METHOD_LEVEL_LAYER_NAME));
        clear(getSettableConfig(RUNTIME_LAYER_NAME));
    }
    
    private SettableConfig getSettableConfig(String configName) {
        return (SettableConfig) super.getConfig(configName);
    }
    
    private void clear(SettableConfig config) {
        Iterator keys = config.getKeys();
        while(keys.hasNext()) {
            config.clearProperty(keys.next());
        }
    }

    @Override
    public void setProperties(Config config) {
        getSettableConfig(RUNTIME_LAYER_NAME).setProperties(config);
    }

    @Override
    public void setProperties(Properties properties) {
        getSettableConfig(RUNTIME_LAYER_NAME).setProperties(properties);
    }

    @Override
    public  void setProperty(String propName, T propValue) {
        getSettableConfig(RUNTIME_LAYER_NAME).setProperty(propName, propValue);
    }

    @Override
    public void clearProperty(String propName) {
        getSettableConfig(RUNTIME_LAYER_NAME).clearProperty(propName);
    }
}