Search in sources :

Example 46 with PortEntity

use of org.apache.nifi.web.api.entity.PortEntity in project nifi by apache.

the class ITOutputPortAccessControl method testReadWriteUserPutOutputPort.

/**
 * Ensures the READ_WRITE user can put an output port.
 *
 * @throws Exception ex
 */
@Test
public void testReadWriteUserPutOutputPort() throws Exception {
    final PortEntity entity = getRandomOutputPort(helper.getReadWriteUser());
    assertTrue(entity.getPermissions().getCanRead());
    assertTrue(entity.getPermissions().getCanWrite());
    assertNotNull(entity.getComponent());
    final String updatedName = "Updated Name" + count++;
    // attempt to update the name
    final long version = entity.getRevision().getVersion();
    entity.getRevision().setClientId(AccessControlHelper.READ_WRITE_CLIENT_ID);
    entity.getComponent().setName(updatedName);
    // perform the request
    final Response response = updateOutputPort(helper.getReadWriteUser(), entity);
    // ensure successful response
    assertEquals(200, response.getStatus());
    // get the response
    final PortEntity responseEntity = response.readEntity(PortEntity.class);
    // verify
    assertEquals(READ_WRITE_CLIENT_ID, responseEntity.getRevision().getClientId());
    assertEquals(version + 1, responseEntity.getRevision().getVersion().longValue());
    assertEquals(updatedName, responseEntity.getComponent().getName());
}
Also used : Response(javax.ws.rs.core.Response) PortEntity(org.apache.nifi.web.api.entity.PortEntity) Test(org.junit.Test)

Example 47 with PortEntity

use of org.apache.nifi.web.api.entity.PortEntity in project nifi by apache.

the class ITOutputPortAccessControl method testReadUserPutOutputPort.

/**
 * Ensures the READ user cannot put an output port.
 *
 * @throws Exception ex
 */
@Test
public void testReadUserPutOutputPort() throws Exception {
    final PortEntity entity = getRandomOutputPort(helper.getReadUser());
    assertTrue(entity.getPermissions().getCanRead());
    assertFalse(entity.getPermissions().getCanWrite());
    assertNotNull(entity.getComponent());
    // attempt update the name
    entity.getRevision().setClientId(READ_CLIENT_ID);
    entity.getComponent().setName("Updated Name" + count++);
    // perform the request
    final Response response = updateOutputPort(helper.getReadUser(), entity);
    // ensure forbidden response
    assertEquals(403, response.getStatus());
}
Also used : Response(javax.ws.rs.core.Response) PortEntity(org.apache.nifi.web.api.entity.PortEntity) Test(org.junit.Test)

Example 48 with PortEntity

use of org.apache.nifi.web.api.entity.PortEntity in project nifi by apache.

the class ITOutputPortAccessControl method createOutputPort.

private PortEntity createOutputPort(final String name) throws Exception {
    String url = helper.getBaseUrl() + "/process-groups/root/output-ports";
    final String updatedName = name + count++;
    // create the output port
    PortDTO outputPort = new PortDTO();
    outputPort.setName(updatedName);
    // create the revision
    final RevisionDTO revision = new RevisionDTO();
    revision.setClientId(READ_WRITE_CLIENT_ID);
    revision.setVersion(0L);
    // create the entity body
    PortEntity entity = new PortEntity();
    entity.setRevision(revision);
    entity.setComponent(outputPort);
    // perform the request
    Response response = helper.getReadWriteUser().testPost(url, entity);
    // ensure the request is successful
    assertEquals(201, response.getStatus());
    // get the entity body
    entity = response.readEntity(PortEntity.class);
    // verify creation
    outputPort = entity.getComponent();
    assertEquals(updatedName, outputPort.getName());
    // get the output port
    return entity;
}
Also used : Response(javax.ws.rs.core.Response) PortDTO(org.apache.nifi.web.api.dto.PortDTO) RevisionDTO(org.apache.nifi.web.api.dto.RevisionDTO) PortEntity(org.apache.nifi.web.api.entity.PortEntity)

Example 49 with PortEntity

use of org.apache.nifi.web.api.entity.PortEntity in project nifi by apache.

the class ITOutputPortAccessControl method testReadWriteUserGetOutputPort.

/**
 * Ensures the READ WRITE user can get an output port.
 *
 * @throws Exception ex
 */
@Test
public void testReadWriteUserGetOutputPort() throws Exception {
    final PortEntity entity = getRandomOutputPort(helper.getReadWriteUser());
    assertTrue(entity.getPermissions().getCanRead());
    assertTrue(entity.getPermissions().getCanWrite());
    assertNotNull(entity.getComponent());
}
Also used : PortEntity(org.apache.nifi.web.api.entity.PortEntity) Test(org.junit.Test)

Example 50 with PortEntity

use of org.apache.nifi.web.api.entity.PortEntity in project nifi by apache.

the class ITOutputPortAccessControl method testReadUserGetOutputPort.

/**
 * Ensures the READ user can get an output port.
 *
 * @throws Exception ex
 */
@Test
public void testReadUserGetOutputPort() throws Exception {
    final PortEntity entity = getRandomOutputPort(helper.getReadUser());
    assertTrue(entity.getPermissions().getCanRead());
    assertFalse(entity.getPermissions().getCanWrite());
    assertNotNull(entity.getComponent());
}
Also used : PortEntity(org.apache.nifi.web.api.entity.PortEntity) Test(org.junit.Test)

Aggregations

PortEntity (org.apache.nifi.web.api.entity.PortEntity)52 Response (javax.ws.rs.core.Response)23 Test (org.junit.Test)18 RevisionDTO (org.apache.nifi.web.api.dto.RevisionDTO)17 Authorizable (org.apache.nifi.authorization.resource.Authorizable)16 PortDTO (org.apache.nifi.web.api.dto.PortDTO)16 HashMap (java.util.HashMap)13 ApiOperation (io.swagger.annotations.ApiOperation)10 ApiResponses (io.swagger.annotations.ApiResponses)10 Map (java.util.Map)10 Consumes (javax.ws.rs.Consumes)10 Path (javax.ws.rs.Path)10 Produces (javax.ws.rs.Produces)10 NodeIdentifier (org.apache.nifi.cluster.protocol.NodeIdentifier)10 FlowDTO (org.apache.nifi.web.api.dto.flow.FlowDTO)9 ProcessGroupFlowEntity (org.apache.nifi.web.api.entity.ProcessGroupFlowEntity)9 ConnectionEntity (org.apache.nifi.web.api.entity.ConnectionEntity)8 FunnelEntity (org.apache.nifi.web.api.entity.FunnelEntity)8 LabelEntity (org.apache.nifi.web.api.entity.LabelEntity)8 ProcessGroupEntity (org.apache.nifi.web.api.entity.ProcessGroupEntity)8