update dtests

palantir · Dec 12, 2024 · a690e73 · a690e73
1 parent 29505d4
commit a690e73
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/test/distributed/org/apache/cassandra/distributed/impl/Coordinator.java b/test/distributed/org/apache/cassandra/distributed/impl/Coordinator.java
@@ -68,7 +68,7 @@ public Future<SimpleQueryResult> asyncExecuteWithTracingWithResult(UUID sessionI
         return instance.async(() -> {
             try
             {
-                Tracing.instance.newSession(sessionId);
+                Tracing.instance.newSession(sessionId, Collections.EMPTY_MAP);
                 return executeInternal(query, consistencyLevelOrigin, boundValues);
             }
             finally

diff --git a/test/distributed/org/apache/cassandra/distributed/impl/Instance.java b/test/distributed/org/apache/cassandra/distributed/impl/Instance.java
@@ -369,7 +369,7 @@ public boolean allowOutgoingMessage(MessageOut messageOut, int id, InetAddress t
                 {
                     byte[] traceTypeBytes = (byte[]) messageOut.parameters.get(Tracing.TRACE_TYPE);
                     Tracing.TraceType traceType = traceTypeBytes == null ? Tracing.TraceType.QUERY : Tracing.TraceType.deserialize(traceTypeBytes[0]);
-                    TraceState.mutateWithTracing(ByteBuffer.wrap(sessionBytes), traceMessage, -1, traceType.getTTL());
+                    Tracing.instance.trace(ByteBuffer.wrap(sessionBytes), traceMessage, -1, traceType.getTTL());
                 }
                 else
                 {