Examples with CommandError org.apache.pulsar.common.api.proto.PulsarApi.CommandError used on opensource projects

Example 11 with CommandError

use of org.apache.pulsar.common.api.proto.PulsarApi.CommandError in project incubator-pulsar by apache.

the class ServerCnxTest method testClusterAccess.

@Test(timeOut = 30000)
public void testClusterAccess() throws Exception {
    svcConfig.setAuthorizationEnabled(true);
    AuthorizationService authorizationService = spy(new AuthorizationService(svcConfig, configCacheService));
    Field providerField = AuthorizationService.class.getDeclaredField("provider");
    providerField.setAccessible(true);
    PulsarAuthorizationProvider authorizationProvider = spy(new PulsarAuthorizationProvider(svcConfig, configCacheService));
    providerField.set(authorizationService, authorizationProvider);
    doReturn(authorizationService).when(brokerService).getAuthorizationService();
    doReturn(true).when(brokerService).isAuthorizationEnabled();
    doReturn(false).when(authorizationProvider).isSuperUser(Mockito.anyString());
    doReturn(CompletableFuture.completedFuture(true)).when(authorizationProvider).checkPermission(any(TopicName.class), Mockito.anyString(), any(AuthAction.class));
    resetChannel();
    setChannelConnected();
    ByteBuf clientCommand = Commands.newProducer(successTopicName, 1, /* producer id */
    1, /* request id */
    "prod-name", Collections.emptyMap());
    channel.writeInbound(clientCommand);
    assertTrue(getResponse() instanceof CommandProducerSuccess);
    resetChannel();
    setChannelConnected();
    clientCommand = Commands.newProducer(topicWithNonLocalCluster, 1, /* producer id */
    1, /* request id */
    "prod-name", Collections.emptyMap());
    channel.writeInbound(clientCommand);
    assertTrue(getResponse() instanceof CommandError);
}

Example 12 with CommandError

use of org.apache.pulsar.common.api.proto.PulsarApi.CommandError in project incubator-pulsar by apache.

the class ServerCnxTest method testProducerCommandWithAuthorizationNegative.

public void testProducerCommandWithAuthorizationNegative() throws Exception {
    AuthorizationService authorizationService = mock(AuthorizationService.class);
    doReturn(CompletableFuture.completedFuture(false)).when(authorizationService).canProduceAsync(Mockito.any(), Mockito.any(), Mockito.any());
    doReturn(authorizationService).when(brokerService).getAuthorizationService();
    doReturn(true).when(brokerService).isAuthenticationEnabled();
    doReturn(true).when(brokerService).isAuthorizationEnabled();
    doReturn("prod1").when(brokerService).generateUniqueProducerName();
    resetChannel();
    setChannelConnected();
    ByteBuf clientCommand = Commands.newProducer(successTopicName, 1, /* producer id */
    1, /* request id */
    null, Collections.emptyMap());
    channel.writeInbound(clientCommand);
    assertTrue(getResponse() instanceof CommandError);
    channel.finish();
}

Example 13 with CommandError

use of org.apache.pulsar.common.api.proto.PulsarApi.CommandError in project incubator-pulsar by apache.

the class ServerCnxTest method testInvalidTopicOnProducer.

@Test(timeOut = 30000)
public void testInvalidTopicOnProducer() throws Exception {
    resetChannel();
    setChannelConnected();
    String invalidTopicName = "xx/ass/aa/aaa";
    resetChannel();
    setChannelConnected();
    ByteBuf clientCommand = Commands.newProducer(invalidTopicName, 1, /* producer id */
    1, /* request id */
    "prod-name", Collections.emptyMap());
    channel.writeInbound(clientCommand);
    Object obj = getResponse();
    assertEquals(obj.getClass(), CommandError.class);
    CommandError res = (CommandError) obj;
    assertEquals(res.getError(), ServerError.InvalidTopicName);
    channel.finish();
}

Example 14 with CommandError

use of org.apache.pulsar.common.api.proto.PulsarApi.CommandError in project incubator-pulsar by apache.

the class ServerCnxTest method testDuplicateConcurrentProducerCommand.

@Test(timeOut = 5000)
public void testDuplicateConcurrentProducerCommand() throws Exception {
    resetChannel();
    setChannelConnected();
    CompletableFuture<Topic> delayFuture = new CompletableFuture<>();
    doReturn(delayFuture).when(brokerService).getTopic(any(String.class));
    // Create producer first time
    ByteBuf clientCommand = Commands.newProducer(successTopicName, 1, /* producer id */
    1, /* request id */
    "prod-name", Collections.emptyMap());
    channel.writeInbound(clientCommand);
    // Create producer second time
    clientCommand = Commands.newProducer(successTopicName, 1, /* producer id */
    1, /* request id */
    "prod-name", Collections.emptyMap());
    channel.writeInbound(clientCommand);
    Object response = getResponse();
    assertTrue(response instanceof CommandError);
    CommandError error = (CommandError) response;
    assertEquals(error.getError(), ServerError.ServiceNotReady);
}

Example 15 with CommandError

use of org.apache.pulsar.common.api.proto.PulsarApi.CommandError in project incubator-pulsar by apache.

the class ServerCnxTest method testDuplicateConcurrentSubscribeCommand.

@Test(timeOut = 5000)
public void testDuplicateConcurrentSubscribeCommand() throws Exception {
    resetChannel();
    setChannelConnected();
    CompletableFuture<Topic> delayFuture = new CompletableFuture<>();
    doReturn(delayFuture).when(brokerService).getTopic(any(String.class));
    // Create subscriber first time
    ByteBuf clientCommand = // 
    Commands.newSubscribe(// 
    successTopicName, successSubName, 1, /* consumer id */
    1, /* request id */
    SubType.Exclusive, 0, "test");
    channel.writeInbound(clientCommand);
    // Create producer second time
    clientCommand = // 
    Commands.newSubscribe(// 
    successTopicName, successSubName, 1, /* consumer id */
    1, /* request id */
    SubType.Exclusive, 0, "test");
    channel.writeInbound(clientCommand);
    Object response = getResponse();
    assertTrue(response instanceof CommandError);
    CommandError error = (CommandError) response;
    assertEquals(error.getError(), ServerError.ServiceNotReady);
}