Examples with DatastoreHelper com.newrelic.agent.introspec.DatastoreHelper used on opensource projects

Example 66 with DatastoreHelper

use of com.newrelic.agent.introspec.DatastoreHelper in project newrelic-java-agent by newrelic.

the class MongoDbAsync380Test method testCollectionApi.

@Test
public void testCollectionApi() throws Exception {
    runMongoDbAsyncQuickStart();
    Introspector introspector = InstrumentationTestRunner.getIntrospector();
    assertEquals(1, introspector.getFinishedTransactionCount(1000));
    DatastoreHelper helper = new DatastoreHelper(MONGODB_PRODUCT);
    helper.assertAggregateMetrics();
    Collection<String> transactionNames = InstrumentationTestRunner.getIntrospector().getTransactionNames();
    assertEquals(1, transactionNames.size());
    String txName = transactionNames.iterator().next();
    // Counts based on operations executed in MongoDbAsyncQuickStart
    int insertOpExpectedCount = 1;
    int insertManyOpExpectedCount = 1;
    int updateOpExpectedCount = 1;
    int updateManyOpExpectedCount = 1;
    int deleteOpExpectedCount = 1;
    int deleteManyOpExpectedCount = 1;
    int findOpExpectedCount = 14;
    int dropOpExpectedCount = 3;
    int countOpExpectedCount = 1;
    int bulkWriteOpExpectedCount = 2;
    helper.assertUnifiedMetricCounts(txName, "insert", "test", insertOpExpectedCount);
    helper.assertUnifiedMetricCounts(txName, "insertMany", "test", insertManyOpExpectedCount);
    helper.assertUnifiedMetricCounts(txName, "update", "test", updateOpExpectedCount);
    helper.assertUnifiedMetricCounts(txName, "updateMany", "test", updateManyOpExpectedCount);
    helper.assertUnifiedMetricCounts(txName, "delete", "test", deleteOpExpectedCount);
    helper.assertUnifiedMetricCounts(txName, "deleteMany", "test", deleteManyOpExpectedCount);
    helper.assertUnifiedMetricCounts(txName, "find", "test", findOpExpectedCount);
    helper.assertUnifiedMetricCounts(txName, "drop", "test", dropOpExpectedCount);
    helper.assertUnifiedMetricCounts(txName, "count", "test", countOpExpectedCount);
    helper.assertUnifiedMetricCounts(txName, "bulkWrite", "test", bulkWriteOpExpectedCount);
    int totalOpCount = insertOpExpectedCount + insertManyOpExpectedCount + updateOpExpectedCount + updateManyOpExpectedCount + deleteOpExpectedCount + deleteManyOpExpectedCount + findOpExpectedCount + dropOpExpectedCount + countOpExpectedCount + bulkWriteOpExpectedCount;
    // Should be equal to the sum of all above metric counts
    assertEquals(totalOpCount, MetricsHelper.getUnscopedMetricCount("Datastore/all"));
    assertEquals(totalOpCount, MetricsHelper.getUnscopedMetricCount("Datastore/MongoDB/allOther"));
    assertEquals(totalOpCount, introspector.getTransactionEvents(txName).iterator().next().getDatabaseCallCount());
}

Example 67 with DatastoreHelper

use of com.newrelic.agent.introspec.DatastoreHelper in project newrelic-java-agent by newrelic.

the class MongoDbAsync370Test method testCollectionApi.

@Test
public void testCollectionApi() throws Exception {
    runMongoDbAsyncQuickStart();
    Introspector introspector = InstrumentationTestRunner.getIntrospector();
    assertEquals(1, introspector.getFinishedTransactionCount(1000));
    DatastoreHelper helper = new DatastoreHelper(MONGODB_PRODUCT);
    helper.assertAggregateMetrics();
    Collection<String> transactionNames = InstrumentationTestRunner.getIntrospector().getTransactionNames();
    assertEquals(1, transactionNames.size());
    String txName = transactionNames.iterator().next();
    // Counts based on operations executed in MongoDbAsyncQuickStart
    int insertOpExpectedCount = 1;
    int insertManyOpExpectedCount = 1;
    int updateOpExpectedCount = 1;
    int updateManyOpExpectedCount = 1;
    int deleteOpExpectedCount = 1;
    int deleteManyOpExpectedCount = 1;
    int findOpExpectedCount = 14;
    int dropOpExpectedCount = 3;
    int countOpExpectedCount = 1;
    int bulkWriteOpExpectedCount = 2;
    helper.assertUnifiedMetricCounts(txName, "insert", "test", insertOpExpectedCount);
    helper.assertUnifiedMetricCounts(txName, "insertMany", "test", insertManyOpExpectedCount);
    helper.assertUnifiedMetricCounts(txName, "update", "test", updateOpExpectedCount);
    helper.assertUnifiedMetricCounts(txName, "updateMany", "test", updateManyOpExpectedCount);
    helper.assertUnifiedMetricCounts(txName, "delete", "test", deleteOpExpectedCount);
    helper.assertUnifiedMetricCounts(txName, "deleteMany", "test", deleteManyOpExpectedCount);
    helper.assertUnifiedMetricCounts(txName, "find", "test", findOpExpectedCount);
    helper.assertUnifiedMetricCounts(txName, "drop", "test", dropOpExpectedCount);
    helper.assertUnifiedMetricCounts(txName, "count", "test", countOpExpectedCount);
    helper.assertUnifiedMetricCounts(txName, "bulkWrite", "test", bulkWriteOpExpectedCount);
    int totalOpCount = insertOpExpectedCount + insertManyOpExpectedCount + updateOpExpectedCount + updateManyOpExpectedCount + deleteOpExpectedCount + deleteManyOpExpectedCount + findOpExpectedCount + dropOpExpectedCount + countOpExpectedCount + bulkWriteOpExpectedCount;
    // Should be equal to the sum of all above metric counts
    assertEquals(totalOpCount, MetricsHelper.getUnscopedMetricCount("Datastore/all"));
    assertEquals(totalOpCount, MetricsHelper.getUnscopedMetricCount("Datastore/MongoDB/allOther"));
    assertEquals(totalOpCount, introspector.getTransactionEvents(txName).iterator().next().getDatabaseCallCount());
}

Example 68 with DatastoreHelper

use of com.newrelic.agent.introspec.DatastoreHelper in project newrelic-java-agent by newrelic.

the class MariadbInstrumentedTest method testSelect.

@Test
public void testSelect() {
    // Given
    Introspector introspector = InstrumentationTestRunner.getIntrospector();
    DatastoreHelper helper = new DatastoreHelper("MariaDB");
    // When
    R2dbcTestUtils.basicRequests(connection);
    // Then
    assertEquals(1, introspector.getFinishedTransactionCount(1000));
    assertEquals(1, introspector.getTransactionNames().size());
    String transactionName = introspector.getTransactionNames().stream().findFirst().orElse("");
    helper.assertScopedStatementMetricCount(transactionName, "INSERT", "USERS", 1);
    helper.assertScopedStatementMetricCount(transactionName, "SELECT", "USERS", 3);
    helper.assertScopedStatementMetricCount(transactionName, "UPDATE", "USERS", 1);
    helper.assertScopedStatementMetricCount(transactionName, "DELETE", "USERS", 1);
    helper.assertAggregateMetrics();
    helper.assertUnscopedOperationMetricCount("INSERT", 1);
    helper.assertUnscopedOperationMetricCount("SELECT", 3);
    helper.assertUnscopedOperationMetricCount("UPDATE", 1);
    helper.assertUnscopedOperationMetricCount("DELETE", 1);
    helper.assertUnscopedStatementMetricCount("INSERT", "USERS", 1);
    helper.assertUnscopedStatementMetricCount("SELECT", "USERS", 3);
    helper.assertUnscopedStatementMetricCount("UPDATE", "USERS", 1);
    helper.assertUnscopedStatementMetricCount("DELETE", "USERS", 1);
}

Example 69 with DatastoreHelper

use of com.newrelic.agent.introspec.DatastoreHelper in project newrelic-java-agent by newrelic.

the class H2InstrumentedTest method testSelect.

@Test
public void testSelect() {
    // Given
    Introspector introspector = InstrumentationTestRunner.getIntrospector();
    DatastoreHelper helper = new DatastoreHelper(DatastoreVendor.H2.toString());
    // When
    R2dbcTestUtils.basicRequests(connection);
    // Then
    assertEquals(1, introspector.getFinishedTransactionCount(1000));
    assertEquals(1, introspector.getTransactionNames().size());
    String transactionName = introspector.getTransactionNames().stream().findFirst().orElse("");
    helper.assertScopedStatementMetricCount(transactionName, "INSERT", "USERS", 1);
    helper.assertScopedStatementMetricCount(transactionName, "SELECT", "USERS", 3);
    helper.assertScopedStatementMetricCount(transactionName, "UPDATE", "USERS", 1);
    helper.assertScopedStatementMetricCount(transactionName, "DELETE", "USERS", 1);
    helper.assertAggregateMetrics();
    helper.assertUnscopedOperationMetricCount("INSERT", 1);
    helper.assertUnscopedOperationMetricCount("SELECT", 3);
    helper.assertUnscopedOperationMetricCount("UPDATE", 1);
    helper.assertUnscopedOperationMetricCount("DELETE", 1);
    helper.assertUnscopedStatementMetricCount("INSERT", "USERS", 1);
    helper.assertUnscopedStatementMetricCount("SELECT", "USERS", 3);
    helper.assertUnscopedStatementMetricCount("UPDATE", "USERS", 1);
    helper.assertUnscopedStatementMetricCount("DELETE", "USERS", 1);
}

Example 70 with DatastoreHelper

use of com.newrelic.agent.introspec.DatastoreHelper in project newrelic-java-agent by newrelic.

the class PostgresqlInstrumentedTest method testBasicRequests.

@Test
public void testBasicRequests() {
    // Given
    Introspector introspector = InstrumentationTestRunner.getIntrospector();
    DatastoreHelper helper = new DatastoreHelper("Postgres");
    // When
    R2dbcTestUtils.basicRequests(connection);
    // Then
    assertEquals(1, introspector.getFinishedTransactionCount(1000));
    assertEquals(1, introspector.getTransactionNames().size());
    String transactionName = introspector.getTransactionNames().stream().findFirst().orElse("");
    helper.assertScopedStatementMetricCount(transactionName, "INSERT", "USERS", 1);
    helper.assertScopedStatementMetricCount(transactionName, "SELECT", "USERS", 3);
    helper.assertScopedStatementMetricCount(transactionName, "UPDATE", "USERS", 1);
    helper.assertScopedStatementMetricCount(transactionName, "DELETE", "USERS", 1);
    helper.assertAggregateMetrics();
    helper.assertUnscopedOperationMetricCount("INSERT", 1);
    helper.assertUnscopedOperationMetricCount("SELECT", 3);
    helper.assertUnscopedOperationMetricCount("UPDATE", 1);
    helper.assertUnscopedOperationMetricCount("DELETE", 1);
    helper.assertUnscopedStatementMetricCount("INSERT", "USERS", 1);
    helper.assertUnscopedStatementMetricCount("SELECT", "USERS", 3);
    helper.assertUnscopedStatementMetricCount("UPDATE", "USERS", 1);
    helper.assertUnscopedStatementMetricCount("DELETE", "USERS", 1);
}