Examples with BlockConditions org.wso2.carbon.apimgt.core.models.BlockConditions used on opensource projects

Example 31 with BlockConditions

use of org.wso2.carbon.apimgt.core.models.BlockConditions in project carbon-apimgt by wso2.

the class BlockingConditionMappingUtil method fromBlockConditionListToListDTO.

/**
 * Converts a List of Block Condition in to REST API LIST DTO Object.
 *
 * @param blockConditionList A List of Block Conditions
 * @return REST API List DTO object derived from Block Condition list
 * @throws UnsupportedThrottleLimitTypeException - If error occurs
 */
public static BlockingConditionListDTO fromBlockConditionListToListDTO(List<BlockConditions> blockConditionList) throws UnsupportedThrottleLimitTypeException {
    BlockingConditionListDTO listDTO = new BlockingConditionListDTO();
    List<BlockingConditionDTO> blockingConditionDTOList = new ArrayList<>();
    if (blockConditionList != null) {
        for (BlockConditions blockCondition : blockConditionList) {
            BlockingConditionDTO dto = fromBlockingConditionToDTO(blockCondition);
            blockingConditionDTOList.add(dto);
        }
    }
    listDTO.setCount(blockingConditionDTOList.size());
    listDTO.setList(blockingConditionDTOList);
    return listDTO;
}

Example 32 with BlockConditions

use of org.wso2.carbon.apimgt.core.models.BlockConditions in project carbon-apimgt by wso2.

the class BlockingConditionMappingUtil method fromBlockingConditionDTOToBlockCondition.

/**
 * Convert BlockingConditionDTO to BlockCondition.
 *
 * @param blockingConditionDTO blockindConditionDTO to be converted
 * @return BlockCondition Object
 * @throws UnsupportedThrottleLimitTypeException - If error occurs
 */
public static BlockConditions fromBlockingConditionDTOToBlockCondition(BlockingConditionDTO blockingConditionDTO) throws UnsupportedThrottleLimitTypeException {
    BlockConditions blockConditions = new BlockConditions();
    blockConditions.setUuid(blockingConditionDTO.getConditionId());
    blockConditions.setConditionType(blockingConditionDTO.getConditionType());
    blockConditions.setConditionValue(blockingConditionDTO.getConditionValue());
    blockConditions.setEnabled(blockingConditionDTO.getStatus());
    if (APIMgtConstants.ThrottlePolicyConstants.BLOCKING_CONDITION_IP_RANGE.equals(blockConditions.getConditionType())) {
        blockConditions.setStartingIP(blockingConditionDTO.getIpCondition().getStartingIP());
        blockConditions.setEndingIP(blockingConditionDTO.getIpCondition().getEndingIP());
    }
    return blockConditions;
}

Example 33 with BlockConditions

use of org.wso2.carbon.apimgt.core.models.BlockConditions in project carbon-apimgt by wso2.

the class BlockingConditionMappingUtil method fromBlockConditionToIpConditionDTO.

/**
 * Block condition IP range details to IPConditionDTO.
 *
 * @param blockConditions blockCondition to be converted into IPConditionDTO.
 * @return IPConditionDTO Object
 */
private static IPConditionDTO fromBlockConditionToIpConditionDTO(BlockConditions blockConditions) {
    IPConditionDTO ipConditionDTO = new IPConditionDTO();
    ipConditionDTO.setStartingIP(blockConditions.getStartingIP());
    ipConditionDTO.setEndingIP(blockConditions.getEndingIP());
    return ipConditionDTO;
}

Example 34 with BlockConditions

use of org.wso2.carbon.apimgt.core.models.BlockConditions in project carbon-apimgt by wso2.

the class BlacklistApiServiceImplTest method blacklistGetTest.

@Test
public void blacklistGetTest() throws APIManagementException, NotFoundException {
    printTestMethodName();
    BlacklistApiServiceImpl blacklistApiService = new BlacklistApiServiceImpl();
    String uuid = UUID.randomUUID().toString();
    APIMgtAdminServiceImpl adminService = Mockito.mock(APIMgtAdminServiceImpl.class);
    PowerMockito.mockStatic(RestApiUtil.class);
    PowerMockito.when(RestApiUtil.getAPIMgtAdminService()).thenReturn(adminService);
    BlockConditions conditions1 = new BlockConditions();
    conditions1.setUuid(uuid);
    BlockConditions conditions2 = new BlockConditions();
    conditions2.setUuid(UUID.randomUUID().toString());
    List<BlockConditions> list = new ArrayList<>();
    list.add(conditions1);
    list.add(conditions2);
    Mockito.doReturn(list).doThrow(new IllegalArgumentException()).when(adminService).getBlockConditions();
    Response response = blacklistApiService.blacklistGet(null, null, getRequest());
    Assert.assertEquals(response.getStatus(), 200);
}

Example 35 with BlockConditions

use of org.wso2.carbon.apimgt.core.models.BlockConditions in project carbon-apimgt by wso2.

the class BlacklistApiServiceImplTest method blacklistPostTest.

@Test
public void blacklistPostTest() throws APIManagementException, NotFoundException {
    printTestMethodName();
    BlacklistApiServiceImpl blacklistApiService = new BlacklistApiServiceImpl();
    String uuid = UUID.randomUUID().toString();
    BlockingConditionDTO dto = new BlockingConditionDTO();
    dto.setConditionId(uuid);
    dto.setStatus(true);
    dto.setConditionType(BLOCKING_CONDITIONS_IP);
    dto.setConditionValue("12.32.45.3");
    APIMgtAdminServiceImpl adminService = Mockito.mock(APIMgtAdminServiceImpl.class);
    PowerMockito.mockStatic(RestApiUtil.class);
    PowerMockito.mockStatic(BlockingConditionMappingUtil.class);
    PowerMockito.when(RestApiUtil.getAPIMgtAdminService()).thenReturn(adminService);
    BlockConditions conditions = BlockingConditionMappingUtil.fromBlockingConditionDTOToBlockCondition(dto);
    Mockito.doReturn(uuid).doThrow(new IllegalArgumentException()).when(adminService).addBlockCondition(conditions);
    Mockito.doReturn(conditions).doThrow(new IllegalArgumentException()).when(adminService).getBlockConditionByUUID(uuid);
    PowerMockito.when(BlockingConditionMappingUtil.fromBlockingConditionDTOToBlockCondition(dto)).thenReturn(conditions);
    Response response = blacklistApiService.blacklistPost(dto, getRequest());
    Assert.assertEquals(201, response.getStatus());
}