ea1998ac64561bb365641226ea2df028e96eb829
[java-idp.git] / tests / edu / internet2 / middleware / shibboleth / aap / AAPTests.java
1 /*
2  * The Shibboleth License, Version 1. Copyright (c) 2002 University Corporation for Advanced Internet Development, Inc.
3  * All rights reserved Redistribution and use in source and binary forms, with or without modification, are permitted
4  * provided that the following conditions are met: Redistributions of source code must retain the above copyright
5  * notice, this list of conditions and the following disclaimer. Redistributions in binary form must reproduce the
6  * above copyright notice, this list of conditions and the following disclaimer in the documentation and/or other
7  * materials provided with the distribution, if any, must include the following acknowledgment: "This product includes
8  * software developed by the University Corporation for Advanced Internet Development <http://www.ucaid.edu> Internet2
9  * Project. Alternately, this acknowledegement may appear in the software itself, if and wherever such third-party
10  * acknowledgments normally appear. Neither the name of Shibboleth nor the names of its contributors, nor Internet2,
11  * nor the University Corporation for Advanced Internet Development, Inc., nor UCAID may be used to endorse or promote
12  * products derived from this software without specific prior written permission. For written permission, please
13  * contact shibboleth@shibboleth.org Products derived from this software may not be called Shibboleth, Internet2,
14  * UCAID, or the University Corporation for Advanced Internet Development, nor may Shibboleth appear in their name,
15  * without prior written permission of the University Corporation for Advanced Internet Development. THIS SOFTWARE IS
16  * PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND WITH ALL FAULTS. ANY EXPRESS OR IMPLIED WARRANTIES,
17  * INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE, AND
18  * NON-INFRINGEMENT ARE DISCLAIMED AND THE ENTIRE RISK OF SATISFACTORY QUALITY, PERFORMANCE, ACCURACY, AND EFFORT IS
19  * WITH LICENSEE. IN NO EVENT SHALL THE COPYRIGHT OWNER, CONTRIBUTORS OR THE UNIVERSITY CORPORATION FOR ADVANCED
20  * INTERNET DEVELOPMENT, INC. BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
21  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS;
22  * OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR
23  * TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
24  * POSSIBILITY OF SUCH DAMAGE.
25  */
26
27 package edu.internet2.middleware.shibboleth.aap;
28
29 import java.io.File;
30 import java.io.FileInputStream;
31 import java.util.Arrays;
32 import java.util.Iterator;
33
34 import junit.framework.TestCase;
35
36 import org.apache.log4j.BasicConfigurator;
37 import org.apache.log4j.Level;
38 import org.apache.log4j.Logger;
39
40 import org.opensaml.SAMLAttribute;
41 import org.opensaml.SAMLException;
42 import org.opensaml.XML;
43 import org.w3c.dom.Element;
44
45 import edu.internet2.middleware.shibboleth.aap.provider.XMLAAP;
46 import edu.internet2.middleware.shibboleth.metadata.Metadata;
47 import edu.internet2.middleware.shibboleth.metadata.provider.XMLMetadata;
48
49 /**
50  * Validation suite for the <code>Metadata</code> interface.
51  * 
52  * @author Walter Hoehn
53  */
54
55 public class AAPTests extends TestCase {
56
57         public AAPTests(String name) {
58                 super(name);
59                 BasicConfigurator.resetConfiguration();
60                 BasicConfigurator.configure();
61                 Logger.getRootLogger().setLevel(Level.OFF);
62         }
63
64         public static void main(String[] args) {
65                 junit.textui.TestRunner.run(AAPTests.class);
66                 BasicConfigurator.configure();
67                 Logger.getRootLogger().setLevel(Level.OFF);
68         }
69
70         protected void setUp() throws Exception {
71                 super.setUp();
72         }
73
74     public void testBasic() {
75
76         try {
77             AAP aap = new XMLAAP(new File("src/conf/AAP.xml").toURL().toString());
78
79             assertFalse("anyAttribute was true",aap.anyAttribute());
80             
81             AttributeRule rule = aap.lookup("affiliation");
82             assertNotNull("Unable to find rule", rule);
83             assertTrue("Rule wasn't scoped",rule.getScoped());
84             assertFalse("Rule was case-sensitive",rule.getCaseSensitive());
85             
86             SAMLAttribute a1 = new SAMLAttribute(new FileInputStream("data/attribute1.xml"));
87             SAMLAttribute a2 = new SAMLAttribute((Element)a1.toDOM().cloneNode(true));
88             
89             rule = aap.lookup(a1.getName(),a1.getNamespace());
90             assertNotNull("Unable to find rule", rule);
91             
92             rule.apply(a1, null);
93             try {
94                 a1.checkValidity();
95                 assertTrue("Attribute should have been stripped clean",false);
96             }
97             catch (SAMLException ex) {
98             }
99             
100             Metadata metadata = new XMLMetadata(new File("src/conf/IQ-sites.xml").toURL().toString());
101             rule.apply(a2, metadata.lookup("urn:mace:inqueue:example.edu").getAttributeAuthorityDescriptor(XML.SAML11_PROTOCOL_ENUM));
102             a2.checkValidity();
103             assertTrue("Value was unexpected","member".equalsIgnoreCase((String)a2.getValues().next()));
104             
105         } catch (Exception e) {
106             fail("Failed to correctly load AAP: " + e);
107         }
108     }
109 }