mirror of https://github.com/jacekkow/keycloak-protocol-cas
view | history | commit | commitdiff
Jacek Kowalski
2024-06-21 d63df349ec82ec25cca4f04ff1600395434dd4c4
commit | author | age
513246 1 package org.keycloak.protocol.cas;
MP 2
bce810 3 import com.jayway.jsonpath.JsonPath;
513246 4 import org.junit.Test;
352436 5 import org.keycloak.protocol.cas.representations.CASErrorCode;
8a5518 6 import org.keycloak.protocol.cas.representations.CASServiceResponse;
513246 7 import org.keycloak.protocol.cas.utils.ServiceResponseHelper;
MP 8 import org.keycloak.protocol.cas.utils.ServiceResponseMarshaller;
bce810 9 import org.w3c.dom.Document;
MP 10 import org.w3c.dom.Node;
11 import org.xmlunit.xpath.JAXPXPathEngine;
12 import org.xmlunit.xpath.XPathEngine;
513246 13
bce810 14 import java.util.*;
513246 15
MP 16 import static org.junit.Assert.assertEquals;
74023a 17 import static org.keycloak.protocol.cas.XMLValidator.parseAndValidate;
EH 18 import static org.keycloak.protocol.cas.XMLValidator.schemaFromClassPath;
513246 19
MP 20 public class ServiceResponseTest {
bce810 21     private final XPathEngine xpath = new JAXPXPathEngine();
MP 22
23     public ServiceResponseTest() {
24         xpath.setNamespaceContext(Collections.singletonMap("cas", "http://www.yale.edu/tp/cas"));
25     }
513246 26
MP 27     @Test
28     public void testSuccessResponse() throws Exception {
29         Map<String, Object> attributes = new HashMap<>();
30         attributes.put("list", Arrays.asList("a", "b"));
31         attributes.put("int", 123);
32         attributes.put("string", "abc");
33
bce810 34         List<String> proxies = Arrays.asList("https://proxy1/pgtUrl", "https://proxy2/pgtUrl");
8a5518 35         CASServiceResponse response = ServiceResponseHelper.createSuccess("username", attributes, "PGTIOU-test",
bce810 36                 proxies);
513246 37
bce810 38         // Build and validate JSON response
MP 39
40         String json = ServiceResponseMarshaller.marshalJson(response);
41         assertEquals("username", JsonPath.read(json, "$.serviceResponse.authenticationSuccess.user"));
42         assertEquals(attributes.get("list"), JsonPath.read(json, "$.serviceResponse.authenticationSuccess.attributes.list"));
43         assertEquals(attributes.get("int"), JsonPath.read(json, "$.serviceResponse.authenticationSuccess.attributes.int"));
44         assertEquals(attributes.get("string"), JsonPath.read(json, "$.serviceResponse.authenticationSuccess.attributes.string"));
45         assertEquals("PGTIOU-test", JsonPath.read(json, "$.serviceResponse.authenticationSuccess.proxyGrantingTicket"));
46         assertEquals(proxies, JsonPath.read(json, "$.serviceResponse.authenticationSuccess.proxies"));
47
48         // Build and validate XML response
49
50         String xml = ServiceResponseMarshaller.marshalXml(response);
74023a 51         Document doc = parseAndValidate(xml, schemaFromClassPath("cas-response-schema.xsd"));
bce810 52         assertEquals("username", xpath.evaluate("/cas:serviceResponse/cas:authenticationSuccess/cas:user", doc));
MP 53         int idx = 0;
54         for (Node node : xpath.selectNodes("/cas:serviceResponse/cas:authenticationSuccess/cas:attributes/cas:list", doc)) {
5d7080 55             assertEquals(((List<?>)attributes.get("list")).get(idx), node.getTextContent());
bce810 56             idx++;
MP 57         }
5d7080 58         assertEquals(((List<?>)attributes.get("list")).size(), idx);
bce810 59         assertEquals(attributes.get("int").toString(), xpath.evaluate("/cas:serviceResponse/cas:authenticationSuccess/cas:attributes/cas:int", doc));
MP 60         assertEquals(attributes.get("string").toString(), xpath.evaluate("/cas:serviceResponse/cas:authenticationSuccess/cas:attributes/cas:string", doc));
61
62         assertEquals("PGTIOU-test", xpath.evaluate("/cas:serviceResponse/cas:authenticationSuccess/cas:proxyGrantingTicket", doc));
63         idx = 0;
64         for (Node node : xpath.selectNodes("/cas:serviceResponse/cas:authenticationSuccess/cas:proxies/cas:proxy", doc)) {
65             assertEquals(proxies.get(idx), node.getTextContent());
66             idx++;
67         }
68         assertEquals(proxies.size(), idx);
513246 69     }
MP 70
71     @Test
72     public void testErrorResponse() throws Exception {
8a5518 73         CASServiceResponse response = ServiceResponseHelper.createFailure(CASErrorCode.INVALID_REQUEST, "Error description");
513246 74
bce810 75         // Build and validate JSON response
MP 76
77         String json = ServiceResponseMarshaller.marshalJson(response);
78         assertEquals(CASErrorCode.INVALID_REQUEST.name(), JsonPath.read(json, "$.serviceResponse.authenticationFailure.code"));
79         assertEquals("Error description", JsonPath.read(json, "$.serviceResponse.authenticationFailure.description"));
80
81         // Build and validate XML response
82
83         String xml = ServiceResponseMarshaller.marshalXml(response);
74023a 84         Document doc = parseAndValidate(xml, schemaFromClassPath("cas-response-schema.xsd"));
bce810 85         assertEquals(CASErrorCode.INVALID_REQUEST.name(), xpath.evaluate("/cas:serviceResponse/cas:authenticationFailure/@code", doc));
MP 86         assertEquals("Error description", xpath.evaluate("/cas:serviceResponse/cas:authenticationFailure", doc));
513246 87     }
MP 88 }