diff options
Diffstat (limited to 'vespaclient-java/src/test/java/com/yahoo/vespastat/BucketStatsRetrieverTest.java')
-rw-r--r-- | vespaclient-java/src/test/java/com/yahoo/vespastat/BucketStatsRetrieverTest.java | 9 |
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() { |