Java tutorial
/* * @(#)TestUtil.java * * Copyright 2004 Sun Microsystems, Inc. All Rights Reserved. * * Redistribution and use in source and binary forms, with or without * modification, are permitted provided that the following conditions are met: * * 1. Redistribution of source code must retain the above copyright notice, * this list of conditions and the following disclaimer. * * 2. Redistribution in binary form must reproduce the above copyright * notice, this list of conditions and the following disclaimer in the * documentation and/or other materials provided with the distribution. * * Neither the name of Sun Microsystems, Inc. or the names of contributors may * be used to endorse or promote products derived from this software without * specific prior written permission. * * This software is provided "AS IS," without a warranty of any kind. ALL * EXPRESS OR IMPLIED CONDITIONS, REPRESENTATIONS AND WARRANTIES, INCLUDING * ANY IMPLIED WARRANTY OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE * OR NON-INFRINGEMENT, ARE HEREBY EXCLUDED. SUN MICROSYSTEMS, INC. ("SUN") * AND ITS LICENSORS SHALL NOT BE LIABLE FOR ANY DAMAGES SUFFERED BY LICENSEE * AS A RESULT OF USING, MODIFYING OR DISTRIBUTING THIS SOFTWARE OR ITS * DERIVATIVES. IN NO EVENT WILL SUN OR ITS LICENSORS BE LIABLE FOR ANY LOST * REVENUE, PROFIT OR DATA, OR FOR DIRECT, INDIRECT, SPECIAL, CONSEQUENTIAL, * INCIDENTAL OR PUNITIVE DAMAGES, HOWEVER CAUSED AND REGARDLESS OF THE THEORY * OF LIABILITY, ARISING OUT OF THE USE OF OR INABILITY TO USE THIS SOFTWARE, * EVEN IF SUN HAS BEEN ADVISED OF THE POSSIBILITY OF SUCH DAMAGES. * * You acknowledge that this software is not designed or intended for use in * the design, construction, operation or maintenance of any nuclear facility. */ package gpms.utils; import java.io.File; import java.io.FileInputStream; import java.io.IOException; import java.io.StringWriter; import java.util.ArrayList; import java.util.List; import java.util.Set; import javax.xml.parsers.DocumentBuilder; import javax.xml.parsers.DocumentBuilderFactory; import javax.xml.transform.Transformer; import javax.xml.transform.TransformerFactory; import javax.xml.transform.dom.DOMSource; import javax.xml.transform.stream.StreamResult; import org.apache.commons.logging.Log; import org.apache.commons.logging.LogFactory; import org.w3c.dom.Document; import org.wso2.balana.ObligationResult; import org.wso2.balana.PDP; import org.wso2.balana.ParsingException; import org.wso2.balana.ctx.AbstractRequestCtx; import org.wso2.balana.ctx.AbstractResult; import org.wso2.balana.ctx.RequestCtxFactory; import org.wso2.balana.ctx.ResponseCtx; import org.wso2.balana.ctx.Status; import org.wso2.balana.ctx.xacml3.Result; import org.wso2.balana.xacml3.Advice; import org.wso2.balana.xacml3.Attributes; /** * Simple utility class * * @author Seth Proctor */ public class PolicyTestUtil { private static Log log = LogFactory.getLog(PolicyTestUtil.class); /** * Checks matching of result that got from PDP and expected response from a * file. * * @param resultResponse * result that got from PDP * @param expectedResponse * expected response from a file * @return True/False */ public static boolean isMatching(ResponseCtx resultResponse, ResponseCtx expectedResponse) { Set<AbstractResult> results = resultResponse.getResults(); Set<AbstractResult> expectedResults = expectedResponse.getResults(); boolean finalResult = false; for (AbstractResult result : results) { boolean match = false; int decision = result.getDecision(); String status = result.getStatus().encode(); List<String> advices = new ArrayList<String>(); if (result.getAdvices() != null) { for (Advice advice : result.getAdvices()) { advices.add(advice.encode()); } } List<String> obligations = new ArrayList<String>(); if (result.getObligations() != null) { for (ObligationResult obligationResult : result.getObligations()) { obligations.add(obligationResult.encode()); } } List<String> attributesList = new ArrayList<String>(); if (result instanceof Result) { Result xacml3Result = (Result) result; if (xacml3Result.getAttributes() != null) { for (Attributes attributesElement : xacml3Result.getAttributes()) { attributesList.add(attributesElement.encode()); } } } for (AbstractResult expectedResult : expectedResults) { int decisionExpected = expectedResult.getDecision(); if (decision == 4 || decision == 5 || decision == 6) { decision = 2; } if (decision != decisionExpected) { continue; } String statusExpected = expectedResult.getStatus().encode(); if (!processResult(statusExpected).equals(processResult(status))) { continue; } List<String> advicesExpected = new ArrayList<String>(); if (expectedResult.getAdvices() != null) { for (Advice advice : expectedResult.getAdvices()) { advicesExpected.add(advice.encode()); } } if (advices.size() != advicesExpected.size()) { continue; } if (advices.size() > 0) { boolean adviceContains = false; for (String advice : advices) { if (!advicesExpected.contains(advice)) { adviceContains = false; break; } else { adviceContains = true; } } if (!adviceContains) { continue; } } List<String> obligationsExpected = new ArrayList<String>(); if (expectedResult.getObligations() != null) { for (ObligationResult obligationResult : expectedResult.getObligations()) { obligationsExpected.add(obligationResult.encode()); } } if (obligations.size() != obligationsExpected.size()) { continue; } if (obligations.size() > 0) { boolean obligationContains = false; for (String obligation : obligations) { if (!obligationsExpected.contains(obligation)) { obligationContains = false; break; } else { obligationContains = true; } } if (!obligationContains) { continue; } } // if only XACML 3.0. result if (expectedResult instanceof Result) { Result xacml3Result = (Result) expectedResult; List<String> attributesExpected = new ArrayList<String>(); if (xacml3Result.getAttributes() != null) { for (Attributes attributes : xacml3Result.getAttributes()) { attributesExpected.add(attributes.encode()); } } if (attributesList.size() != attributesExpected.size()) { continue; } if (attributesList.size() > 0) { boolean attributeContains = false; for (String attribute : attributesList) { if (!attributesExpected.contains(attribute)) { attributeContains = false; break; } else { attributeContains = true; } } if (!attributeContains) { continue; } } } match = true; break; } if (match) { finalResult = true; } else { finalResult = false; break; } } if (finalResult) { log.info("Test is Passed........!!! " + "Result received from the PDP is matched with expected result"); } else { log.info("Test is Failed........!!! " + "Result received from the PDP is NOT match with expected result"); } return finalResult; } /** * Evaluates XACML request * * @param pdp * PDP instance * @param request * XACML request * @return XACML response as ResponseCtx */ public static ResponseCtx evaluate(PDP pdp, String request) { AbstractRequestCtx requestCtx; ResponseCtx responseCtx; try { requestCtx = RequestCtxFactory.getFactory().getRequestCtx(request.replaceAll(">\\s+<", "><")); responseCtx = pdp.evaluate(requestCtx); } catch (ParsingException e) { String error = "Invalid request : " + e.getMessage(); // there was something wrong with the request, so we return // Indeterminate with a status of syntax error...though this // may change if a more appropriate status type exists ArrayList<String> code = new ArrayList<String>(); code.add(Status.STATUS_SYNTAX_ERROR); Status status = new Status(code, error); // As invalid request, by default XACML 3.0 response is created. responseCtx = new ResponseCtx(new Result(AbstractResult.DECISION_INDETERMINATE, status)); } return responseCtx; } // From PDP class // public ResponseCtx evaluate(AbstractRequestCtx request) { // // EvaluationCtx evalContext = null; // try { // evalContext = EvaluationCtxFactory.getFactory().getEvaluationCtx( // request, pdpConfig); // return evaluate(evalContext); // } catch (ParsingException e) { // logger.error("Invalid request : " + e.getMessage()); // // there was something wrong with the request, so we return // // Indeterminate with a status of syntax error...though this // // may change if a more appropriate status type exists // ArrayList<String> code = new ArrayList<String>(); // code.add(Status.STATUS_SYNTAX_ERROR); // Status status = new Status(code, e.getMessage()); // return new ResponseCtx(ResultFactory.getFactory().getResult( // AbstractResult.DECISION_INDETERMINATE, status, // request.getXacmlVersion())); // // } // } /** * This creates the XACML request from a file * * @param rootDirectory * root directory of the request files * @param versionDirectory * version directory of the request files * @param requestId * request file name * @return String or null if any error */ public static String createRequest(String rootDirectory, String versionDirectory, String requestId) { File file = new File("."); StringWriter writer = null; try { String filePath = file.getCanonicalPath() + File.separator + TestConstants.RESOURCE_PATH + File.separator + rootDirectory + File.separator + versionDirectory + File.separator + TestConstants.REQUEST_DIRECTORY + File.separator + requestId; DocumentBuilderFactory factory = DocumentBuilderFactory.newInstance(); factory.setIgnoringComments(true); factory.setNamespaceAware(true); DocumentBuilder db = factory.newDocumentBuilder(); Document doc = db.parse(new FileInputStream(filePath)); DOMSource domSource = new DOMSource(doc); writer = new StringWriter(); StreamResult result = new StreamResult(writer); TransformerFactory tf = TransformerFactory.newInstance(); Transformer transformer = tf.newTransformer(); transformer.transform(domSource, result); return writer.toString(); } catch (Exception e) { log.error("Error while reading expected response from file ", e); // ignore any exception and return null } finally { if (writer != null) { try { writer.close(); } catch (IOException e) { log.error("Error closing stream ", e); // ignore any exception and return null } } } return null; } /** * This creates the expected XACML response from a file * * @param rootDirectory * root directory of the response files * @param versionDirectory * version directory of the response files * @param responseId * response file name * @return ResponseCtx or null if any error */ public static ResponseCtx createResponse(String rootDirectory, String versionDirectory, String responseId) { File file = new File("."); try { String filePath = file.getCanonicalPath() + File.separator + TestConstants.RESOURCE_PATH + File.separator + rootDirectory + File.separator + versionDirectory + File.separator + TestConstants.RESPONSE_DIRECTORY + File.separator + responseId; DocumentBuilderFactory factory = DocumentBuilderFactory.newInstance(); factory.setIgnoringComments(true); factory.setNamespaceAware(true); factory.setValidating(false); DocumentBuilder db = factory.newDocumentBuilder(); Document doc = db.parse(new FileInputStream(filePath)); return ResponseCtx.getInstance(doc.getDocumentElement()); } catch (Exception e) { log.error("Error while reading expected response from file ", e); // ignore any exception and return null } return null; } /** * This would remove the StatusMessage from the response. Because * StatusMessage depends on the how you have defined it with the PDP, * Therefore we can not compare it with conformance tests. * * @param response * XACML response String * @return XACML response String with out StatusMessage */ private static String processResult(String response) { if (response.contains("StatusMessage")) { response = response.substring(0, response.indexOf("<StatusMessage>")) + response.substring(response.indexOf("</Status>")); } return response; } }