use of org.wso2.carbon.apimgt.rest.api.admin.dto.BlockingConditionDTO 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());
}
use of org.wso2.carbon.apimgt.rest.api.admin.dto.BlockingConditionDTO in project carbon-apimgt by wso2.
the class BlacklistApiServiceImpl method blacklistConditionIdPut.
/**
* Update blacklist/block condition statues
*
* @param conditionId uuid of the blacklist condition
* @param body body of the blacklist status to be updated
* @param ifMatch If-Match header value
* @param ifUnmodifiedSince If-Unmodified-Since header value
* @param request msf4j request object
* @return Response Object
* @throws NotFoundException if an error occurred when particular resource does not exits in the system.
*/
@Override
public Response blacklistConditionIdPut(String conditionId, BlockingConditionDTO body, String ifMatch, String ifUnmodifiedSince, Request request) throws NotFoundException {
if (log.isDebugEnabled()) {
log.debug("Received BlockCondition GET request with id: " + conditionId);
}
try {
APIMgtAdminService apiMgtAdminService = RestApiUtil.getAPIMgtAdminService();
// This will give BlockConditionNotFoundException if there's no block condition exists with UUID
Boolean status = apiMgtAdminService.updateBlockConditionStateByUUID(conditionId, body.getStatus());
BlockingConditionDTO dto = null;
if (status) {
BlockConditions blockCondition = apiMgtAdminService.getBlockConditionByUUID(conditionId);
dto = BlockingConditionMappingUtil.fromBlockingConditionToDTO(blockCondition);
}
return Response.ok().entity(dto).build();
} catch (APIManagementException e) {
String errorMessage = "Error occurred while getting blacklist condition with UUID " + conditionId;
ErrorDTO errorDTO = RestApiUtil.getErrorDTO(e.getErrorHandler());
log.error(errorMessage, e);
return Response.status(e.getErrorHandler().getHttpStatusCode()).entity(errorDTO).build();
}
}
use of org.wso2.carbon.apimgt.rest.api.admin.dto.BlockingConditionDTO in project carbon-apimgt by wso2.
the class BlacklistApiServiceImpl method blacklistConditionIdGet.
/**
* Get blacklist condition by ID.
*
* @param conditionId ID of the blacklist condition to be retrieved
* @param ifNoneMatch IF-None-Match header value
* @param ifModifiedSince If-Modified-Since header value
* @param request msf4j request object
* @return Response object
* @throws NotFoundException if an error occurred when particular resource does not exits in the system.
*/
@Override
public Response blacklistConditionIdGet(String conditionId, String ifNoneMatch, String ifModifiedSince, Request request) throws NotFoundException {
if (log.isDebugEnabled()) {
log.debug("Received BlockCondition GET request with id: " + conditionId);
}
try {
APIMgtAdminService apiMgtAdminService = RestApiUtil.getAPIMgtAdminService();
// This will give BlockConditionNotFoundException if there's no block condition exists with UUID
BlockConditions blockCondition = apiMgtAdminService.getBlockConditionByUUID(conditionId);
BlockingConditionDTO dto = BlockingConditionMappingUtil.fromBlockingConditionToDTO(blockCondition);
return Response.ok().entity(dto).build();
} catch (APIManagementException e) {
String errorMessage = "Error occurred while getting blacklist condition with UUID " + conditionId;
ErrorDTO errorDTO = RestApiUtil.getErrorDTO(e.getErrorHandler());
log.error(errorMessage, e);
return Response.status(e.getErrorHandler().getHttpStatusCode()).entity(errorDTO).build();
}
}
use of org.wso2.carbon.apimgt.rest.api.admin.dto.BlockingConditionDTO in project carbon-apimgt by wso2.
the class BlacklistApiServiceImpl method blacklistPost.
/**
* Add a blacklist condition.
*
* @param body DTO object including the blacklist condition data
* @param request msf4j request object
* @return Response object
* @throws NotFoundException if an error occurred when particular resource does not exits in the system.
*/
@Override
public Response blacklistPost(BlockingConditionDTO body, Request request) throws NotFoundException {
if (log.isDebugEnabled()) {
log.debug("Received BlockCondition POST request with body: " + body);
}
try {
APIMgtAdminService apiMgtAdminService = RestApiUtil.getAPIMgtAdminService();
// Add the block condition. It will throw BlockConditionAlreadyExistsException if the condition already
// exists in the system
BlockConditions blockConditions = BlockingConditionMappingUtil.fromBlockingConditionDTOToBlockCondition(body);
String uuid = apiMgtAdminService.addBlockCondition(blockConditions);
// retrieve the new blocking condition and send back as the response
BlockConditions newBlockingCondition = apiMgtAdminService.getBlockConditionByUUID(uuid);
BlockingConditionDTO dto = BlockingConditionMappingUtil.fromBlockingConditionToDTO(newBlockingCondition);
return Response.status(Response.Status.CREATED).entity(dto).build();
} catch (APIManagementException e) {
String errorMessage = "Error occurred while adding blocking condition with UUID " + body.getConditionId();
ErrorDTO errorDTO = RestApiUtil.getErrorDTO(e.getErrorHandler());
log.error(errorMessage, e);
return Response.status(e.getErrorHandler().getHttpStatusCode()).entity(errorDTO).build();
}
}
use of org.wso2.carbon.apimgt.rest.api.admin.dto.BlockingConditionDTO in project carbon-apimgt by wso2.
the class MappingUtilTestCase method fromBlockingConditionToDTOIPRangeTest.
@Test
public void fromBlockingConditionToDTOIPRangeTest() {
BlockConditions blockConditionIPRANGE = SampleTestObjectCreator.createUniqueBlockConditions(APIMgtConstants.ThrottlePolicyConstants.BLOCKING_CONDITION_IP_RANGE);
BlockingConditionDTO blockingConditionDTo = MappingUtil.fromBlockingConditionToDTO(blockConditionIPRANGE);
Assert.assertEquals(blockConditionIPRANGE.getUuid(), blockingConditionDTo.getUuid());
Assert.assertEquals(blockConditionIPRANGE.getConditionType(), blockingConditionDTo.getConditionType());
Assert.assertEquals(blockConditionIPRANGE.getConditionValue(), blockingConditionDTo.getConditionValue());
Assert.assertEquals(Boolean.valueOf(blockConditionIPRANGE.isEnabled()), Boolean.valueOf(blockingConditionDTo.getEnabled()));
Assert.assertEquals(Long.valueOf(APIUtils.ipToLong(blockConditionIPRANGE.getEndingIP())), Long.valueOf(blockingConditionDTo.getEndingIP()));
Assert.assertEquals(Long.valueOf(APIUtils.ipToLong(blockConditionIPRANGE.getStartingIP())), Long.valueOf(blockingConditionDTo.getStartingIP()));
// Test for null handling
blockConditionIPRANGE.setUuid(null);
Assert.assertNull(MappingUtil.fromBlockingConditionToDTO(blockConditionIPRANGE));
}
Aggregations