Project

General

Profile

1 6324 leinfelder
/**
2
 *  '$RCSfile$'
3
 *  Copyright: 2010 Regents of the University of California and the
4
 *              National Center for Ecological Analysis and Synthesis
5
 *  Purpose: To test the Access Controls in metacat by JUnit
6
 *
7
 *   '$Author$'
8
 *     '$Date$'
9
 * '$Revision$'
10
 *
11
 * This program is free software; you can redistribute it and/or modify
12
 * it under the terms of the GNU General Public License as published by
13
 * the Free Software Foundation; either version 2 of the License, or
14
 * (at your option) any later version.
15
 *
16
 * This program is distributed in the hope that it will be useful,
17
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
18
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
19
 * GNU General Public License for more details.
20
 *
21
 * You should have received a copy of the GNU General Public License
22
 * along with this program; if not, write to the Free Software
23
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
24
 */
25
26
package edu.ucsb.nceas.metacat.dataone;
27
28 9104 tao
import java.io.IOException;
29 6324 leinfelder
import java.io.InputStream;
30
import java.io.InputStreamReader;
31
import java.io.Reader;
32 6385 cjones
import java.math.BigInteger;
33 6324 leinfelder
import java.util.Date;
34 7163 leinfelder
import java.util.List;
35 6324 leinfelder
36 9180 tao
import junit.framework.Test;
37
import junit.framework.TestSuite;
38
39 7622 leinfelder
import org.apache.wicket.protocol.http.mock.MockHttpServletRequest;
40 8834 leinfelder
import org.dataone.client.D1Node;
41
import org.dataone.client.NodeLocator;
42
import org.dataone.client.exception.ClientSideException;
43 8810 leinfelder
import org.dataone.client.v2.CNode;
44
import org.dataone.client.v2.itk.D1Client;
45
import org.dataone.client.v2.formats.ObjectFormatCache;
46 9187 tao
import org.dataone.configuration.Settings;
47 6366 leinfelder
import org.dataone.service.types.v1.AccessPolicy;
48
import org.dataone.service.types.v1.AccessRule;
49
import org.dataone.service.types.v1.Checksum;
50
import org.dataone.service.types.v1.Identifier;
51 8810 leinfelder
import org.dataone.service.types.v2.Node;
52 8834 leinfelder
import org.dataone.service.types.v2.ObjectFormatList;
53 6366 leinfelder
import org.dataone.service.types.v1.NodeReference;
54 7163 leinfelder
import org.dataone.service.types.v1.NodeType;
55 6366 leinfelder
import org.dataone.service.types.v1.Permission;
56
import org.dataone.service.types.v1.Session;
57
import org.dataone.service.types.v1.Subject;
58 8810 leinfelder
import org.dataone.service.types.v2.SystemMetadata;
59 6755 leinfelder
import org.dataone.service.types.v1.util.ChecksumUtil;
60 8810 leinfelder
import org.dataone.service.types.v2.util.ObjectFormatServiceImpl;
61 6755 leinfelder
import org.dataone.service.util.Constants;
62 8834 leinfelder
import org.dataone.service.util.TypeMarshaller;
63 6324 leinfelder
64
import edu.ucsb.nceas.MCTestCase;
65
import edu.ucsb.nceas.metacat.client.Metacat;
66
import edu.ucsb.nceas.metacat.client.MetacatFactory;
67 9998 tao
import edu.ucsb.nceas.metacat.dataone.D1NodeService;
68 6324 leinfelder
69
/**
70
 * A JUnit superclass for testing the dataone Node implementations
71
 */
72
public class D1NodeServiceTest extends MCTestCase {
73
74 6542 leinfelder
    protected MockHttpServletRequest request;
75
76
	/**
77 6324 leinfelder
    * constructor for the test
78
    */
79
    public D1NodeServiceTest(String name) {
80
        super(name);
81 6542 leinfelder
        // set up the fake request (for logging)
82
        request = new MockHttpServletRequest(null, null, null);
83 6324 leinfelder
    }
84 9180 tao
85
    public static Test suite()
86
    {
87
        TestSuite suite = new TestSuite();
88
        suite.addTest(new D1NodeServiceTest("initialize"));
89 10004 tao
        suite.addTest(new D1NodeServiceTest("testExpandRighsHolder"));
90 9180 tao
        return suite;
91
    }
92
93 6324 leinfelder
    /**
94
	 * Establish a testing framework by initializing appropriate objects
95
	 */
96 8834 leinfelder
    public void setUp() throws Exception {
97
    	super.setUp();
98
		NodeLocator nodeLocator = new NodeLocator() {
99
			@Override
100
			public D1Node getCNode() throws ClientSideException {
101 9104 tao
			    D1Node node = null;
102
			    try {
103
			        node = new MockCNode();
104
			    } catch (IOException e) {
105
			        throw new ClientSideException(e.getMessage());
106
			    }
107
				return node;
108 8834 leinfelder
			}
109
		};
110
		D1Client.setNodeLocator(nodeLocator );
111
112
    }
113 6324 leinfelder
114
	/**
115
	 * Release any objects after tests are complete
116
	 */
117 8834 leinfelder
	public void tearDown() {
118
		// set back to force it to use defaults
119
		D1Client.setNodeLocator(null);
120
	}
121 6324 leinfelder
122 10004 tao
	public void testExpandRighsHolder() throws Exception {
123
	      // set back to force it to use defaults
124
	       D1Client.setNodeLocator(null);
125
	       Subject rightsHolder = new Subject();
126
	       rightsHolder.setValue("CN=arctic-data-admins,DC=dataone,DC=org");
127
	       Subject user = new Subject();
128
129
	       user.setValue("CN=Christopher Jones A2108,O=Google,C=US,DC=cilogon,DC=org");
130
	       assertTrue(D1NodeService.expandRightsHolder(rightsHolder, user));
131
132
	       user.setValue("uid=foo");
133
	       assertTrue(!D1NodeService.expandRightsHolder(rightsHolder, user));
134
135 10110 tao
	       user.setValue("http://orcid.org/0000-0002-8121-2341");
136 10004 tao
	       assertTrue(D1NodeService.expandRightsHolder(rightsHolder, user));
137
138
	       rightsHolder.setValue("CN=foo,,DC=dataone,DC=org");
139
	       assertTrue(!D1NodeService.expandRightsHolder(rightsHolder, user));
140
141
	       user.setValue("uid=foo");
142
	       assertTrue(!D1NodeService.expandRightsHolder(rightsHolder, user));
143
144
	       rightsHolder.setValue(null);
145
	       assertTrue(!D1NodeService.expandRightsHolder(rightsHolder, user));
146
147
	       rightsHolder.setValue("CN=foo,,DC=dataone,DC=org");
148
	       user.setValue(null);
149
	       assertTrue(!D1NodeService.expandRightsHolder(rightsHolder, user));
150
151
	       rightsHolder.setValue(null);
152
	       assertTrue(!D1NodeService.expandRightsHolder(rightsHolder, user));
153
154
	       rightsHolder.setValue("");
155
	       user.setValue("");
156
	       assertTrue(!D1NodeService.expandRightsHolder(rightsHolder, user));
157
	       NodeLocator nodeLocator = new NodeLocator() {
158
	           @Override
159
	           public D1Node getCNode() throws ClientSideException {
160
	               D1Node node = null;
161
	               try {
162
	                   node = new MockCNode();
163
	               } catch (IOException e) {
164
	                   throw new ClientSideException(e.getMessage());
165
	               }
166
	               return node;
167
	           }
168
	       };
169
	       D1Client.setNodeLocator(nodeLocator );
170
	   }
171 9998 tao
172 6324 leinfelder
	/**
173
	 * constructs a "fake" session with a test subject
174
	 * @return
175
	 */
176 6697 leinfelder
	public Session getTestSession() throws Exception {
177 6324 leinfelder
		Session session = new Session();
178
        Subject subject = new Subject();
179
        subject.setValue("cn=test,dc=dataone,dc=org");
180
        session.setSubject(subject);
181
        return session;
182
	}
183 7163 leinfelder
184
	/**
185
	 * constructs a "fake" session with the MN subject
186
	 * @return
187
	 */
188
	public Session getMNSession() throws Exception {
189
		Session session = new Session();
190
        Subject subject = MNodeService.getInstance(request).getCapabilities().getSubject(0);
191
        session.setSubject(subject);
192
        return session;
193
	}
194 6755 leinfelder
195 7163 leinfelder
	public Session getCNSession() throws Exception {
196
		Session session = new Session();
197
		Subject subject = null;
198
		CNode cn = D1Client.getCN();
199
		List<Node> nodes = cn.listNodes().getNodeList();
200
201
		// find the first CN in the node list
202
		for (Node node : nodes) {
203 9273 tao
			if (node.getType().equals(NodeType.CN)) {
204 7163 leinfelder
				subject = node.getSubject(0);
205
				break;
206
			}
207
		}
208
		session.setSubject(subject);
209
		return session;
210
211
	}
212
213 9191 tao
	public Session getAnotherSession() throws Exception {
214
	    Session session = new Session();
215
        Subject subject = new Subject();
216
        subject.setValue("cn=test2,dc=dataone,dc=org");
217
        session.setSubject(subject);
218
        return session;
219
220
	}
221
222 6324 leinfelder
	/**
223
	 * Run an initial test that always passes to check that the test harness is
224
	 * working.
225
	 */
226
	public void initialize()
227
	{
228
	    printTestHeader("initialize");
229
		assertTrue(1 == 1);
230
	}
231
232
	/**
233
	 * create system metadata with a specified id
234
	 */
235 6431 leinfelder
	public SystemMetadata createSystemMetadata(Identifier id, Subject owner, InputStream object)
236 6324 leinfelder
	  throws Exception
237
	{
238
	    SystemMetadata sm = new SystemMetadata();
239 6561 leinfelder
	    sm.setSerialVersion(BigInteger.valueOf(1));
240 6324 leinfelder
        // set the id
241
        sm.setIdentifier(id);
242 6561 leinfelder
        sm.setFormatId(ObjectFormatCache.getInstance().getFormat("application/octet-stream").getFormatId());
243 6324 leinfelder
        // create the checksum
244 6755 leinfelder
        Checksum checksum = new Checksum();
245
        String ca = "MD5";
246
        checksum.setValue("test");
247
        checksum.setAlgorithm(ca);
248
        // actually generate one
249 6328 leinfelder
        if (object != null) {
250 6755 leinfelder
            checksum = ChecksumUtil.checksum(object, ca);
251 6328 leinfelder
        }
252 6324 leinfelder
        sm.setChecksum(checksum);
253
        // set the size
254 6385 cjones
        sm.setSize(new BigInteger("0"));
255 6324 leinfelder
        sm.setSubmitter(owner);
256
        sm.setRightsHolder(owner);
257
        sm.setDateUploaded(new Date());
258
        sm.setDateSysMetadataModified(new Date());
259 9187 tao
        String currentNodeId = Settings.getConfiguration().getString("dataone.nodeId");
260
        if(currentNodeId == null || currentNodeId.trim().equals("")) {
261
            throw new Exception("there should be value in the dataone.nodeId in the metacat.properties file.");
262
        }
263 6324 leinfelder
        NodeReference nr = new NodeReference();
264 9187 tao
        nr.setValue(currentNodeId);
265 6324 leinfelder
        sm.setOriginMemberNode(nr);
266
        sm.setAuthoritativeMemberNode(nr);
267
		// set the access to public read
268
        AccessPolicy accessPolicy = new AccessPolicy();
269
        AccessRule allow = new AccessRule();
270
        allow.addPermission(Permission.READ);
271
        Subject publicSubject = new Subject();
272 6532 leinfelder
        publicSubject.setValue(Constants.SUBJECT_PUBLIC);
273 6324 leinfelder
		allow.addSubject(publicSubject);
274
		accessPolicy.addAllow(allow);
275
        sm.setAccessPolicy(accessPolicy);
276
277
        return sm;
278
	}
279
280
	/**
281
	 * For fresh Metacat installations without the Object Format List
282
	 * we insert the default version from d1_common.jar
283
	 */
284
	protected void setUpFormats() {
285
		try {
286
			Metacat m = MetacatFactory.createMetacatConnection(metacatUrl);
287
			m.login(username, password);
288
			// check if it exists already
289
			InputStream is = null;
290
			try {
291
				is = m.read(ObjectFormatService.OBJECT_FORMAT_DOCID);
292
			} catch (Exception e) {
293
				// probably missing the doc
294
			}
295 8834 leinfelder
296
			if (is != null) {
297
				// check for v2 OFL
298
				try {
299
					ObjectFormatList ofl = TypeMarshaller.unmarshalTypeFromStream(ObjectFormatList.class, is);
300
				} catch (ClassCastException cce) {
301
					// need to update it
302
					InputStream formats = ObjectFormatServiceImpl.getInstance().getObjectFormatFile();
303
					Reader xmlDocument = new InputStreamReader(formats);
304
					int rev = m.getNewestDocRevision(ObjectFormatService.OBJECT_FORMAT_DOCID);
305
					rev++;
306
					m.update(ObjectFormatService.OBJECT_FORMAT_DOCID + "." + rev, xmlDocument, null);
307
				}
308
309
			}
310
			else {
311 6755 leinfelder
				// get the default from d1_common
312
				InputStream formats = ObjectFormatServiceImpl.getInstance().getObjectFormatFile();
313
				Reader xmlDocument = new InputStreamReader(formats);
314 6324 leinfelder
				m.insert(ObjectFormatService.OBJECT_FORMAT_DOCID + ".1", xmlDocument, null);
315
			}
316
			m.logout();
317
		} catch (Exception e) {
318
			// any number of things could go wrong
319
			e.printStackTrace();
320
		}
321
	}
322
323
	/**
324
	 * print a header to start each test
325
	 */
326
	protected void printTestHeader(String testName)
327
	{
328
	    System.out.println();
329
	    System.out.println("*************** " + testName + " ***************");
330
	}
331
332
}