Search in sources :

Example 1 with MockRpcClientFactory

use of org.opennms.core.rpc.mock.MockRpcClientFactory in project opennms by OpenNMS.

the class CollectorTestUtils method createLocationAwareCollectorClient.

static LocationAwareCollectorClient createLocationAwareCollectorClient() {
    final DefaultServiceCollectorRegistry serviceCollectorRegistry = new DefaultServiceCollectorRegistry();
    final CollectorClientRpcModule collectorClientRpcModule = new CollectorClientRpcModule();
    collectorClientRpcModule.setServiceCollectorRegistry(serviceCollectorRegistry);
    collectorClientRpcModule.setExecutor(Executors.newSingleThreadExecutor());
    final MockRpcClientFactory rpcClientFactory = new MockRpcClientFactory();
    final LocationAwareCollectorClientImpl locationAwareCollectorClient = new LocationAwareCollectorClientImpl(rpcClientFactory);
    locationAwareCollectorClient.setRpcModule(collectorClientRpcModule);
    locationAwareCollectorClient.afterPropertiesSet();
    return locationAwareCollectorClient;
}
Also used : DefaultServiceCollectorRegistry(org.opennms.netmgt.collection.support.DefaultServiceCollectorRegistry) LocationAwareCollectorClientImpl(org.opennms.netmgt.collection.client.rpc.LocationAwareCollectorClientImpl) CollectorClientRpcModule(org.opennms.netmgt.collection.client.rpc.CollectorClientRpcModule) MockRpcClientFactory(org.opennms.core.rpc.mock.MockRpcClientFactory)

Example 2 with MockRpcClientFactory

use of org.opennms.core.rpc.mock.MockRpcClientFactory in project opennms by OpenNMS.

the class CollectorTestUtils method createLocationAwareCollectorClient.

public static LocationAwareCollectorClient createLocationAwareCollectorClient() {
    final DefaultServiceCollectorRegistry serviceCollectorRegistry = new DefaultServiceCollectorRegistry();
    final CollectorClientRpcModule collectorClientRpcModule = new CollectorClientRpcModule();
    collectorClientRpcModule.setServiceCollectorRegistry(serviceCollectorRegistry);
    collectorClientRpcModule.setExecutor(Executors.newSingleThreadExecutor());
    final MockRpcClientFactory rpcClientFactory = new MockRpcClientFactory();
    final LocationAwareCollectorClientImpl locationAwareCollectorClient = new LocationAwareCollectorClientImpl(rpcClientFactory);
    locationAwareCollectorClient.setRpcModule(collectorClientRpcModule);
    locationAwareCollectorClient.setRpcTargetHelper(new RpcTargetHelper());
    locationAwareCollectorClient.afterPropertiesSet();
    return locationAwareCollectorClient;
}
Also used : DefaultServiceCollectorRegistry(org.opennms.netmgt.collection.support.DefaultServiceCollectorRegistry) LocationAwareCollectorClientImpl(org.opennms.netmgt.collection.client.rpc.LocationAwareCollectorClientImpl) RpcTargetHelper(org.opennms.core.rpc.utils.RpcTargetHelper) CollectorClientRpcModule(org.opennms.netmgt.collection.client.rpc.CollectorClientRpcModule) MockRpcClientFactory(org.opennms.core.rpc.mock.MockRpcClientFactory)

Example 3 with MockRpcClientFactory

use of org.opennms.core.rpc.mock.MockRpcClientFactory in project opennms by OpenNMS.

the class PersistRegexSelectorStrategyTest method setUp.

@Before
public void setUp() throws Exception {
    ipInterfaceDao = EasyMock.createMock(IpInterfaceDao.class);
    String localhost = InetAddress.getLocalHost().getHostAddress();
    NetworkBuilder builder = new NetworkBuilder();
    builder.addNode("myNode");
    builder.addInterface(localhost).setIsManaged("M").setIsSnmpPrimary("P");
    OnmsNode node = builder.getCurrentNode();
    node.setId(1);
    OnmsIpInterface ipInterface = node.getIpInterfaces().iterator().next();
    EasyMock.expect(ipInterfaceDao.load(1)).andReturn(ipInterface).anyTimes();
    EasyMock.replay(ipInterfaceDao);
    Package pkg = new Package();
    pkg.setName("junitTestPackage");
    Filter filter = new Filter();
    filter.setContent("IPADDR != '0.0.0.0'");
    pkg.setFilter(filter);
    Service service = new Service();
    service.setName("SNMP");
    pkg.addService(service);
    Map<String, Object> map = new TreeMap<String, Object>();
    List<org.opennms.netmgt.config.collectd.Parameter> params = pkg.getService("SNMP").getParameters();
    for (org.opennms.netmgt.config.collectd.Parameter p : params) {
        map.put(p.getKey(), p.getValue());
    }
    map.put("collection", "default");
    serviceParams = new ServiceParameters(map);
    LocationAwareSnmpClient locationAwareSnmpClient = new LocationAwareSnmpClientRpcImpl(new MockRpcClientFactory());
    PlatformTransactionManager ptm = new MockPlatformTransactionManager();
    SnmpCollectionAgent agent = DefaultSnmpCollectionAgent.create(1, ipInterfaceDao, ptm);
    OnmsSnmpCollection snmpCollection = new OnmsSnmpCollection(agent, serviceParams, new MockDataCollectionConfigDao(), locationAwareSnmpClient);
    org.opennms.netmgt.config.datacollection.ResourceType rt = new org.opennms.netmgt.config.datacollection.ResourceType();
    rt.setName("myResourceType");
    StorageStrategy storageStrategy = new StorageStrategy();
    storageStrategy.setClazz("org.opennms.netmgt.collection.support.IndexStorageStrategy");
    rt.setStorageStrategy(storageStrategy);
    PersistenceSelectorStrategy persistenceSelectorStrategy = new PersistenceSelectorStrategy();
    persistenceSelectorStrategy.setClazz("org.opennms.netmgt.collectd.PersistRegexSelectorStrategy");
    Parameter param = new Parameter();
    param.setKey(PersistRegexSelectorStrategy.MATCH_EXPRESSION);
    param.setValue("#name matches '^agalue.*$'");
    persistenceSelectorStrategy.addParameter(param);
    rt.setPersistenceSelectorStrategy(persistenceSelectorStrategy);
    GenericIndexResourceType resourceType = new GenericIndexResourceType(agent, snmpCollection, rt);
    resourceA = new GenericIndexResource(resourceType, rt.getName(), new SnmpInstId("1.2.3.4.5.6.7.8.9.1.1"));
    AttributeGroupType groupType = new AttributeGroupType("mib2-interfaces", AttributeGroupType.IF_TYPE_ALL);
    MibObject mibObject = new MibObject();
    mibObject.setOid(".1.2.3.4.5.6.7.8.9.2.1");
    mibObject.setInstance("1");
    mibObject.setAlias("name");
    mibObject.setType("string");
    StringAttributeType attributeType = new StringAttributeType(resourceType, snmpCollection.getName(), mibObject, groupType);
    SnmpValue snmpValue = new Snmp4JValueFactory().getOctetString("agalue rules!".getBytes());
    resourceA.setAttributeValue(attributeType, snmpValue);
    resourceB = new GenericIndexResource(resourceType, rt.getName(), new SnmpInstId("1.2.3.4.5.6.7.8.9.1.2"));
}
Also used : StorageStrategy(org.opennms.netmgt.config.datacollection.StorageStrategy) Snmp4JValueFactory(org.opennms.netmgt.snmp.snmp4j.Snmp4JValueFactory) NetworkBuilder(org.opennms.netmgt.model.NetworkBuilder) LocationAwareSnmpClientRpcImpl(org.opennms.netmgt.snmp.proxy.common.LocationAwareSnmpClientRpcImpl) SnmpInstId(org.opennms.netmgt.snmp.SnmpInstId) MockPlatformTransactionManager(org.opennms.core.test.MockPlatformTransactionManager) MockDataCollectionConfigDao(org.opennms.netmgt.dao.mock.MockDataCollectionConfigDao) OnmsNode(org.opennms.netmgt.model.OnmsNode) Service(org.opennms.netmgt.config.collectd.Service) TreeMap(java.util.TreeMap) MockPlatformTransactionManager(org.opennms.core.test.MockPlatformTransactionManager) PlatformTransactionManager(org.springframework.transaction.PlatformTransactionManager) IpInterfaceDao(org.opennms.netmgt.dao.api.IpInterfaceDao) MockRpcClientFactory(org.opennms.core.rpc.mock.MockRpcClientFactory) LocationAwareSnmpClient(org.opennms.netmgt.snmp.proxy.LocationAwareSnmpClient) SnmpValue(org.opennms.netmgt.snmp.SnmpValue) OnmsIpInterface(org.opennms.netmgt.model.OnmsIpInterface) Filter(org.opennms.netmgt.config.collectd.Filter) AttributeGroupType(org.opennms.netmgt.collection.api.AttributeGroupType) Parameter(org.opennms.netmgt.config.datacollection.Parameter) MibObject(org.opennms.netmgt.config.datacollection.MibObject) Package(org.opennms.netmgt.config.collectd.Package) ServiceParameters(org.opennms.netmgt.collection.api.ServiceParameters) MibObject(org.opennms.netmgt.config.datacollection.MibObject) PersistenceSelectorStrategy(org.opennms.netmgt.config.datacollection.PersistenceSelectorStrategy) Before(org.junit.Before)

Aggregations

MockRpcClientFactory (org.opennms.core.rpc.mock.MockRpcClientFactory)3 CollectorClientRpcModule (org.opennms.netmgt.collection.client.rpc.CollectorClientRpcModule)2 LocationAwareCollectorClientImpl (org.opennms.netmgt.collection.client.rpc.LocationAwareCollectorClientImpl)2 DefaultServiceCollectorRegistry (org.opennms.netmgt.collection.support.DefaultServiceCollectorRegistry)2 TreeMap (java.util.TreeMap)1 Before (org.junit.Before)1 RpcTargetHelper (org.opennms.core.rpc.utils.RpcTargetHelper)1 MockPlatformTransactionManager (org.opennms.core.test.MockPlatformTransactionManager)1 AttributeGroupType (org.opennms.netmgt.collection.api.AttributeGroupType)1 ServiceParameters (org.opennms.netmgt.collection.api.ServiceParameters)1 Filter (org.opennms.netmgt.config.collectd.Filter)1 Package (org.opennms.netmgt.config.collectd.Package)1 Service (org.opennms.netmgt.config.collectd.Service)1 MibObject (org.opennms.netmgt.config.datacollection.MibObject)1 Parameter (org.opennms.netmgt.config.datacollection.Parameter)1 PersistenceSelectorStrategy (org.opennms.netmgt.config.datacollection.PersistenceSelectorStrategy)1 StorageStrategy (org.opennms.netmgt.config.datacollection.StorageStrategy)1 IpInterfaceDao (org.opennms.netmgt.dao.api.IpInterfaceDao)1 MockDataCollectionConfigDao (org.opennms.netmgt.dao.mock.MockDataCollectionConfigDao)1 NetworkBuilder (org.opennms.netmgt.model.NetworkBuilder)1