/**
*
* Copyright 2004 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.qa;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.fail;
import java.util.ArrayList;
import java.util.Collection;
import java.util.Iterator;
import javax.xml.registry.BulkResponse;
import javax.xml.registry.BusinessQueryManager;
import javax.xml.registry.FindQualifier;
import javax.xml.registry.JAXRException;
import javax.xml.registry.JAXRResponse;
import javax.xml.registry.RegistryService;
import javax.xml.registry.infomodel.Association;
import javax.xml.registry.infomodel.Concept;
import javax.xml.registry.infomodel.Key;
import javax.xml.registry.infomodel.Organization;
import javax.xml.registry.infomodel.RegistryObject;
import junit.framework.JUnit4TestAdapter;
import org.apache.ws.scout.BaseTestCase;
import org.apache.ws.scout.Creator;
import org.junit.After;
import org.junit.Before;
import org.junit.Test;
/**
* Tests Publish, Delete (and indirectly, find) for associations.
*
* You can comment out the deletion portion and use Open source UDDI Browser
* <http://www.uddibrowser.org> to check your intermediate results.
*
* Based on query/publish tests written by <a href="mailto:anil@apache.org">Anil
* Saldhana</a>.
*
* @author <a href="mailto:dbhole@redhat.com">Deepak Bhole</a>
* @author <a href="mailto:anil@apache.org">Anil Saldhana</a>
*
* @since Sep 27, 2005
*/
public class JAXR030AssociationsTest extends BaseTestCase {
private BusinessQueryManager bqm = null;
String associationType = "AssociationType/RelatedTo";
private static String tempSrcOrgName = "Apache Source Org -- APACHE SCOUT TEST";
private static String tempTgtOrgName = "Apache Target Org -- APACHE SCOUT TEST";
private Organization sOrg=null;
private Organization tOrg=null;
@Before
public void setUp() {
super.setUp();
}
@After
public void tearDown() {
super.tearDown();
}
/**
* Tests publishing and deleting of associations.
*
* Do not break this into testPublish(), testDelete(), etc. Order is
* important, and not all jvms can guarantee order since the JUnit framework
* uses getMethods() to gather test methods, and getMethods() does not
* guarantee order.
*/
@Test
public void testPublishFindAndDeleteAssociation() {
login();
try {
RegistryService rs = connection.getRegistryService();
bqm = rs.getBusinessQueryManager();
blm = rs.getBusinessLifeCycleManager();
//deleting any pre-exisiting organizations
ArrayList<Organization> orgs = findTempOrgs();
for (Organization organization : orgs) {
Collection<Key> keys = new ArrayList<Key>();
keys.add(organization.getKey());
blm.deleteOrganizations(keys);
}
Creator creator = new Creator(blm);
System.out.println("\nCreating temporary organizations...\n");
Organization org1 = creator.createOrganization(tempSrcOrgName);
Organization org2 = creator.createOrganization(tempTgtOrgName);
Collection<Organization> organizations = new ArrayList<Organization>();
organizations.add(org1);
organizations.add(org2);
blm.saveOrganizations(organizations);
System.out.println("\nSearching for newly created organizations...\n");
ArrayList<Organization> newOrgs = findTempOrgs();
sOrg = newOrgs.get(0);
tOrg = newOrgs.get(1);
System.out.println("\nCreating association...\n");
createAssociation(sOrg, tOrg);
// All created ... now try to delete.
String associationID = findAndDeleteAssociation();
//Let us look for associations now
BulkResponse associationResp =
bqm.findCallerAssociations(null, Boolean.TRUE, Boolean.TRUE, null);
if(associationResp.getExceptions() != null)
{
System.out.println(associationResp.getExceptions());
fail("Association lookup failed");
}
else
{
Collection retAssocs = associationResp.getCollection();
if (retAssocs.size() == 0)
{
//Pass
} else
{
Iterator iterAss = retAssocs.iterator();
while(iterAss.hasNext())
{
Association assc = (Association) iterAss.next();
if(assc.getKey().getId().equals(associationID)) {
System.out.println("found: " + associationID);
fail("Deleted Association found");
}
}
}
}
} catch (Exception e) {
e.printStackTrace();
fail(e.getMessage());
}
}
private void createAssociation(Organization sOrg, Organization tOrg)
throws JAXRException {
Concept type = bqm.findConceptByPath(associationType);
Association association = blm.createAssociation(tOrg, type);
sOrg.addAssociation(association);
ArrayList<Association> associations = new ArrayList<Association>();
associations.add(association);
BulkResponse br = blm.saveAssociations(associations, true);
if (br.getStatus() == JAXRResponse.STATUS_SUCCESS) {
System.out.println("Association Saved");
Collection coll = br.getCollection();
Iterator iter = coll.iterator();
while (iter.hasNext()) {
System.out.println("Saved Key=" + iter.next());
}// end while
} else {
System.err.println("JAXRExceptions " + "occurred during save:");
Collection exceptions = br.getExceptions();
Iterator iter = exceptions.iterator();
while (iter.hasNext()) {
Exception e = (Exception) iter.next();
System.err.println(e.toString());
}
deleteTempOrgs();
}
}
private String findAndDeleteAssociation() throws JAXRException {
String id = null;
String sOrgID = sOrg.getKey().getId();
String tOrgID = tOrg.getKey().getId();
Collection<String> findQualifiers = new ArrayList<String>();
findQualifiers.add(FindQualifier.SORT_BY_NAME_ASC);
Concept type = bqm.findConceptByPath(associationType);
ArrayList<Concept> conceptTypes = new ArrayList<Concept>(1);
conceptTypes.add(type);
BulkResponse br = bqm.findAssociations(findQualifiers, sOrgID, tOrgID,
conceptTypes);
Collection associations = br.getCollection();
if (associations == null) {
System.out.println("\n-- Matched 0 orgs");
fail("Expected 1 association");
} else {
System.out.println("\n-- Matched " + associations.size()
+ " associations --\n");
assertEquals(1,associations.size());
// then step through them
for (Iterator conceptIter = associations.iterator(); conceptIter
.hasNext();) {
Association a = (Association) conceptIter.next();
System.out.println("Id: " + a.getKey().getId());
System.out.println("Name: " + a.getName().getValue());
// Print spacer between messages
System.out.println(" --- ");
id = a.getKey().getId();
deleteAssociation(a.getKey());
System.out.println("\n ============================== \n");
}
}
return id;
}
private void deleteAssociation(Key key) throws JAXRException {
String id = key.getId();
System.out.println("\nDeleting association with id " + id + "\n");
Collection<Key> keys = new ArrayList<Key>();
keys.add(key);
BulkResponse response = blm.deleteAssociations(keys);
assertEquals(BulkResponse.STATUS_SUCCESS, response.getStatus());
Collection exceptions = response.getExceptions();
if (exceptions == null) {
Collection retKeys = response.getCollection();
Iterator keyIter = retKeys.iterator();
javax.xml.registry.infomodel.Key orgKey = null;
if (keyIter.hasNext()) {
orgKey = (javax.xml.registry.infomodel.Key) keyIter.next();
id = orgKey.getId();
System.out
.println("Association with ID=" + id + " was deleted");
}
}
}
private ArrayList<Organization> findTempOrgs() throws JAXRException {
ArrayList<Organization> toReturn = new ArrayList<Organization>();
// Define find qualifiers and name patterns
Collection<String> findQualifiers = new ArrayList<String>();
findQualifiers.add(FindQualifier.SORT_BY_NAME_ASC);
Collection<String> namePatterns = new ArrayList<String>();
if ("3.0".equals(uddiversion)) {
namePatterns.add(tempSrcOrgName);
namePatterns.add(tempTgtOrgName);
} else {
namePatterns.add("%" + tempSrcOrgName + "%");
namePatterns.add("%" + tempTgtOrgName + "%");
}
// Find based upon qualifier type and values
System.out.println("\n-- searching the registry --\n");
BulkResponse response = bqm.findOrganizations(findQualifiers,
namePatterns, null, null, null, null);
// check how many organisation we have matched
Collection orgs = response.getCollection();
if (orgs == null) {
System.out.println("\n-- Matched 0 orgs");
} else {
System.out.println("\n-- Matched " + orgs.size()
+ " organisations --\n");
// then step through them
for (Iterator orgIter = orgs.iterator(); orgIter.hasNext();) {
Organization org = (Organization) orgIter.next();
System.out.println("Org name: " + getName(org));
System.out.println("Org description: " + getDescription(org));
System.out.println("Org key id: " + getKey(org));
if (getName(org).indexOf(tempSrcOrgName) > -1) {
toReturn.add(0, org);
} else {
toReturn.add(1, org);
}
// Print spacer between organizations
System.out.println("\n ============================== \n");
}
}// end else
return toReturn;
}
private void deleteTempOrgs() {
try {
Key sOrgKey = sOrg.getKey();
Key tOrgKey = tOrg.getKey();
System.out.println("\nDeleting temporary organizations with ids "
+ sOrgKey + " and " + tOrgKey + "\n");
Collection<Key> keys = new ArrayList<Key>();
keys.add(sOrgKey);
keys.add(tOrgKey);
BulkResponse response = blm.deleteOrganizations(keys);
Collection exceptions = response.getExceptions();
if (exceptions == null) {
Collection retKeys = response.getCollection();
Iterator keyIter = retKeys.iterator();
Key orgKey = null;
while (keyIter.hasNext()) {
orgKey = (javax.xml.registry.infomodel.Key) keyIter.next();
String id = orgKey.getId();
System.out.println("Organization with ID=" + id
+ " was deleted");
}
}
} catch (JAXRException jaxre) {
jaxre.printStackTrace();
}
}
private static String getName(RegistryObject ro) throws JAXRException {
if (ro != null && ro.getName() != null) {
return ro.getName().getValue();
}
return "";
}
private static String getDescription(RegistryObject ro)
throws JAXRException {
if (ro != null && ro.getDescription() != null) {
return ro.getDescription().getValue();
}
return "";
}
private static String getKey(RegistryObject ro) throws JAXRException {
if (ro != null && ro.getKey() != null) {
return ro.getKey().getId();
}
return "";
}
public static junit.framework.Test suite() {
return new JUnit4TestAdapter(JAXR030AssociationsTest.class);
}
}