org.collectionspace.chain.csp.persistence.services.relation.TestRelations.java Source code

Introduction

Here is the source code for org.collectionspace.chain.csp.persistence.services.relation.TestRelations.java

Source

/* Copyright 2010 University of Cambridge
 * Licensed under the Educational Community License (ECL), Version 2.0. You may not use this file except in 
 * compliance with this License.
 *
 * You may obtain a copy of the ECL 2.0 License at https://source.collectionspace.org/collection-space/LICENSE.txt
 */
package org.collectionspace.chain.csp.persistence.services.relation;

import static org.junit.Assert.*;

import java.io.IOException;
import java.io.InputStream;
import java.util.HashMap;
import java.util.Map;

import org.apache.commons.io.IOUtils;
import org.collectionspace.chain.csp.persistence.services.ServicesBaseClass;
import org.collectionspace.chain.csp.persistence.services.connection.ConnectionException;
import org.collectionspace.chain.csp.persistence.services.connection.RequestMethod;
import org.collectionspace.chain.csp.persistence.services.connection.ReturnedMultipartDocument;
import org.collectionspace.chain.csp.persistence.services.connection.ReturnedURL;
import org.collectionspace.chain.util.json.JSONUtils;
import org.collectionspace.csp.api.persistence.ExistException;
import org.collectionspace.csp.api.persistence.Storage;
import org.collectionspace.csp.api.persistence.UnderlyingStorageException;
import org.collectionspace.csp.api.persistence.UnimplementedException;
import org.dom4j.Document;
import org.dom4j.DocumentException;
import org.json.JSONException;
import org.json.JSONObject;
import org.junit.Before;
import org.junit.Test;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

public class TestRelations extends ServicesBaseClass {
    private static final Logger log = LoggerFactory.getLogger(TestRelations.class);

    // XXX refactor
    private String getResourceString(String name) throws IOException {
        InputStream in = getResource(name);
        return IOUtils.toString(in);
    }

    @Before
    public void checkServicesRunning() throws ConnectionException {
        setup();
    }

    private Map<String, Document> getMultipartCommon(String part, String file) throws DocumentException {
        return makeMultipartCommon(part, getDocument(file));
    }

    private Map<String, Document> makeMultipartCommon(String part, Document doc) throws DocumentException {
        Map<String, Document> out = new HashMap<String, Document>();
        out.put(part, doc);
        return out;
    }

    private String makeRecord(Storage ss, String id) throws Exception {
        JSONObject obj = getJSON("obj3.json");
        obj.put("accessionNumber", id);
        return ss.autocreateJSON("collection-object/", obj, null);
    }

    @Test
    public void testRelationsThroughAPI() throws Exception {
        Storage ss = makeServicesStorage();
        //create 3 objects
        String obj1 = makeRecord(ss, "A");
        String obj2 = makeRecord(ss, "B");
        String obj3 = makeRecord(ss, "C");

        // relate obj1 and obj2
        String path = relate(ss, obj1, obj2);
        //relate  obj2 and obj3
        String path2 = relate(ss, obj2, obj3);

        // test relationship
        JSONObject data2 = ss.retrieveJSON("relations/main/" + path, new JSONObject());
        assertTrue(JSONUtils.checkJSONEquiv("CollectionObject/" + obj1, data2.getString("src")));
        assertTrue(JSONUtils.checkJSONEquiv("CollectionObject/" + obj2, data2.getString("dst")));
        assertTrue(JSONUtils.checkJSONEquiv("affects", data2.getString("type")));

        // update
        updaterelate(ss, path, obj1, obj3);

        // get
        JSONObject data3 = ss.retrieveJSON("relations/main/" + path, new JSONObject());
        assertTrue(JSONUtils.checkJSONEquiv("CollectionObject/" + obj1, data3.getString("src")));
        assertTrue(JSONUtils.checkJSONEquiv("CollectionObject/" + obj3, data3.getString("dst")));
        assertTrue(JSONUtils.checkJSONEquiv("affects", data3.getString("type")));

        //get list

        JSONObject searchRestriction = new JSONObject();
        searchRestriction.put("src", "collection-object/" + obj2);
        searchRestriction.put("type", "affects");

        // simple list
        //XXX CSPACE-1080 - will need to update if this is improved
        JSONObject datalist = ss.getPathsJSON("relations/main", searchRestriction);
        int truecount = 0;
        String[] paths = (String[]) datalist.get("listItems");
        log.info(datalist.toString());
        JSONObject pagination = datalist.getJSONObject("pagination");
        boolean pagbool = false;
        for (int i = 0; i < paths.length; i++) {
            if (paths[i].equals(path) || paths[i].equals(path2)) {
                truecount++;
            }
        }

        assertTrue(truecount > 0);

        // delete
        ss.deleteJSON("/relations/main/" + path);
        ss.deleteJSON("/relations/main/" + path2);

        // delete objects
        ss.deleteJSON("collection-object/" + obj1);
        ss.deleteJSON("collection-object/" + obj2);
        ss.deleteJSON("collection-object/" + obj3);

    }

    private String relate(Storage ss, String obj1, String obj2)
            throws JSONException, ExistException, UnimplementedException, UnderlyingStorageException {
        JSONObject data = new JSONObject();
        data.put("src", "collection-object/" + obj1);
        data.put("dst", "collection-object/" + obj2);
        data.put("type", "affects");
        // create
        return ss.autocreateJSON("relations/main/", data, null);
    }

    private void updaterelate(Storage ss, String path, String obj1, String obj2)
            throws JSONException, ExistException, UnimplementedException, UnderlyingStorageException {
        JSONObject data = new JSONObject();
        data.put("src", "collection-object/" + obj1);
        data.put("dst", "collection-object/" + obj2);
        data.put("type", "affects");
        // udpate
        ss.updateJSON("relations/main/" + path, data, new JSONObject());
    }

}