1 files changed, 17 insertions, 2 deletions
diff --git a/pw_transfer/java/main/dev/pigweed/pw_transfer/TransferClient.java b/pw_transfer/java/main/dev/pigweed/pw_transfer/TransferClient.java
index 248e50ea9..13779005a 100644
--- a/pw_transfer/java/main/dev/pigweed/pw_transfer/TransferClient.java
+++ b/pw_transfer/java/main/dev/pigweed/pw_transfer/TransferClient.java
@@ -35,7 +35,7 @@ public class TransferClient {
   private final TransferEventHandler transferEventHandler;
   private final Thread transferEventHandlerThread;
 
-  private ProtocolVersion desiredProtocolVersion = ProtocolVersion.latest();
+  private ProtocolVersion desiredProtocolVersion = ProtocolVersion.LEGACY;
 
   /**
    * Creates a new transfer client for sending and receiving data with pw_transfer.
@@ -153,7 +153,22 @@ public class TransferClient {
     transferEventHandlerThread.join();
   }
 
-  void waitUntilEventsAreProcessedForTest() {
+  // Functions for test use only.
+  // TODO: b/279808806 - These could be annotated with test-only visibility.
+
+  final void waitUntilEventsAreProcessedForTest() {
     transferEventHandler.waitUntilEventsAreProcessedForTest();
   }
+
+  final int getNextSessionIdForTest() {
+    return transferEventHandler.getNextSessionIdForTest();
+  }
+
+  final WriteTransfer getWriteTransferForTest(ListenableFuture<?> transferFuture) {
+    return (WriteTransfer) transferFuture;
+  }
+
+  final ReadTransfer getReadTransferForTest(ListenableFuture<?> transferFuture) {
+    return (ReadTransfer) transferFuture;
+  }
 }