Examples with Config com.hortonworks.streamline.common.Config used on opensource projects

Example 16 with Config

use of com.hortonworks.streamline.common.Config in project streamline by hortonworks.

the class StormServiceRegistrarTest method testRegister.

@Test
public void testRegister() throws Exception {
    Cluster cluster = getTestCluster(1L);
    StormServiceRegistrar registrar = initializeServiceRegistrar();
    // explicit convert Object
    Config config = new Config();
    config.put(StormServiceRegistrar.PARAM_NIMBUS_SEEDS, "storm-1,storm-2");
    config.put(StormServiceRegistrar.PARAM_NIMBUS_THRIFT_PORT, (Object) 6627);
    config.put(StormServiceRegistrar.PARAM_UI_HOST, "storm-1");
    config.put(StormServiceRegistrar.PARAM_UI_PORT, (Object) 8080);
    config.put(StormServiceRegistrar.PARAM_NIMBUS_THRIFT_MAX_BUFFER_SIZE, (Object) 102476800);
    config.put(StormServiceRegistrar.PARAM_THRIFT_TRANSPORT, "org.apache.storm.security.auth.SimpleTransportPlugin");
    config.put(StormServiceRegistrar.PARAM_PRINCIPAL_TO_LOCAL, "org.apache.storm.security.auth.DefaultPrincipalToLocal");
    config.put(StormServiceRegistrar.PARAM_NIMBUS_PRINCIPAL_NAME, "nimbus/_HOST@EXAMPLE.COM");
    registrar.register(cluster, config, Collections.emptyList());
    Service stormService = environmentService.getServiceByName(cluster.getId(), Constants.Storm.SERVICE_NAME);
    assertNotNull(stormService);
    Component nimbus = environmentService.getComponentByName(stormService.getId(), ComponentPropertyPattern.NIMBUS.name());
    assertNotNull(nimbus);
    Collection<ComponentProcess> nimbusProcesses = environmentService.listComponentProcesses(nimbus.getId());
    List<String> hosts = nimbusProcesses.stream().map(ComponentProcess::getHost).collect(Collectors.toList());
    assertEquals(Sets.newHashSet("storm-1", "storm-2"), new HashSet<>(hosts));
    List<Integer> ports = nimbusProcesses.stream().map(ComponentProcess::getPort).collect(Collectors.toList());
    assertEquals(Sets.newHashSet(6627, 6627), new HashSet<>(ports));
    Component ui = environmentService.getComponentByName(stormService.getId(), ComponentPropertyPattern.STORM_UI_SERVER.name());
    assertNotNull(ui);
    Collection<ComponentProcess> uiProcesses = environmentService.listComponentProcesses(ui.getId());
    assertEquals(Sets.newHashSet("storm-1"), uiProcesses.stream().map(ComponentProcess::getHost).collect(Collectors.toSet()));
    assertEquals(Sets.newHashSet(8080), uiProcesses.stream().map(ComponentProcess::getPort).collect(Collectors.toSet()));
    ServiceConfiguration stormYamlConf = environmentService.getServiceConfigurationByName(stormService.getId(), CONFIGURATION_NAME_STORM_YAML);
    assertNotNull(stormYamlConf);
    Map<String, String> stormYamlConfMap = stormYamlConf.getConfigurationMap();
    assertEquals(config.getAny(StormServiceRegistrar.PARAM_NIMBUS_THRIFT_MAX_BUFFER_SIZE), Integer.valueOf(stormYamlConfMap.get(StormServiceRegistrar.PARAM_NIMBUS_THRIFT_MAX_BUFFER_SIZE)));
    assertEquals(config.get(StormServiceRegistrar.PARAM_THRIFT_TRANSPORT), stormYamlConfMap.get(StormServiceRegistrar.PARAM_THRIFT_TRANSPORT));
    assertEquals(config.get(StormServiceRegistrar.PARAM_PRINCIPAL_TO_LOCAL), stormYamlConfMap.get(StormServiceRegistrar.PARAM_PRINCIPAL_TO_LOCAL));
    ServiceConfiguration stormEnvConf = environmentService.getServiceConfigurationByName(stormService.getId(), CONFIGURATION_NAME_STORM_ENV);
    assertNotNull(stormEnvConf);
    Map<String, String> stormEnvConfMap = stormEnvConf.getConfigurationMap();
    assertEquals(config.get(StormServiceRegistrar.PARAM_NIMBUS_PRINCIPAL_NAME), stormEnvConfMap.get(StormServiceRegistrar.PARAM_NIMBUS_PRINCIPAL_NAME));
}

Example 17 with Config

use of com.hortonworks.streamline.common.Config in project streamline by hortonworks.

the class StormServiceRegistrarTest method testRegister_component_storm_ui_server_notPresent.

@Test
public void testRegister_component_storm_ui_server_notPresent() throws Exception {
    Cluster cluster = getTestCluster(1L);
    StormServiceRegistrar registrar = initializeServiceRegistrar();
    try {
        Config config = new Config();
        config.put(StormServiceRegistrar.PARAM_NIMBUS_SEEDS, "storm-1,storm-2");
        config.put(StormServiceRegistrar.PARAM_NIMBUS_THRIFT_PORT, (Object) 6627);
        // no ui params
        registrar.register(cluster, config, Collections.emptyList());
        fail("Should throw IllegalArgumentException");
    } catch (IllegalArgumentException e) {
        // OK
        Service stormService = environmentService.getServiceByName(cluster.getId(), Constants.Storm.SERVICE_NAME);
        assertNull(stormService);
    }
}

Example 18 with Config

use of com.hortonworks.streamline.common.Config in project streamline by hortonworks.

the class ZookeeperServiceRegistrarTest method testRegister.

@Test
public void testRegister() throws Exception {
    Cluster cluster = getTestCluster(1L);
    ZookeeperServiceRegistrar registrar = initializeServiceRegistrar();
    Config config = new Config();
    config.put(ZookeeperServiceRegistrar.PARAM_ZOOKEEPER_SERVER_HOSTNAMES, Lists.newArrayList("zookeeper-1", "zookeeper-2"));
    config.put(ZookeeperServiceRegistrar.PARAM_ZOOKEEPER_PORT, (Object) 2181);
    registrar.register(cluster, config, Collections.emptyList());
    Service zkService = environmentService.getServiceByName(cluster.getId(), Constants.Zookeeper.SERVICE_NAME);
    assertNotNull(zkService);
    Component zkServer = environmentService.getComponentByName(zkService.getId(), ComponentPropertyPattern.ZOOKEEPER_SERVER.name());
    assertNotNull(zkServer);
    Collection<ComponentProcess> zkServerProcesses = environmentService.listComponentProcesses(zkServer.getId());
    assertEquals(Sets.newHashSet("zookeeper-1", "zookeeper-2"), zkServerProcesses.stream().map(ComponentProcess::getHost).collect(Collectors.toSet()));
    assertEquals(Sets.newHashSet(2181, 2181), zkServerProcesses.stream().map(ComponentProcess::getPort).collect(Collectors.toSet()));
    ServiceConfiguration zooConf = environmentService.getServiceConfigurationByName(zkService.getId(), CONFIGURATION_NAME_ZOO_CFG);
    assertNotNull(zooConf);
}

Example 19 with Config

use of com.hortonworks.streamline.common.Config in project streamline by hortonworks.

the class TestTopologyDagCreatingVisitor method visit.

@Override
public void visit(RulesProcessor rulesProcessor) {
    String id = rulesProcessor.getId();
    String processorName = rulesProcessor.getName();
    if (!testRunRulesProcessorsForEachProcessor.containsKey(processorName)) {
        throw new IllegalStateException("Not all processors have corresponding TestRunRulesProcessor instance. processor name: " + processorName);
    }
    Config config = new Config(rulesProcessor.getConfig());
    TestRunRulesProcessor testRunRulesProcessor = testRunRulesProcessorsForEachProcessor.get(processorName);
    testRunRulesProcessor.setId(id);
    testRunRulesProcessor.setName(processorName);
    testRunRulesProcessor.setConfig(config);
    testRunRulesProcessor.setTransformationClass(TestRunRulesProcessorBoltFluxComponent.class.getName());
    testTopologyDag.add(testRunRulesProcessor);
    processorToReplacedTestProcessorMap.put(processorName, testRunRulesProcessor);
    copyEdges(rulesProcessor);
}

Example 20 with Config

use of com.hortonworks.streamline.common.Config in project streamline by hortonworks.

the class TestTopologyDagCreatingVisitor method visit.

@Override
public void visit(StreamlineProcessor processor) {
    String id = processor.getId();
    String processorName = processor.getName();
    if (!testRunProcessorsForEachProcessor.containsKey(processorName)) {
        throw new IllegalStateException("Not all processors have corresponding TestRunProcessor instance. processor name: " + processorName);
    }
    Config config = new Config(processor.getConfig());
    TestRunProcessor testRunProcessor = testRunProcessorsForEachProcessor.get(processorName);
    testRunProcessor.setId(id);
    testRunProcessor.setName(processorName);
    testRunProcessor.setConfig(config);
    testRunProcessor.setTransformationClass(TestRunProcessorBoltFluxComponent.class.getName());
    testTopologyDag.add(testRunProcessor);
    processorToReplacedTestProcessorMap.put(processorName, testRunProcessor);
    copyEdges(processor);
}