Examples with TransactionSegment com.newrelic.agent.trace.TransactionSegment used on opensource projects

Example 16 with TransactionSegment

use of com.newrelic.agent.trace.TransactionSegment in project newrelic-java-agent by newrelic.

the class DefaultSqlTracerTest method testHighSecurityErrorFinish.

@SuppressWarnings("unchecked")
@Test
public void testHighSecurityErrorFinish() throws Exception {
    TransactionTracerConfig ttConfig = mock(TransactionTracerConfig.class);
    // These are set to -1 to allow this tracer to always go through slow sql logic. Otherwise if the tracer
    // executes in less than 1 millisecond (e.g. 0 ms rounded) it won't be greater than the threshold (0) so
    // we won't run this through the slow sql code.
    when(ttConfig.getExplainThresholdInMillis()).thenReturn(-1d);
    when(ttConfig.getExplainThresholdInNanos()).thenReturn(-1d);
    when(ttConfig.getRecordSql()).thenReturn(RecordSql.obfuscated.name());
    when(ttConfig.isExplainEnabled()).thenReturn(true);
    when(ttConfig.isEnabled()).thenReturn(true);
    AgentConfig agentConfig = AgentHelper.mockAgentConfig(ttConfig);
    when(agentConfig.isHighSecurity()).thenReturn(true);
    when(agentConfig.getTransactionTracerConfig()).thenReturn(ttConfig);
    String inputSql = "select * from dudette where ssn = 123456789";
    String obfuscatedInputSql = "select * from dudette where ssn = ?";
    DefaultSqlTracer tracer = newTracer(inputSql);
    assertTrue(tracer.getAgentAttributes().isEmpty());
    tracer.finish(new RuntimeException());
    SqlObfuscator sqlObfuscator = ServiceFactory.getDatabaseService().getDefaultSqlObfuscator();
    TransactionSegment segment = new TransactionSegment(ttConfig, sqlObfuscator, 0, tracer);
    // exclusive_duration_millis, sql
    assertEquals(2, tracer.getAgentAttributes().size());
    // shouldn't be obfuscated yet
    assertEquals(inputSql, (String) tracer.getAgentAttributes().get("sql"));
    JSONArray json = (JSONArray) AgentHelper.serializeJSON(segment);
    Map params = (Map) json.get(3);
    // the name changes when it is obfuscated
    String sqlStatement = (String) params.get("sql_obfuscated");
    assertEquals(obfuscatedInputSql, sqlStatement);
    String rawSql = (String) params.get("sql");
    // raw sql should be null since high security is on
    assertNull(rawSql);
    // Ensure that we do not send up unobfuscated sql in the intrinsics
    Transaction transaction = Transaction.getTransaction(false);
    assertNotNull(transaction);
    Map<String, Object> intrinsics = transaction.getIntrinsicAttributes();
    assertNotNull(intrinsics);
    assertEquals(obfuscatedInputSql, intrinsics.get(SqlTracer.SQL_PARAMETER_NAME));
}