Search in sources :

Example 6 with AclDeleteResult

use of org.apache.kafka.server.authorizer.AclDeleteResult in project kafka by apache.

the class ClusterMetadataAuthorizerTest method testDeleteAclsError.

@Test
public void testDeleteAclsError() throws Exception {
    MockAclMutator mutator = new MockAclMutator();
    MockClusterMetadataAuthorizer authorizer = new MockClusterMetadataAuthorizer();
    authorizer.setAclMutator(mutator);
    CompletableFuture<List<AclDeleteResult>> response = new CompletableFuture<>();
    response.completeExceptionally(new AuthorizationException("not authorized"));
    mutator.setDeleteAclsResponse(response);
    List<? extends CompletionStage<AclDeleteResult>> results = authorizer.deleteAcls(new MockAuthorizableRequestContext.Builder().build(), TEST_FILTERS);
    assertEquals(2, results.size());
    for (int i = 0; i < 2; i++) {
        AclDeleteResult deleteResult = results.get(i).toCompletableFuture().get();
        assertEquals(0, deleteResult.aclBindingDeleteResults().size());
        assertEquals(AuthorizationException.class, deleteResult.exception().get().getClass());
    }
}
Also used : CompletableFuture(java.util.concurrent.CompletableFuture) AuthorizationException(org.apache.kafka.common.errors.AuthorizationException) List(java.util.List) AclDeleteResult(org.apache.kafka.server.authorizer.AclDeleteResult) Endpoint(org.apache.kafka.common.Endpoint) Test(org.junit.jupiter.api.Test)

Example 7 with AclDeleteResult

use of org.apache.kafka.server.authorizer.AclDeleteResult in project kafka by apache.

the class AclControlManagerTest method testCreateAclDeleteAcl.

@Test
public void testCreateAclDeleteAcl() {
    SnapshotRegistry snapshotRegistry = new SnapshotRegistry(new LogContext());
    AclControlManager manager = new AclControlManager(snapshotRegistry, Optional.empty());
    MockClusterMetadataAuthorizer authorizer = new MockClusterMetadataAuthorizer();
    authorizer.loadSnapshot(manager.idToAcl());
    List<AclBinding> toCreate = new ArrayList<>();
    for (int i = 0; i < 3; i++) {
        toCreate.add(TEST_ACLS.get(i).toBinding());
    }
    toCreate.add(new AclBinding(new ResourcePattern(TOPIC, "*", PatternType.UNKNOWN), new AccessControlEntry("User:*", "*", ALTER, ALLOW)));
    ControllerResult<List<AclCreateResult>> createResult = manager.createAcls(toCreate);
    List<AclCreateResult> expectedResults = new ArrayList<>();
    for (int i = 0; i < 3; i++) {
        expectedResults.add(AclCreateResult.SUCCESS);
    }
    expectedResults.add(new AclCreateResult(new InvalidRequestException("Invalid patternType UNKNOWN")));
    for (int i = 0; i < expectedResults.size(); i++) {
        AclCreateResult expectedResult = expectedResults.get(i);
        if (expectedResult.exception().isPresent()) {
            assertEquals(expectedResult.exception().get().getMessage(), createResult.response().get(i).exception().get().getMessage());
        } else {
            assertFalse(createResult.response().get(i).exception().isPresent());
        }
    }
    RecordTestUtils.replayAll(manager, createResult.records());
    assertTrue(manager.iterator(Long.MAX_VALUE).hasNext());
    ControllerResult<List<AclDeleteResult>> deleteResult = manager.deleteAcls(Arrays.asList(new AclBindingFilter(new ResourcePatternFilter(ResourceType.ANY, null, LITERAL), AccessControlEntryFilter.ANY), new AclBindingFilter(new ResourcePatternFilter(ResourceType.UNKNOWN, null, LITERAL), AccessControlEntryFilter.ANY)));
    assertEquals(2, deleteResult.response().size());
    Set<AclBinding> deleted = new HashSet<>();
    for (AclDeleteResult.AclBindingDeleteResult result : deleteResult.response().get(0).aclBindingDeleteResults()) {
        assertEquals(Optional.empty(), result.exception());
        deleted.add(result.aclBinding());
    }
    assertEquals(new HashSet<>(Arrays.asList(TEST_ACLS.get(0).toBinding(), TEST_ACLS.get(2).toBinding())), deleted);
    assertEquals(InvalidRequestException.class, deleteResult.response().get(1).exception().get().getClass());
    RecordTestUtils.replayAll(manager, deleteResult.records());
    Iterator<List<ApiMessageAndVersion>> iterator = manager.iterator(Long.MAX_VALUE);
    assertTrue(iterator.hasNext());
    List<ApiMessageAndVersion> list = iterator.next();
    assertEquals(1, list.size());
    assertEquals(TEST_ACLS.get(1).toBinding(), StandardAcl.fromRecord((AccessControlEntryRecord) list.get(0).message()).toBinding());
    assertFalse(iterator.hasNext());
}
Also used : ResourcePatternFilter(org.apache.kafka.common.resource.ResourcePatternFilter) ResourcePattern(org.apache.kafka.common.resource.ResourcePattern) ArrayList(java.util.ArrayList) AclDeleteResult(org.apache.kafka.server.authorizer.AclDeleteResult) AclCreateResult(org.apache.kafka.server.authorizer.AclCreateResult) ApiMessageAndVersion(org.apache.kafka.server.common.ApiMessageAndVersion) List(java.util.List) ArrayList(java.util.ArrayList) InvalidRequestException(org.apache.kafka.common.errors.InvalidRequestException) AclBinding(org.apache.kafka.common.acl.AclBinding) HashSet(java.util.HashSet) AclBindingFilter(org.apache.kafka.common.acl.AclBindingFilter) LogContext(org.apache.kafka.common.utils.LogContext) AccessControlEntry(org.apache.kafka.common.acl.AccessControlEntry) Endpoint(org.apache.kafka.common.Endpoint) SnapshotRegistry(org.apache.kafka.timeline.SnapshotRegistry) StandardAclWithIdTest(org.apache.kafka.metadata.authorizer.StandardAclWithIdTest) StandardAclTest(org.apache.kafka.metadata.authorizer.StandardAclTest) Test(org.junit.jupiter.api.Test)

Aggregations

AclDeleteResult (org.apache.kafka.server.authorizer.AclDeleteResult)7 ArrayList (java.util.ArrayList)4 List (java.util.List)4 CompletableFuture (java.util.concurrent.CompletableFuture)3 AclBinding (org.apache.kafka.common.acl.AclBinding)3 ApiMessageAndVersion (org.apache.kafka.server.common.ApiMessageAndVersion)3 Test (org.junit.jupiter.api.Test)3 Endpoint (org.apache.kafka.common.Endpoint)2 AccessControlEntry (org.apache.kafka.common.acl.AccessControlEntry)2 AclBindingFilter (org.apache.kafka.common.acl.AclBindingFilter)2 InvalidRequestException (org.apache.kafka.common.errors.InvalidRequestException)2 ResourcePattern (org.apache.kafka.common.resource.ResourcePattern)2 AclBindingDeleteResult (org.apache.kafka.server.authorizer.AclDeleteResult.AclBindingDeleteResult)2 ByteBuffer (java.nio.ByteBuffer)1 HashSet (java.util.HashSet)1 Map (java.util.Map)1 Collectors (java.util.stream.Collectors)1 Uuid (org.apache.kafka.common.Uuid)1 AclOperation (org.apache.kafka.common.acl.AclOperation)1 AclPermissionType (org.apache.kafka.common.acl.AclPermissionType)1