Set route on send session instead of individual messages

Session route is "default" unless otherwise specified, which overrides
per-message routes. This means that the route parameter has not worked
as expected.

1 files changed, 3 insertions, 6 deletions

diff --git a/vespaclient-java/src/test/java/com/yahoo/vespastat/BucketStatsRetrieverTest.java b/vespaclient-java/src/test/java/com/yahoo/vespastat/BucketStatsRetrieverTest.java
index a34e3a73205..475547546d3 100644
--- a/vespaclient-java/src/test/java/com/yahoo/vespastat/BucketStatsRetrieverTest.java
+++ b/vespaclient-java/src/test/java/com/yahoo/vespastat/BucketStatsRetrieverTest.java
@@ -20,6 +20,7 @@ import java.util.List;
 import static org.junit.Assert.assertEquals;
 import static org.mockito.Matchers.any;
 import static org.mockito.Matchers.argThat;
+import static org.mockito.Matchers.eq;
 import static org.mockito.Mockito.mock;
 import static org.mockito.Mockito.times;
 import static org.mockito.Mockito.verify;
@@ -127,12 +128,8 @@ public class BucketStatsRetrieverTest {
         BucketStatsRetriever retriever = new BucketStatsRetriever(mockedFactory, route, t -> {});
         retriever.retrieveBucketList(new BucketId(0), bucketSpace);
 
-        verify(mockedSession).syncSend(argThat(new ArgumentMatcher<Message>() {
-            @Override
-            public boolean matches(Object o) {
-                return ((Message) o).getRoute().equals(Route.parse(route));
-            }
-        }));
+        // Route is set at session-level, not per message sent.
+        verify(mockedSession).setRoute(eq(route));
     }
 
     private BucketStatsRetriever createRetriever() {