use of com.yahoo.documentapi.messagebus.protocol.GetBucketListReply in project vespa by vespa-engine.
the class BucketStatsRetriever method retrieveBucketList.
public List<GetBucketListReply.BucketInfo> retrieveBucketList(BucketId bucketId, String bucketSpace) throws BucketStatsException {
GetBucketListMessage msg = new GetBucketListMessage(bucketId, bucketSpace);
GetBucketListReply bucketListReply = sendMessage(msg, GetBucketListReply.class);
return bucketListReply.getBuckets();
}
use of com.yahoo.documentapi.messagebus.protocol.GetBucketListReply in project vespa by vespa-engine.
the class BucketStatsRetrieverTest method testRetrieveBucketList.
@Test
public void testRetrieveBucketList() throws BucketStatsException {
String bucketInfo = "I like turtles!";
BucketId bucketId = bucketIdFactory.getBucketId(new DocumentId("id:ns:type::another"));
GetBucketListReply reply = new GetBucketListReply();
reply.getBuckets().add(new GetBucketListReply.BucketInfo(bucketId, bucketInfo));
when(mockedSession.syncSend(any())).thenReturn(reply);
List<GetBucketListReply.BucketInfo> bucketList = createRetriever().retrieveBucketList(bucketId, bucketSpace);
verify(mockedSession, times(1)).syncSend(any());
assertEquals(1, bucketList.size());
assertEquals(bucketInfo, bucketList.get(0).getBucketInformation());
}
use of com.yahoo.documentapi.messagebus.protocol.GetBucketListReply in project vespa by vespa-engine.
the class BucketStatsRetrieverTest method testRoute.
@Test
public void testRoute() throws BucketStatsException {
String route = "default";
BucketId bucketId = bucketIdFactory.getBucketId(new DocumentId("id:ns:type::another"));
GetBucketListReply reply = new GetBucketListReply();
reply.getBuckets().add(new GetBucketListReply.BucketInfo(bucketId, "I like turtles!"));
when(mockedSession.syncSend(any())).thenReturn(reply);
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));
}
}));
}
use of com.yahoo.documentapi.messagebus.protocol.GetBucketListReply in project vespa by vespa-engine.
the class BucketStatsRetrieverTest method testShouldFailOnReplyError.
@Test(expected = BucketStatsException.class)
public void testShouldFailOnReplyError() throws BucketStatsException {
GetBucketListReply reply = new GetBucketListReply();
reply.addError(new Error(0, "errormsg"));
when(mockedSession.syncSend(any())).thenReturn(reply);
createRetriever().retrieveBucketList(new BucketId(1), bucketSpace);
}
Aggregations