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/BucketStatsPrinterTest.java | |
parent | 96c103ab9e0a9e87eb04846a40ba7096898415fb (diff) |
Make vespa-stat use bucketSpace.
Diffstat (limited to 'vespaclient-java/src/test/java/com/yahoo/vespastat/BucketStatsPrinterTest.java')
-rw-r--r-- | vespaclient-java/src/test/java/com/yahoo/vespastat/BucketStatsPrinterTest.java | 13 |
1 files changed, 7 insertions, 6 deletions
diff --git a/vespaclient-java/src/test/java/com/yahoo/vespastat/BucketStatsPrinterTest.java b/vespaclient-java/src/test/java/com/yahoo/vespastat/BucketStatsPrinterTest.java index 874a2a382ea..aaeada5be25 100644 --- a/vespaclient-java/src/test/java/com/yahoo/vespastat/BucketStatsPrinterTest.java +++ b/vespaclient-java/src/test/java/com/yahoo/vespastat/BucketStatsPrinterTest.java @@ -22,13 +22,14 @@ public class BucketStatsPrinterTest { private BucketStatsRetriever retriever; private final ByteArrayOutputStream out = new ByteArrayOutputStream(); + private final String bucketSpace = "default"; @Before public void mockBucketStatsRetriever() throws BucketStatsException { retriever = mock(BucketStatsRetriever.class); when(retriever.getBucketIdForType(any(), any())).thenReturn(new BucketId(0x42)); - when(retriever.retrieveBucketList(any())).thenReturn(Collections.emptyList()); - when(retriever.retrieveBucketStats(any(), any(), any())).thenReturn(""); + when(retriever.retrieveBucketList(any(), any())).thenReturn(Collections.emptyList()); + when(retriever.retrieveBucketStats(any(), any(), any(), any())).thenReturn(""); } @After @@ -44,7 +45,7 @@ public class BucketStatsPrinterTest { private String retreiveAndPrintBucketStats(ClientParameters.SelectionType type, String id, boolean dumpData) throws BucketStatsException { BucketStatsPrinter printer = new BucketStatsPrinter(retriever, new PrintStream(out)); - printer.retrieveAndPrintBucketStats(type, id, dumpData); + printer.retrieveAndPrintBucketStats(type, id, dumpData, bucketSpace); return getOutputString(); } @@ -68,7 +69,7 @@ public class BucketStatsPrinterTest { List<GetBucketListReply.BucketInfo> bucketList = new ArrayList<>(); String dummyInfoString = "dummyinformation"; bucketList.add(new GetBucketListReply.BucketInfo(new BucketId(0), dummyInfoString)); - when(retriever.retrieveBucketList(any())).thenReturn(bucketList); + when(retriever.retrieveBucketList(any(), any())).thenReturn(bucketList); String output = retreiveAndPrintBucketStats(ClientParameters.SelectionType.USER, "1234", false); assertTrue(output.contains(dummyInfoString)); @@ -78,8 +79,8 @@ public class BucketStatsPrinterTest { public void testShouldPrintBucketStats() throws BucketStatsException { String dummyBucketStats = "dummystats"; GetBucketListReply.BucketInfo bucketInfo = new GetBucketListReply.BucketInfo(new BucketId(0), "dummy"); - when(retriever.retrieveBucketList(any())).thenReturn(Collections.singletonList(bucketInfo)); - when(retriever.retrieveBucketStats(any(), any(), any())).thenReturn(dummyBucketStats); + when(retriever.retrieveBucketList(any(), any())).thenReturn(Collections.singletonList(bucketInfo)); + when(retriever.retrieveBucketStats(any(), any(), any(), any())).thenReturn(dummyBucketStats); String output = retreiveAndPrintBucketStats(ClientParameters.SelectionType.USER, "1234", true); assertTrue(output.contains(dummyBucketStats)); |