com.netflix.archaius.test.TestPropertyOverrideAnnotationReader 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.io.FileNotFoundException;
import java.io.IOException;
import java.io.InputStream;
import java.util.Arrays;
import java.util.Properties;

/**
 * Utility class to read {@literal@}{@link TestPropertyOverride} annotations and
 * transform them into {@link Properties} objects. This is intended for use in 
 * testing utilities such as the {@link Archaius2TestConfig} JUnit Rule. 
 */
public class TestPropertyOverrideAnnotationReader {
    
    public Properties getPropertiesForAnnotation(TestPropertyOverride annotation) {
        Properties properties = new Properties();
        if (annotation == null) {
            return properties;
        }

        for (String fileName : annotation.propertyFiles()) {
            try {
                InputStream propFileStream = this.getClass().getClassLoader().getResourceAsStream(fileName);
                if(propFileStream != null) {
                    properties.load(propFileStream);
                }
                else {
                    throw new FileNotFoundException(fileName);
                }
            } catch (IOException e) {
                throw new TestConfigException("Failed to load property file from classpath", e);
            }
        }
        for (String override : annotation.value()) {
            String[] parts = override.split("=", 2);
            if (parts.length < 2) {
                throw new TestConfigException("Error parsing TestPropertyOverride for: " + Arrays.toString(annotation.value())
                        + " Please ensure you are specifying overrides in the form \"key=value\"");
            }
            properties.put(parts[0], parts[1]);
        }
        return properties;
    }
}