The beginnings of a test suite. Slightly more flesh on the ARP interface. Memory...
[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.util.Properties;
53
54 import org.apache.log4j.BasicConfigurator;
55
56 import junit.framework.TestCase;
57
58 /**
59  * Validation suite for <code>Arp</code> processing.
60  * 
61  * @author Walter Hoehn (wassa@columbia.edu)
62  */
63
64 public class ArpTests extends TestCase {
65
66         public ArpTests(String name) {
67                 super(name);
68                 BasicConfigurator.configure();
69         }
70
71         public static void main(String[] args) {
72                 junit.textui.TestRunner.run(ArpTests.class);
73                 BasicConfigurator.configure();
74         }
75
76         public void testRepositories() {
77
78                 /*
79                  * Test the Factory
80                  */
81
82                 //Make sure we fail if no Repository is specified
83                 Properties props = new Properties();
84                 try {
85                         ArpRepositoryFactory.getInstance(props);
86                 } catch (ArpRepositoryException e) {
87                         //This is supposed to fail
88                 }
89
90                 // Make sure we can create an Arp Repository
91                 props.setProperty(
92                         "edu.internet2.middleware.shibboleth.aa.arp.ArpRepository.implementation",
93                         "edu.internet2.middleware.shibboleth.aa.arp.provider.MemoryArpRepository");
94                 ArpRepository repository = null;
95                 try {
96                         repository = ArpRepositoryFactory.getInstance(props);
97                 } catch (ArpRepositoryException e) {
98                         fail("Failed to create memory-based Arp Repository" + e);
99                 }
100                 assertNotNull("Failed to create memory-based Arp Repository: Factory returned null.", repository);
101
102                 /*
103                  * Exercise the Memory Arp Repository
104                  */
105
106                 //Set/retrieve/remove a Site ARP
107                 Arp siteArp1 = new Arp();
108                 siteArp1.setDescription("Test Site Arp 1.");
109                 try {
110                         repository.update(siteArp1);
111                         assertEquals(
112                                 "Memory Repository does not store and retrieve Site ARPs properly.",
113                                 siteArp1,
114                                 repository.getSitePolicy());
115                         repository.remove(repository.getSitePolicy());
116                         assertNull("Memorty Repository does not properly delete Site ARPs.", repository.getSitePolicy());
117                 } catch (ArpRepositoryException e) {
118                         fail("Error adding Site ARP to Memory Repository.");
119                 }
120
121                 //Set/retrieve/delete some user ARPs
122                 Arp userArp1 = new Arp();
123                 userArp1.setDescription("Broken User Arp 1.");
124                 try {
125                         repository.update(userArp1);
126                         assertTrue(
127                                 "Memory Repository does not store and retrieve User ARPs properly.",
128                                 (!userArp1.equals(repository.getUserPolicy(userArp1.getPrincipal()))));
129                 } catch (ArpRepositoryException e) {
130                         fail("Error adding User ARP to Memory Repository.");
131                 }
132
133                 Arp userArp2 = new Arp(new AAPrincipal("TestPrincipal"));
134                 userArp2.setDescription("Test User Arp 2.");
135                 try {
136                         repository.update(userArp2);
137                         assertEquals(
138                                 "Memory Repository does not store and retrieve User ARPs properly.",
139                                 userArp2,
140                                 repository.getUserPolicy(userArp2.getPrincipal()));
141                         repository.remove(repository.getUserPolicy(userArp2.getPrincipal()));
142                         assertNull(
143                                 "Memorty Repository does not properly delete User ARPs.",
144                                 repository.getUserPolicy(userArp2.getPrincipal()));
145                 } catch (ArpRepositoryException e) {
146                         fail("Error adding User ARP to Memory Repository.");
147                 }
148
149         }
150
151 }