Some small refactorings and cleanups.
[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
32 import junit.framework.TestCase;
33
34 import org.apache.log4j.BasicConfigurator;
35 import org.apache.log4j.Level;
36 import org.apache.log4j.Logger;
37 import org.opensaml.SAMLAttribute;
38 import org.opensaml.SAMLException;
39 import org.opensaml.XML;
40 import org.w3c.dom.Element;
41
42 import edu.internet2.middleware.shibboleth.aap.provider.XMLAAP;
43 import edu.internet2.middleware.shibboleth.metadata.Metadata;
44 import edu.internet2.middleware.shibboleth.metadata.provider.XMLMetadata;
45
46 /**
47  * Validation suite for the <code>Metadata</code> interface.
48  * 
49  * @author Walter Hoehn
50  */
51
52 public class AAPTests extends TestCase {
53
54         public AAPTests(String name) {
55                 super(name);
56                 BasicConfigurator.resetConfiguration();
57                 BasicConfigurator.configure();
58                 Logger.getRootLogger().setLevel(Level.OFF);
59         }
60
61         public static void main(String[] args) {
62                 junit.textui.TestRunner.run(AAPTests.class);
63                 BasicConfigurator.configure();
64                 Logger.getRootLogger().setLevel(Level.OFF);
65         }
66
67         protected void setUp() throws Exception {
68                 super.setUp();
69         }
70
71     public void testBasic() {
72
73         try {
74             AAP aap = new XMLAAP(new File("src/conf/AAP.xml").toURL().toString());
75
76             assertFalse("anyAttribute was true",aap.anyAttribute());
77             
78             AttributeRule rule = aap.lookup("affiliation");
79             assertNotNull("Unable to find rule", rule);
80             assertTrue("Rule wasn't scoped",rule.getScoped());
81             assertFalse("Rule was case-sensitive",rule.getCaseSensitive());
82             
83             SAMLAttribute a1 = new SAMLAttribute(new FileInputStream("data/attribute1.xml"));
84             SAMLAttribute a2 = new SAMLAttribute((Element)a1.toDOM().cloneNode(true));
85             
86             rule = aap.lookup(a1.getName(),a1.getNamespace());
87             assertNotNull("Unable to find rule", rule);
88             
89             rule.apply(a1, null);
90             try {
91                 a1.checkValidity();
92                 assertTrue("Attribute should have been stripped clean",false);
93             }
94             catch (SAMLException ex) {
95             }
96             
97             Metadata metadata = new XMLMetadata(new File("src/conf/IQ-sites.xml").toURL().toString());
98             rule.apply(a2, metadata.lookup("urn:mace:inqueue:example.edu").getAttributeAuthorityDescriptor(XML.SAML11_PROTOCOL_ENUM));
99             a2.checkValidity();
100             assertTrue("Value was unexpected","member".equalsIgnoreCase((String)a2.getValues().next()));
101             
102         } catch (Exception e) {
103             fail("Failed to correctly load AAP: " + e);
104         }
105     }
106 }