diff options
author | Tor Egge <Tor.Egge@oath.com> | 2018-02-06 14:37:59 +0000 |
---|---|---|
committer | Tor Egge <Tor.Egge@oath.com> | 2018-02-06 15:19:06 +0000 |
commit | ddae19a0f1d041d68e795bc4c3a25a1f0126a6cc (patch) | |
tree | 747eed88e79588ed6fa2771bc8db163487cc4473 /vespaclient-java/src/test/java/com/yahoo/vespastat/BucketStatsRetrieverTest.java | |
parent | 96c103ab9e0a9e87eb04846a40ba7096898415fb (diff) |
Make vespa-stat use bucketSpace.
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, 5 insertions, 4 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 557cefd8504..a34e3a73205 100644 --- a/vespaclient-java/src/test/java/com/yahoo/vespastat/BucketStatsRetrieverTest.java +++ b/vespaclient-java/src/test/java/com/yahoo/vespastat/BucketStatsRetrieverTest.java @@ -31,6 +31,7 @@ public class BucketStatsRetrieverTest { private DocumentAccessFactory mockedFactory; private MessageBusDocumentAccess mockedDocumentAccess; private MessageBusSyncSession mockedSession; + private final String bucketSpace = "default"; @Before @@ -67,7 +68,7 @@ public class BucketStatsRetrieverTest { reply.getBuckets().add(new GetBucketListReply.BucketInfo(bucketId, bucketInfo)); when(mockedSession.syncSend(any())).thenReturn(reply); - List<GetBucketListReply.BucketInfo> bucketList = createRetriever().retrieveBucketList(bucketId); + List<GetBucketListReply.BucketInfo> bucketList = createRetriever().retrieveBucketList(bucketId, bucketSpace); verify(mockedSession, times(1)).syncSend(any()); assertEquals(1, bucketList.size()); @@ -83,7 +84,7 @@ public class BucketStatsRetrieverTest { StatBucketReply reply = new StatBucketReply(); reply.setResults(bucketInfo); when(mockedSession.syncSend(any())).thenReturn(reply); - String result = createRetriever().retrieveBucketStats(ClientParameters.SelectionType.DOCUMENT, docId, bucketId); + String result = createRetriever().retrieveBucketStats(ClientParameters.SelectionType.DOCUMENT, docId, bucketId, bucketSpace); verify(mockedSession, times(1)).syncSend(any()); assertEquals(bucketInfo, result); @@ -112,7 +113,7 @@ public class BucketStatsRetrieverTest { reply.addError(new Error(0, "errormsg")); when(mockedSession.syncSend(any())).thenReturn(reply); - createRetriever().retrieveBucketList(new BucketId(1)); + createRetriever().retrieveBucketList(new BucketId(1), bucketSpace); } @Test @@ -124,7 +125,7 @@ public class BucketStatsRetrieverTest { when(mockedSession.syncSend(any())).thenReturn(reply); BucketStatsRetriever retriever = new BucketStatsRetriever(mockedFactory, route, t -> {}); - retriever.retrieveBucketList(new BucketId(0)); + retriever.retrieveBucketList(new BucketId(0), bucketSpace); verify(mockedSession).syncSend(argThat(new ArgumentMatcher<Message>() { @Override |