Search in sources :

Example 6 with RPMServiceManager

use of com.newrelic.agent.RPMServiceManager in project newrelic-java-agent by newrelic.

the class NoticeErrorAttributesTest method testStringifiesAndTruncates.

@Test
public void testStringifiesAndTruncates() {
    try {
        Map<String, Object> settings = new HashMap<>();
        settings.put("app_name", APP_NAME);
        manager.setConfigService(ConfigServiceFactory.createConfigServiceUsingSettings(settings));
        manager.setTransactionService(new TransactionService());
        manager.setTransactionTraceService(new TransactionTraceService());
        AttributesService service = new AttributesService();
        manager.setAttributesService(service);
        RPMServiceManager mockRPMServiceManager = manager.getRPMServiceManager();
        RPMService mockRPMService = mock(RPMService.class);
        ErrorService errorService = new ErrorServiceImpl(APP_NAME);
        when(mockRPMServiceManager.getRPMService()).thenReturn(mockRPMService);
        when(mockRPMService.getErrorService()).thenReturn(errorService);
        Transaction t = Transaction.getTransaction();
        BasicRequestRootTracer tracer = createDispatcherTracer();
        t.getTransactionActivity().tracerStarted(tracer);
        NewRelicApiImplementation impl = new NewRelicApiImplementation();
        Map<String, Object> atts = new HashMap<>();
        atts.put("test.foo", new Object() {

            @Override
            public String toString() {
                String base = "a";
                for (int i = 0; i < 10; i++) {
                    base += base;
                }
                return base;
            }
        });
        impl.noticeError("hello", atts);
        Assert.assertEquals(t.getErrorAttributes().keySet(), Sets.newHashSet("test.foo"));
        Assert.assertEquals(255, ((String) t.getErrorAttributes().get("test.foo")).length());
        Assert.assertNotEquals(255, atts.get("test.foo").toString().length());
    } finally {
        Transaction.clearTransaction();
    }
}
Also used : TransactionService(com.newrelic.agent.TransactionService) ErrorServiceImpl(com.newrelic.agent.errors.ErrorServiceImpl) HashMap(java.util.HashMap) RPMServiceManager(com.newrelic.agent.RPMServiceManager) NewRelicApiImplementation(com.newrelic.api.agent.NewRelicApiImplementation) TransactionTraceService(com.newrelic.agent.trace.TransactionTraceService) ErrorService(com.newrelic.agent.errors.ErrorService) Transaction(com.newrelic.agent.Transaction) RPMService(com.newrelic.agent.RPMService) BasicRequestRootTracer(com.newrelic.agent.tracers.servlet.BasicRequestRootTracer) Test(org.junit.Test)

Example 7 with RPMServiceManager

use of com.newrelic.agent.RPMServiceManager in project newrelic-java-agent by newrelic.

the class JfrServiceTest method jfrLoopDoesStart.

@Test
public void jfrLoopDoesStart() {
    JfrService jfrService = new JfrService(jfrConfig, agentConfig);
    JfrService spyJfr = spy(jfrService);
    when(spyJfr.coreApisExist()).thenReturn(true);
    when(spyJfr.isEnabled()).thenReturn(true);
    MockServiceManager manager = new MockServiceManager();
    ServiceFactory.setServiceManager(manager);
    RPMServiceManager mockRPMServiceManager = manager.getRPMServiceManager();
    RPMService mockRPMService = mock(RPMService.class);
    when(mockRPMServiceManager.getRPMService()).thenReturn(mockRPMService);
    when(mockRPMService.getEntityGuid()).thenReturn("test_guid");
    spyJfr.doStart();
    try {
        // The timeout wait is necessary because jfr loop is being executed on async thread.
        verify(spyJfr, timeout(100)).startJfrLoop();
        spyJfr.doStop();
    } catch (JfrRecorderException e) {
        fail("Should not have thrown any exception");
    }
}
Also used : JfrRecorderException(com.newrelic.jfr.daemon.JfrRecorderException) MockServiceManager(com.newrelic.agent.MockServiceManager) RPMService(com.newrelic.agent.RPMService) RPMServiceManager(com.newrelic.agent.RPMServiceManager) Java10IncompatibleTest(com.newrelic.test.marker.Java10IncompatibleTest) Test(org.junit.Test) Java7IncompatibleTest(com.newrelic.test.marker.Java7IncompatibleTest) Java9IncompatibleTest(com.newrelic.test.marker.Java9IncompatibleTest)

Aggregations

RPMServiceManager (com.newrelic.agent.RPMServiceManager)7 RPMService (com.newrelic.agent.RPMService)5 ErrorService (com.newrelic.agent.errors.ErrorService)5 Test (org.junit.Test)5 Transaction (com.newrelic.agent.Transaction)4 TransactionService (com.newrelic.agent.TransactionService)4 ErrorServiceImpl (com.newrelic.agent.errors.ErrorServiceImpl)4 TransactionTraceService (com.newrelic.agent.trace.TransactionTraceService)4 BasicRequestRootTracer (com.newrelic.agent.tracers.servlet.BasicRequestRootTracer)4 NewRelicApiImplementation (com.newrelic.api.agent.NewRelicApiImplementation)4 HashMap (java.util.HashMap)4 MockServiceManager (com.newrelic.agent.MockServiceManager)2 IRPMService (com.newrelic.agent.IRPMService)1 ServiceManager (com.newrelic.agent.service.ServiceManager)1 JfrRecorderException (com.newrelic.jfr.daemon.JfrRecorderException)1 Java10IncompatibleTest (com.newrelic.test.marker.Java10IncompatibleTest)1 Java7IncompatibleTest (com.newrelic.test.marker.Java7IncompatibleTest)1 Java9IncompatibleTest (com.newrelic.test.marker.Java9IncompatibleTest)1 HashSet (java.util.HashSet)1 Before (org.junit.Before)1