Examples with ConfigBuilder io.pravega.common.util.ConfigBuilder used on opensource projects

Example 1 with ConfigBuilder

use of io.pravega.common.util.ConfigBuilder in project pravega by pravega.

the class ServiceBuilderConfigTests method testGetConfig.

/**
 * Verifies the include(ConfigBuilder) and getConfig() work properly. This test follows a bit of an unusual approach
 * in that it picks a few known config classes, populates them using their builders and reflection, then uses reflection
 * once again to compare expected output (generated using their builders) with output from the ServiceBuilderConfig.getConfig().
 * <p>
 * This verifies that the namespacing inside ServiceBuilderConfig.builder() works correctly, as well as the constructors
 * for various configs.
 */
@Test
@SuppressWarnings("unchecked")
public void testGetConfig() throws Exception {
    // Select a few classes to test dynamically.
    val testClasses = new HashMap<Class<?>, Supplier<ConfigBuilder<?>>>();
    testClasses.put(ReadIndexConfig.class, ReadIndexConfig::builder);
    testClasses.put(WriterConfig.class, WriterConfig::builder);
    testClasses.put(MetricsConfig.class, MetricsConfig::builder);
    testClasses.put(DurableLogConfig.class, DurableLogConfig::builder);
    testClasses.put(ServiceConfig.class, ServiceConfig::builder);
    // Value generator.
    val nextValue = new AtomicInteger(1000 * 1000 * 1000);
    // Create instances of each test class and dynamically assign their properties some arbitrary values
    val expectedValues = new HashMap<Class<?>, Object>();
    val b = ServiceBuilderConfig.builder();
    for (Map.Entry<Class<?>, Supplier<ConfigBuilder<?>>> e : testClasses.entrySet()) {
        Class<?> c = e.getKey();
        ConfigBuilder<?> configBuilder = e.getValue().get();
        for (Field f : c.getDeclaredFields()) {
            // type contains one of the supported types.
            if (Modifier.isStatic(f.getModifiers()) && f.getType().isAssignableFrom(Property.class) && isSupportedType(f.getGenericType().getTypeName())) {
                Property p = (Property) f.get(null);
                if (p.getDefaultValue() != null && p.getDefaultValue() instanceof Boolean) {
                    configBuilder.with(p, nextValue.incrementAndGet() % 2 == 0);
                } else {
                    // Any number can be interpreted as a string or number.
                    configBuilder.with(p, Integer.toString(nextValue.incrementAndGet()));
                }
            }
        }
        // Collect the built config object for later use.
        expectedValues.put(c, configBuilder.build());
        // Include the builder in the main builder.
        b.include(configBuilder);
    }
    // Create the ServiceBuilderConfig, and verify that the created Config classes (using getConfig()) match the
    // expected ones.
    val builderConfig = b.build();
    for (Map.Entry<Class<?>, Supplier<ConfigBuilder<?>>> e : testClasses.entrySet()) {
        Class<?> c = e.getKey();
        Object expectedConfig = expectedValues.get(c);
        Object actualConfig = builderConfig.getConfig(e.getValue());
        // All the properties we care about are public getters with no arguments - only check those.
        for (Method m : c.getDeclaredMethods()) {
            if (m.getName().startsWith("get") && m.getParameterCount() == 0 && !Modifier.isStatic(m.getModifiers()) && Modifier.isPublic(m.getModifiers())) {
                Object expectedValue = m.invoke(expectedConfig);
                Object actualValue = m.invoke(actualConfig);
                if (expectedValue == null) {
                    Assert.assertNull("Expected a null value for " + getPropName(c, m), actualValue);
                } else {
                    Assert.assertNotNull("Not expected a null value for " + getPropName(c, m), actualValue);
                }
                if (isSupportedType(expectedValue)) {
                    Assert.assertEquals("Unexpected value for " + getPropName(c, m), expectedValue, actualValue);
                }
            }
        }
    }
}

Example 2 with ConfigBuilder

use of io.pravega.common.util.ConfigBuilder in project pravega by pravega.

the class ServiceBuilderConfigTests method testBuilder.

/**
 * Tests the Builder for ServiceBuilderConfig.
 */
@Test
public void testBuilder() throws Exception {
    final int propertyCount = 20;
    // Create three raw Properties.
    val p1 = buildProperties(0, propertyCount / 2, 0);
    val p2 = buildProperties(propertyCount / 4, propertyCount / 2, 100);
    val p3 = buildProperties(propertyCount / 2, propertyCount / 2, 1000);
    // Save #1 to a temp file.
    @Cleanup("delete") File tmpFile = File.createTempFile("pravega-", "-servicebuildertest");
    try (val writer = new FileWriter(tmpFile)) {
        p1.store(writer, "");
    }
    // Create a builder and load up #1 from file
    val b = ServiceBuilderConfig.builder().include(tmpFile.getPath());
    val expected = new Properties();
    expected.putAll(p1);
    verifyContents("Unexpected contents after: file. ", expected, b);
    // Include #2.
    b.include(p2);
    expected.putAll(p2);
    verifyContents("Unexpected contents after: file + Properties.", expected, b);
    // Include a test config builder with #3.
    b.include(new ConfigBuilderSupplier<>(p3, s -> s));
    expected.putAll(p3);
    verifyContents("Unexpected contents after: file + Properties + ConfigBuilder.", expected, b);
}