From a690e7393c898b90ea4626bf92823f538034b070 Mon Sep 17 00:00:00 2001
From: Tom Petracca <tpetracca@palantir.com>
Date: Thu, 12 Dec 2024 18:07:04 -0500
Subject: [PATCH] update dtests

---
 .../org/apache/cassandra/distributed/impl/Coordinator.java      | 2 +-
 .../org/apache/cassandra/distributed/impl/Instance.java         | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/test/distributed/org/apache/cassandra/distributed/impl/Coordinator.java b/test/distributed/org/apache/cassandra/distributed/impl/Coordinator.java
index 6553fb94de..01354a3f20 100644
--- 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
index aca1cfd479..139786f140 100644
--- 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
                 {