Fixed test

DataDog · Jun 24, 2024 · c154a26 · c154a26
1 parent b3305e9
commit c154a26
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/internal-api/src/test/java/datadog/trace/api/gateway/InstrumentationGatewayTest.java b/internal-api/src/test/java/datadog/trace/api/gateway/InstrumentationGatewayTest.java
@@ -201,9 +201,9 @@ public void testNormalCalls() {
     assertThat(cbp.getCallback(events.graphqlServerRequestMessage()).apply(null, null).getAction())
         .isEqualTo(Flow.Action.Noop.INSTANCE);
     ss.registerCallback(events.databaseConnection(), callback);
-    cbp.getCallback(events.databaseConnection()).accept(null, null);
+    cbp.getCallback(events.databaseConnection()).apply(null, null);
     ss.registerCallback(events.databaseSqlQuery(), callback);
-    cbp.getCallback(events.databaseSqlQuery()).accept(null, null);
+    cbp.getCallback(events.databaseSqlQuery()).apply(null, null);
     assertThat(callback.count).isEqualTo(Events.MAX_EVENTS);
   }
 
@@ -257,9 +257,9 @@ public void testThrowableBlocking() {
     assertThat(cbp.getCallback(events.graphqlServerRequestMessage()).apply(null, null).getAction())
         .isEqualTo(Flow.Action.Noop.INSTANCE);
     ss.registerCallback(events.databaseConnection(), throwback);
-    cbp.getCallback(events.databaseConnection()).accept(null, null);
+    cbp.getCallback(events.databaseConnection()).apply(null, null);
     ss.registerCallback(events.databaseSqlQuery(), throwback);
-    cbp.getCallback(events.databaseSqlQuery()).accept(null, null);
+    cbp.getCallback(events.databaseSqlQuery()).apply(null, null);
     assertThat(throwback.count).isEqualTo(Events.MAX_EVENTS);
   }