The beginnings of ARP serialization. Tests for the same.
[java-idp.git] / src / edu / internet2 / middleware / shibboleth / aa / arp / ArpTests.java
1 /* 
2  * The Shibboleth License, Version 1. 
3  * Copyright (c) 2002 
4  * University Corporation for Advanced Internet Development, Inc. 
5  * All rights reserved
6  * 
7  * 
8  * Redistribution and use in source and binary forms, with or without 
9  * modification, are permitted provided that the following conditions are met:
10  * 
11  * Redistributions of source code must retain the above copyright notice, this 
12  * list of conditions and the following disclaimer.
13  * 
14  * Redistributions in binary form must reproduce the above copyright notice, 
15  * this list of conditions and the following disclaimer in the documentation 
16  * and/or other materials provided with the distribution, if any, must include 
17  * the following acknowledgment: "This product includes software developed by 
18  * the University Corporation for Advanced Internet Development 
19  * <http://www.ucaid.edu>Internet2 Project. Alternately, this acknowledegement 
20  * may appear in the software itself, if and wherever such third-party 
21  * acknowledgments normally appear.
22  * 
23  * Neither the name of Shibboleth nor the names of its contributors, nor 
24  * Internet2, nor the University Corporation for Advanced Internet Development, 
25  * Inc., nor UCAID may be used to endorse or promote products derived from this 
26  * software without specific prior written permission. For written permission, 
27  * please contact shibboleth@shibboleth.org
28  * 
29  * Products derived from this software may not be called Shibboleth, Internet2, 
30  * UCAID, or the University Corporation for Advanced Internet Development, nor 
31  * may Shibboleth appear in their name, without prior written permission of the 
32  * University Corporation for Advanced Internet Development.
33  * 
34  * 
35  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" 
36  * AND WITH ALL FAULTS. ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 
37  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY, FITNESS FOR A 
38  * PARTICULAR PURPOSE, AND NON-INFRINGEMENT ARE DISCLAIMED AND THE ENTIRE RISK 
39  * OF SATISFACTORY QUALITY, PERFORMANCE, ACCURACY, AND EFFORT IS WITH LICENSEE. 
40  * IN NO EVENT SHALL THE COPYRIGHT OWNER, CONTRIBUTORS OR THE UNIVERSITY 
41  * CORPORATION FOR ADVANCED INTERNET DEVELOPMENT, INC. BE LIABLE FOR ANY DIRECT, 
42  * INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES 
43  * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; 
44  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND 
45  * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 
46  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS 
47  * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
48  */
49
50 package edu.internet2.middleware.shibboleth.aa.arp;
51
52 import java.io.FileInputStream;
53 import java.io.InputStream;
54 import java.util.Properties;
55
56 import junit.framework.TestCase;
57
58 import org.apache.log4j.BasicConfigurator;
59 import org.apache.xerces.parsers.DOMParser;
60 import org.xml.sax.InputSource;
61
62 /**
63  * Validation suite for <code>Arp</code> processing.
64  * 
65  * @author Walter Hoehn (wassa@columbia.edu)
66  */
67
68 public class ArpTests extends TestCase {
69
70         public ArpTests(String name) {
71                 super(name);
72                 BasicConfigurator.configure();
73         }
74
75         public static void main(String[] args) {
76                 junit.textui.TestRunner.run(ArpTests.class);
77                 BasicConfigurator.configure();
78         }
79
80         public void testArpMarshalling() {
81
82                 //Test ARP description
83                 try {
84                         InputStream inStream = new FileInputStream("test/arp1.xml");
85                         DOMParser parser = new DOMParser();
86                         parser.parse(new InputSource(inStream));
87                         Arp arp1 = new Arp();
88                         arp1.marshall(parser.getDocument().getDocumentElement());
89                         assertEquals(
90                                 "ARP Description not marshalled properly",
91                                 arp1.getDescription(),
92                                 "Simplest possible ARP.");
93
94                         //Test Rule description
95                         assertEquals(
96                                 "ARP Rule Description not marshalled properly",
97                                 arp1.getAllRules()[0].getDescription(),
98                                 "Example Rule Description.");
99                 } catch (Exception e) {
100                         fail("Failed to marshall ARP.");
101                 }
102
103                 //Test case where ARP description does not exist
104                 try {
105                         InputStream inStream = new FileInputStream("test/arp2.xml");
106                         DOMParser parser = new DOMParser();
107                         parser.parse(new InputSource(inStream));
108                         Arp arp2 = new Arp();
109                         arp2.marshall(parser.getDocument().getDocumentElement());
110                         assertNull("ARP Description not marshalled properly", arp2.getDescription());
111
112                         //Test case where ARP Rule description does not exist   
113                         assertNull(
114                                 "ARP Rule Description not marshalled properly",
115                                 arp2.getAllRules()[0].getDescription());
116                 } catch (Exception e) {
117                         fail("Failed to marshall ARP.");
118                 }
119
120         }
121
122         public void testRepositories() {
123
124                 /*
125                  * Test the Factory
126                  */
127
128                 //Make sure we fail if no Repository is specified
129                 Properties props = new Properties();
130                 try {
131                         ArpRepositoryFactory.getInstance(props);
132                 } catch (ArpRepositoryException e) {
133                         //This is supposed to fail
134                 }
135
136                 // Make sure we can create an Arp Repository
137                 props.setProperty(
138                         "edu.internet2.middleware.shibboleth.aa.arp.ArpRepository.implementation",
139                         "edu.internet2.middleware.shibboleth.aa.arp.provider.MemoryArpRepository");
140                 ArpRepository repository = null;
141                 try {
142                         repository = ArpRepositoryFactory.getInstance(props);
143                 } catch (ArpRepositoryException e) {
144                         fail("Failed to create memory-based Arp Repository" + e);
145                 }
146                 assertNotNull("Failed to create memory-based Arp Repository: Factory returned null.", repository);
147
148                 /*
149                  * Exercise the Memory Arp Repository
150                  */
151
152                 //Set/retrieve/remove a Site ARP
153                 Arp siteArp1 = new Arp();
154                 siteArp1.setDescription("Test Site Arp 1.");
155                 try {
156                         repository.update(siteArp1);
157                         assertEquals(
158                                 "Memory Repository does not store and retrieve Site ARPs properly.",
159                                 siteArp1,
160                                 repository.getSitePolicy());
161                         repository.remove(repository.getSitePolicy());
162                         assertNull("Memorty Repository does not properly delete Site ARPs.", repository.getSitePolicy());
163                 } catch (ArpRepositoryException e) {
164                         fail("Error adding Site ARP to Memory Repository.");
165                 }
166
167                 //Set/retrieve/delete some user ARPs
168                 Arp userArp1 = new Arp();
169                 userArp1.setDescription("Broken User Arp 1.");
170                 try {
171                         repository.update(userArp1);
172                         assertTrue(
173                                 "Memory Repository does not store and retrieve User ARPs properly.",
174                                 (!userArp1.equals(repository.getUserPolicy(userArp1.getPrincipal()))));
175                 } catch (ArpRepositoryException e) {
176                         fail("Error adding User ARP to Memory Repository.");
177                 }
178
179                 Arp userArp2 = new Arp(new AAPrincipal("TestPrincipal"));
180                 userArp2.setDescription("Test User Arp 2.");
181                 try {
182                         repository.update(userArp2);
183                         assertEquals(
184                                 "Memory Repository does not store and retrieve User ARPs properly.",
185                                 userArp2,
186                                 repository.getUserPolicy(userArp2.getPrincipal()));
187                         repository.remove(repository.getUserPolicy(userArp2.getPrincipal()));
188                         assertNull(
189                                 "Memorty Repository does not properly delete User ARPs.",
190                                 repository.getUserPolicy(userArp2.getPrincipal()));
191                 } catch (ArpRepositoryException e) {
192                         fail("Error adding User ARP to Memory Repository.");
193                 }
194
195         }
196
197 }