@Override protected void setUp() throws Exception { super.setUp(); m_ds = m_ezMock.createMock(DataSource.class); m_ds2 = m_ezMock.createMock(DataSource.class); m_conn = m_ezMock.createMock(Connection.class); m_conn2 = m_ezMock.createMock(Connection.class); DataSourceFactory.setInstance("ds", m_ds); DataSourceFactory.setInstance("ds2", m_ds2); }
public void testCloseResources() throws Exception { EasyMock.expect(m_ds.getConnection()).andReturn(m_conn); m_conn.setAutoCommit(false); Statement stmt = m_ezMock.createMock(Statement.class); ResultSet rs = m_ezMock.createMock(ResultSet.class); rs.close(); stmt.close(); m_conn.close(); m_conn.commit(); m_ezMock.replayAll(); Transaction.begin(); Transaction.getConnection("ds"); Transaction.register(stmt); Transaction.register(rs); Transaction.end(); m_ezMock.verifyAll(); }
private void setupCollector(String svcName, ServiceCollector svcCollector) throws CollectionInitializationException { MockServiceCollector.setDelegate(svcCollector); // Tell the config to use the MockServiceCollector for the specified service Collector collector = new Collector(); collector.setService(svcName); collector.setClassName(MockServiceCollector.class.getName()); m_collectdConfigFactory = m_easyMockUtils.createMock(CollectdConfigFactory.class); m_collectdConfig = m_easyMockUtils.createMock(CollectdConfiguration.class); expect(m_collectdConfigFactory.getCollectdConfig()).andReturn(m_collectdConfig).anyTimes(); expect(m_collectdConfig.getCollectors()).andReturn(Collections.singletonList(collector)).anyTimes(); expect(m_collectdConfig.getThreads()).andReturn(1).anyTimes(); m_collectd.setCollectdConfigFactory(m_collectdConfigFactory); } }
parameters.put("thresholding-enabled", "true"); FilterDao filterDao = m_mocks.createMock(FilterDao.class); expect(filterDao.getActiveIPAddressList((String)EasyMock.anyObject())).andReturn(Collections.singletonList(addr("127.0.0.1"))).anyTimes(); filterDao.flushActiveIpAddressListCache(); FilterDaoFactory.setInstance(filterDao); MonitoredService svc = m_mocks.createMock(MonitoredService.class); expect(svc.getNodeId()).andReturn(1); expect(svc.getIpAddr()).andReturn("127.0.0.1").atLeastOnce();
/** * Add a dummy transaction manager that has mock calls to commit() and rollback() */ private void setupTransactionManager() { PlatformTransactionManager m_transactionManager = m_easyMockUtils.createMock(PlatformTransactionManager.class); TransactionTemplate transactionTemplate = new TransactionTemplate(m_transactionManager); m_collectd.setTransactionTemplate(transactionTemplate); expect(m_transactionManager.getTransaction(isA(TransactionDefinition.class))).andReturn(new SimpleTransactionStatus()).anyTimes(); m_transactionManager.rollback(isA(TransactionStatus.class)); expectLastCall().anyTimes(); m_transactionManager.commit(isA(TransactionStatus.class)); expectLastCall().anyTimes(); }
@SuppressWarnings("unchecked") private void setupCollector(String svcName) throws CollectionInitializationException { ServiceCollector svcCollector = m_easyMockUtils.createMock(ServiceCollector.class); svcCollector.initialize(); svcCollector.validateAgent(isA(CollectionAgent.class), isA(Map.class)); expectLastCall().anyTimes(); setupCollector(svcName, svcCollector); }
m_filterDao = m_mockUtils.createMock(FilterDao.class); FilterDaoFactory.setInstance(m_filterDao); collector.addParameter(param); m_collectdConfigFactory = m_mockUtils.createMock(CollectdConfigFactory.class); m_collectdConfiguration = m_mockUtils.createMock(CollectdConfiguration.class); EasyMock.expect(m_collectdConfigFactory.getCollectdConfig()).andReturn(m_collectdConfiguration).anyTimes(); EasyMock.expect(m_collectdConfiguration.getCollectors()).andReturn(Collections.singletonList(collector)).anyTimes(); EasyMock.expect(m_collectdConfiguration.getThreads()).andReturn(1).anyTimes(); m_ifaceDao = m_mockUtils.createMock(IpInterfaceDao.class); m_nodeDao = m_mockUtils.createMock(NodeDao.class);
m_eventIpcManager = m_easyMockUtils.createMock(EventIpcManager.class); EventIpcManagerFactory.setIpcManager(m_eventIpcManager); m_nodeDao = m_easyMockUtils.createMock(NodeDao.class); m_ipIfDao = m_easyMockUtils.createMock(IpInterfaceDao.class); m_scheduler = new MockScheduler();
@Before // Cannot avoid this warning since there is no way to fetch the class object for an interface // that uses generics public void setUp() throws Exception { BeanUtils.setStaticApplicationContext(m_context); m_mocks = new EasyMockUtils(); m_pollerConfig = m_mocks.createMock(PollerConfig.class); MockLogAppender.setupLogging(); String previousOpennmsHome = System.setProperty("opennms.home", "src/test/resources"); PollOutagesConfigFactory.init(); System.setProperty("opennms.home", previousOpennmsHome); MockNetwork network = new MockNetwork(); network.setCriticalService("ICMP"); network.addNode(1, "testNode"); network.addInterface("127.0.0.1"); network.setIfAlias("eth0"); network.addService("ICMP"); network.addService("SNMP"); m_db.populate(network); }
m_filterDao = m_mockUtils.createMock(FilterDao.class); FilterDaoFactory.setInstance(m_filterDao); collector.setClassName(MockServiceCollector.class.getName()); m_collectdConfigFactory = m_mockUtils.createMock(CollectdConfigFactory.class); m_collectdConfiguration = m_mockUtils.createMock(CollectdConfiguration.class); EasyMock.expect(m_collectdConfigFactory.getCollectdConfig()).andReturn(m_collectdConfiguration).anyTimes(); EasyMock.expect(m_collectdConfiguration.getCollectors()).andReturn(Collections.singletonList(collector)).anyTimes(); EasyMock.expect(m_collectdConfiguration.getThreads()).andReturn(2).anyTimes(); m_ifaceDao = m_mockUtils.createMock(IpInterfaceDao.class); m_nodeDao = m_mockUtils.createMock(NodeDao.class);
@Before public void setUp() throws Exception { MockUtil.println("------------ Begin Test " + m_testName.getMethodName() + " --------------------------"); MockLogAppender.setupLogging(); m_fileAnticipator = new FileAnticipator(); m_rrdStrategy = new JRobinRrdStrategy(); m_resourceStorageDao = new FilesystemResourceStorageDao(); m_resourceStorageDao.setRrdDirectory(m_fileAnticipator.getTempDir()); m_intf = new OnmsIpInterface(); m_node = new OnmsNode(); m_node.setId(1); m_intf.setId(25); m_intf.setNode(m_node); m_intf.setIpAddress(InetAddressUtils.addr("1.1.1.1")); m_ifDao = m_easyMockUtils.createMock(IpInterfaceDao.class); m_serviceParams = new ServiceParameters(new HashMap<String,Object>()); }
@Before @Override public void setUp() throws Exception { setStartEventd(false); super.setUp(); m_mockAgent = MockSnmpAgent.createAgentAndRun(new ClassPathResource("org/opennms/netmgt/snmp/snmpTestData1.properties").getURL(), InetAddressUtils.str(myLocalHost()) + "/9161"); m_config = new MockDataCollectionConfig(); DataCollectionConfigFactory.setInstance(m_config); m_sysNameOid = SnmpObjId.get(".1.3.6.1.2.1.1.5"); m_ifOutOctets = SnmpObjId.get("..1.3.6.1.2.1.2.2.1.16"); m_invalid = SnmpObjId.get(".1.5.6.1.2.1.1.5"); m_ifDescr = SnmpObjId.get(".1.3.6.1.2.1.2.2.1.2"); m_easyMockUtils = new EasyMockUtils(); m_ifaceDao = m_easyMockUtils.createMock(IpInterfaceDao.class); createAgent(1, PrimaryType.PRIMARY); }
/** * NMS-9413: Verifies that collectd does not schedule interfaces when the * {@link ServiceCollector} throws a {@link CollectionInitializationException} * while validating the agent. */ @SuppressWarnings("unchecked") @Test public void testInterfaceIsNotScheduledWhenValidationFails() throws Exception { ServiceCollector svcCollector = m_easyMockUtils.createMock(ServiceCollector.class); svcCollector.initialize(); svcCollector.validateAgent(isA(CollectionAgent.class), isA(Map.class)); expectLastCall().andThrow(new CollectionInitializationException("No!")).once(); setupCollector("SNMP", svcCollector); OnmsIpInterface iface = getInterface(); setupInterface(iface); setupTransactionManager(); expect(m_collectdConfig.getPackages()).andReturn(Collections.singletonList(getCollectionPackageThatMatchesSNMP())); expect(m_collectdConfigFactory.interfaceInPackage(iface, getCollectionPackageThatMatchesSNMP())).andReturn(true); m_easyMockUtils.replayAll(); assertEquals("scheduler entry count", 0, m_scheduler.getEntryCount()); m_collectd.afterPropertiesSet(); m_collectd.start(); m_scheduler.next(); assertEquals("scheduler entry count", 0, m_scheduler.getEntryCount()); m_collectd.stop(); m_easyMockUtils.verifyAll(); }
@Test public void testCreate() throws Exception { setupCollector("SNMP"); setupTransactionManager(); // Use a mock scheduler to track calls to the Collectd scheduler Scheduler m_scheduler = m_easyMockUtils.createMock(Scheduler.class); m_collectd.setScheduler(m_scheduler); // Expect one task to be scheduled m_scheduler.schedule(eq(0L), isA(ReadyRunnable.class)); // Expect the scheduler to be started and stopped during Collectd // start() and stop() m_scheduler.start(); m_scheduler.stop(); m_easyMockUtils.replayAll(); // Initialize Collectd m_collectd.afterPropertiesSet(); // Start and stop collectd m_collectd.start(); m_collectd.stop(); m_easyMockUtils.verifyAll(); }