use of org.apache.kafka.common.message.FindCoordinatorResponseData in project kafka by apache.
the class CoordinatorStrategyTest method testSuccessfulOldCoordinatorLookup.
@Test
public void testSuccessfulOldCoordinatorLookup() {
CoordinatorKey group = CoordinatorKey.byGroupId("foo");
FindCoordinatorResponseData responseData = new FindCoordinatorResponseData().setErrorCode(Errors.NONE.code()).setHost("localhost").setPort(9092).setNodeId(1);
AdminApiLookupStrategy.LookupResult<CoordinatorKey> result = runOldLookup(group, responseData);
assertEquals(singletonMap(group, 1), result.mappedKeys);
assertEquals(emptyMap(), result.failedKeys);
}
use of org.apache.kafka.common.message.FindCoordinatorResponseData in project kafka by apache.
the class CoordinatorStrategyTest method testHandleOldResponseRequiresOneKey.
@Test
public void testHandleOldResponseRequiresOneKey() {
FindCoordinatorResponseData responseData = new FindCoordinatorResponseData().setErrorCode(Errors.NONE.code());
FindCoordinatorResponse response = new FindCoordinatorResponse(responseData);
CoordinatorStrategy strategy = new CoordinatorStrategy(CoordinatorType.GROUP, new LogContext());
strategy.disableBatch();
assertThrows(IllegalArgumentException.class, () -> strategy.handleResponse(Collections.emptySet(), response));
CoordinatorKey group1 = CoordinatorKey.byGroupId("foo");
CoordinatorKey group2 = CoordinatorKey.byGroupId("bar");
assertThrows(IllegalArgumentException.class, () -> strategy.handleResponse(mkSet(group1, group2), response));
}
use of org.apache.kafka.common.message.FindCoordinatorResponseData in project kafka by apache.
the class CoordinatorStrategyTest method assertFatalLookup.
public Throwable assertFatalLookup(CoordinatorKey key, Errors error) {
FindCoordinatorResponseData responseData = new FindCoordinatorResponseData().setCoordinators(Collections.singletonList(new FindCoordinatorResponseData.Coordinator().setKey(key.idValue).setErrorCode(error.code())));
AdminApiLookupStrategy.LookupResult<CoordinatorKey> result = runLookup(singleton(key), responseData);
assertEquals(emptyMap(), result.mappedKeys);
assertEquals(singleton(key), result.failedKeys.keySet());
Throwable throwable = result.failedKeys.get(key);
assertTrue(error.exception().getClass().isInstance(throwable));
return throwable;
}
use of org.apache.kafka.common.message.FindCoordinatorResponseData in project kafka by apache.
the class CoordinatorStrategyTest method testSuccessfulCoordinatorLookup.
@Test
public void testSuccessfulCoordinatorLookup() {
CoordinatorKey group1 = CoordinatorKey.byGroupId("foo");
CoordinatorKey group2 = CoordinatorKey.byGroupId("bar");
FindCoordinatorResponseData responseData = new FindCoordinatorResponseData().setCoordinators(Arrays.asList(new FindCoordinatorResponseData.Coordinator().setKey("foo").setErrorCode(Errors.NONE.code()).setHost("localhost").setPort(9092).setNodeId(1), new FindCoordinatorResponseData.Coordinator().setKey("bar").setErrorCode(Errors.NONE.code()).setHost("localhost").setPort(9092).setNodeId(2)));
AdminApiLookupStrategy.LookupResult<CoordinatorKey> result = runLookup(new HashSet<>(Arrays.asList(group1, group2)), responseData);
Map<CoordinatorKey, Integer> expectedResult = new HashMap<>();
expectedResult.put(group1, 1);
expectedResult.put(group2, 2);
assertEquals(expectedResult, result.mappedKeys);
assertEquals(emptyMap(), result.failedKeys);
}
use of org.apache.kafka.common.message.FindCoordinatorResponseData in project kafka by apache.
the class CoordinatorStrategyTest method testRetriableOldCoordinatorLookup.
private void testRetriableOldCoordinatorLookup(Errors error) {
CoordinatorKey group = CoordinatorKey.byGroupId("foo");
FindCoordinatorResponseData responseData = new FindCoordinatorResponseData().setErrorCode(error.code());
AdminApiLookupStrategy.LookupResult<CoordinatorKey> result = runOldLookup(group, responseData);
assertEquals(emptyMap(), result.failedKeys);
assertEquals(emptyMap(), result.mappedKeys);
}
Aggregations