src/test/java/org/apache/ws/scout/registry/OwnershipTest.java

/*
 * Copyright 2001-2009 The Apache Software Foundation.
 * 
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 * 
 *      http://www.apache.org/licenses/LICENSE-2.0
 * 
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package org.apache.ws.scout.registry;

import static org.junit.Assert.fail;

import java.util.ArrayList;
import java.util.Collection;
import java.util.Iterator;
import java.util.Locale;

import javax.xml.registry.BulkResponse;
import javax.xml.registry.BusinessLifeCycleManager;
import javax.xml.registry.BusinessQueryManager;
import javax.xml.registry.JAXRException;
import javax.xml.registry.JAXRResponse;
import javax.xml.registry.LifeCycleManager;
import javax.xml.registry.RegistryService;
import javax.xml.registry.infomodel.InternationalString;
import javax.xml.registry.infomodel.Key;
import javax.xml.registry.infomodel.Organization;

import org.apache.ws.scout.BaseTestCase;
import org.junit.After;
import org.junit.Before;
import org.junit.Test;

/**
 *
 * Test for SCOUT-55 which attempts to test whether getRegistryObjects() returns caller structures, or all structures. 
 * At the moment there appears no way of implementing this using the UDDI API, so the actual test is being checked in
 * commented out (it will fail).    Would be good to have if we can figure out how to fix SCOUT-55.
 *
 * @author <a href="mailto:tcunning@apache.org">Tom Cunningham</a>
 */
public class OwnershipTest extends BaseTestCase {
	RegistryService rs, rs2;
	private BusinessQueryManager bqm, bqm2;
	@SuppressWarnings("unused")
	private BusinessLifeCycleManager blm, blm2;
	Collection associationKeys = null;
	
    @Before
    public void setUp() {
        super.setUp();
        login();
        loginSecondUser();
		try {
			rs = connection.getRegistryService();
	    	bqm = rs.getBusinessQueryManager();
	    	blm = rs.getBusinessLifeCycleManager();
	        
	    	rs2 = connection2.getRegistryService();
	        blm2 = rs2.getBusinessLifeCycleManager();
	        bqm2 = rs2.getBusinessQueryManager();

		} catch (JAXRException e) {
			fail(e.getMessage());
		}
    }
    
    @After
    public void tearDown() {
        super.tearDown();

    }
    
    private InternationalString getIString(String str)
    throws JAXRException
    {
        return blm.createInternationalString(str);
    }

    
    private Organization createTempOrg(String tempOrgName, BusinessLifeCycleManager blm) throws JAXRException {
        Key orgKey = null;
        Organization org = blm.createOrganization(getIString(tempOrgName));
        org.setDescription(getIString("Temporary organization to test saveService()"));

        Collection<Organization> orgs = new ArrayList<Organization>();
        orgs.add(org);
        BulkResponse br = blm.saveOrganizations(orgs);

        if (br.getStatus() == JAXRResponse.STATUS_SUCCESS)
        {
            orgKey = (Key) br.getCollection().iterator().next();
            System.out.println("Temporary Organization Created with id=" + orgKey.getId());
            org.setKey(orgKey);
        }  else
        {
            System.err.println("JAXRExceptions " +
                    "occurred during creation of temporary organization:");

            Iterator iter = br.getCollection().iterator();

            while (iter.hasNext()) {
                Exception e = (Exception) iter.next();
                System.err.println(e.toString());
            }

            fail();
        }
        return org;
    }

    private void deleteTempOrg(Key key) throws JAXRException {

        if (key == null) {
                return;
        }

        String id = key.getId();

        System.out.println("\nDeleting temporary organization with id " + id + "\n");

        Collection<Key> keys = new ArrayList<Key>();
        keys.add(key);
        BulkResponse response = blm.deleteOrganizations(keys);

        Collection exceptions = response.getExceptions();
        if (exceptions == null) {
                Collection retKeys = response.getCollection();
                Iterator keyIter = retKeys.iterator();
                Key orgKey = null;
                if (keyIter.hasNext()) {
                        orgKey =
                                (javax.xml.registry.infomodel.Key) keyIter.next();
                        id = orgKey.getId();
                        System.out.println("Organization with ID=" + id + " was deleted");
                }
        }
    }
    
    @Test
	public void testGetRegistryObjects() {

    	login();
        try {

        	Organization org1 = createTempOrg("OWNERSHIPTEST Organization 1", blm);
        	Organization org2 = createTempOrg("OWNERSHIPTEST Organization 2", blm);
        	
        	BulkResponse br = bqm.getRegistryObjects(LifeCycleManager.ORGANIZATION);
            if (br.getStatus() == JAXRResponse.STATUS_SUCCESS)
            {
            	System.out.println("BR.size : " + br.getCollection().size());
            	Collection coll1 = br.getCollection();
            	Iterator iterator = coll1.iterator();
            	Collection<String> orgKeys = new ArrayList<String>();
            	while (iterator.hasNext()) {
            		Organization org = (Organization) iterator.next();
            		System.out.println("BR " + org.getName().getValue(Locale.US));
            		orgKeys.add(org.getKey().getId());
            	}

            	if (br.getCollection().size() == 0) {
            		fail("Found no organizations for user 1");
            	}
            	
            	BulkResponse br2 = bqm2.getRegistryObjects(LifeCycleManager.ORGANIZATION);
                if (br2.getStatus() == JAXRResponse.STATUS_SUCCESS)
                {
                	Collection coll = br2.getCollection();
                	Iterator iterator2 = coll.iterator();
                	while (iterator2.hasNext()) {
                		Organization org = (Organization) iterator2.next();
                		System.out.println("BR2 " + org.getName().getValue(Locale.US));
                		if (orgKeys.contains(org.getKey().getId())) {
                			fail("Should not find organizations from user1");
                		}
                		
                	}
                }        	
            	
            	deleteTempOrg(org1.getKey());
            	deleteTempOrg(org2.getKey());

            }
        	
        	
        
		} catch (JAXRException e) {
			e.printStackTrace();
		}
		
	}
}