/**
    *
    * 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.assertNull;
  import static org.junit.Assert.assertTrue;
  
  import java.util.ArrayList;
  import java.util.Collection;
  import java.util.Iterator;
  
  import javax.xml.registry.BulkResponse;
  import javax.xml.registry.FindQualifier;
  import javax.xml.registry.JAXRException;
  import javax.xml.registry.JAXRResponse;
  import static javax.xml.registry.LifeCycleManager.PERSON_NAME;
  import javax.xml.registry.RegistryService;
  import javax.xml.registry.infomodel.Classification;
  import javax.xml.registry.infomodel.ClassificationScheme;
  import javax.xml.registry.infomodel.EmailAddress;
  import javax.xml.registry.infomodel.Key;
  import javax.xml.registry.infomodel.Organization;
  import javax.xml.registry.infomodel.PersonName;
  import javax.xml.registry.infomodel.PostalAddress;
  import javax.xml.registry.infomodel.Service;
  import javax.xml.registry.infomodel.ServiceBinding;
  import javax.xml.registry.infomodel.TelephoneNumber;
  import javax.xml.registry.infomodel.User;
  
  import junit.framework.JUnit4TestAdapter;
  
  import org.apache.ws.scout.BaseTestCase;
  import org.apache.ws.scout.Creator;
  import org.apache.ws.scout.Finder;
  import org.apache.ws.scout.Printer;
  import org.apache.ws.scout.Remover;
  import static org.apache.ws.scout.registry.qa.JAXR015PrimaryContactTest.CITY;
  import static org.apache.ws.scout.registry.qa.JAXR015PrimaryContactTest.COUNTRY;
  import static org.apache.ws.scout.registry.qa.JAXR015PrimaryContactTest.EMAIL;
  import static org.apache.ws.scout.registry.qa.JAXR015PrimaryContactTest.PHONE_NUMBER;
  import static org.apache.ws.scout.registry.qa.JAXR015PrimaryContactTest.POSTAL_CODE;
  import static org.apache.ws.scout.registry.qa.JAXR015PrimaryContactTest.STATE;
  import static org.apache.ws.scout.registry.qa.JAXR015PrimaryContactTest.STREET;
  import static org.apache.ws.scout.registry.qa.JAXR015PrimaryContactTest.STREET_NUMBER;
  import org.junit.After;
  import static org.junit.Assert.assertEquals;
  import static org.junit.Assert.fail;
  import org.junit.Before;
  import org.junit.Test;
  
  /**
   * Test to check Jaxr Publish
   * Open source UDDI Browser  http://www.uddibrowser.org or using the juddi-gui project
   * can be used to check your results
   * @author <mailto:anil@apache.org>Anil Saldhana
   * @since Nov 20, 2004
   */
  public class JAXR010OrganizationTest extends BaseTestCase
  {
      @Before
      public void setUp() {
          super.setUp();
          login();
          try {
              RegistryService rs = connection.getRegistryService();
              bqm = rs.getBusinessQueryManager();
              blm = rs.getBusinessLifeCycleManager();
              ClassificationScheme cScheme = blm.createClassificationScheme("org.jboss.soa.esb.:testcategory", "JBossESB Classification Scheme");
              ArrayList<ClassificationScheme> cSchemes = new ArrayList<ClassificationScheme>();
              cSchemes.add(cScheme);
              BulkResponse br = blm.saveClassificationSchemes(cSchemes);
              assertEquals(JAXRResponse.STATUS_SUCCESS, br.getStatus());
          } catch (Exception je) {
              je.printStackTrace();
              fail(je.getMessage());
          }
      }
  
      @After
      public void tearDown() {
          super.tearDown();
          login();
          try {
              RegistryService rs = connection.getRegistryService();
              bqm = rs.getBusinessQueryManager();
             blm = rs.getBusinessLifeCycleManager();
             Collection<String> findQualifiers = new ArrayList<String>();
             findQualifiers.add(FindQualifier.AND_ALL_KEYS);
             //findQualifiers.add(FindQualifier.SORT_BY_NAME_DESC);
             ClassificationScheme cScheme = bqm.findClassificationSchemeByName(findQualifiers, "org.jboss.soa.esb.:testcategory");
             Removerver.html#Remover">Remover remover = new Remover(blm);
             remover.removeClassificationScheme(cScheme);
         } catch (Exception je) {
             je.printStackTrace();
             fail(je.getMessage());
         }
     }
     
     @Test 
     public void publishClassificationScheme()
     {
         login();
         try
         {
             RegistryService rs = connection.getRegistryService();
             blm = rs.getBusinessLifeCycleManager();
             Creatortor.html#Creator">Creator creator = new Creator(blm);
             
             Collection<ClassificationScheme> schemes = new ArrayList<ClassificationScheme>();
             ClassificationScheme classificationScheme = creator.createClassificationScheme(this.getClass().getName());
             schemes.add(classificationScheme);
             
             BulkResponse bulkResponse = blm.saveClassificationSchemes(schemes);
             assertEquals(JAXRResponse.STATUS_SUCCESS,bulkResponse.getStatus());
             
             
         } catch (JAXRException e) {
             e.printStackTrace();
             assertTrue(false);
         }   
     }
     
     @Test
     public void publishOrganization()
     {
         BulkResponse response = null;
         login();
         try
         {
             RegistryService rs = connection.getRegistryService();
             blm = rs.getBusinessLifeCycleManager();
             bqm = rs.getBusinessQueryManager();
             Creatortor.html#Creator">Creator creator = new Creator(blm);
             Findernder.html#Finder">Finder finder = new Finder(bqm, uddiversion);
             
             Collection<Organization> orgs = new ArrayList<Organization>();
             Organization organization = creator.createOrganization(this.getClass().getName());
 //          Add a Service
             Service service = creator.createService(this.getClass().getName());
             ServiceBinding serviceBinding = creator.createServiceBinding();
             service.addServiceBinding(serviceBinding);
             organization.addService(service);
             //Add a classification
             ClassificationScheme cs = finder.findClassificationSchemeByName(this.getClass().getName());
             Classification classification = creator.createClassification(cs);
             organization.addClassification(classification);
             
             orgs.add(organization);
 
             //Now save the Organization along with a Service, ServiceBinding and Classification
             BulkResponse br = blm.saveOrganizations(orgs);
             if (br.getStatus() == JAXRResponse.STATUS_SUCCESS)
             {
                 System.out.println("Organization Saved");
                 Collection coll = br.getCollection();
                 Iterator iter = coll.iterator();
                 while (iter.hasNext())
                 {
                     Key key = (Key) iter.next();
                     System.out.println("Saved Key=" + key.getId());
                 }//end while
             } else
             {
                 System.err.println("JAXRExceptions " +
                         "occurred during save:");
                 Collection exceptions = br.getExceptions();
                 if (exceptions!=null) {
                     Iterator iter = exceptions.iterator();
                     while (iter.hasNext())
                     {
                         Exception e = (Exception) iter.next();
                         System.err.println(e.toString());
                     }
                 }
             }
             
         } catch (JAXRException e) {
             e.printStackTrace();
 			assertTrue(false);
         }
         assertNull(response);
     }
     
     @Test
     public void queryOrganization()
     {
         login();
         try
         {
             RegistryService rs = connection.getRegistryService();
             blm = rs.getBusinessLifeCycleManager();
             bqm = rs.getBusinessQueryManager();
             Creatortor.html#Creator">Creator creator = new Creator(blm);
             Findernder.html#Finder">Finder finder = new Finder(bqm, uddiversion);
 
             Collection<Organization> orgs = new ArrayList<Organization>();
             Organization organization = creator.createOrganization(this.getClass().getName());
 //          Add a Service
             Service service = creator.createService(this.getClass().getName());
             ServiceBinding serviceBinding = creator.createServiceBinding();
             service.addServiceBinding(serviceBinding);
             organization.addService(service);
             
 
             User user = blm.createUser();
             PersonName personName = blm.createPersonName(PERSON_NAME);
             TelephoneNumber telephoneNumber = blm.createTelephoneNumber();
             telephoneNumber.setNumber(PHONE_NUMBER);
             telephoneNumber.setType(null);
             PostalAddress address = blm.createPostalAddress(STREET_NUMBER,
                     STREET, CITY, STATE, COUNTRY, POSTAL_CODE, "");
 
             Collection<PostalAddress> postalAddresses = new ArrayList<PostalAddress>();
             postalAddresses.add(address);
             Collection<EmailAddress> emailAddresses = new ArrayList<EmailAddress>();
             EmailAddress emailAddress = blm.createEmailAddress(EMAIL);
             emailAddresses.add(emailAddress);
 
             Collection<TelephoneNumber> numbers = new ArrayList<TelephoneNumber>();
             numbers.add(telephoneNumber);
             user.setPersonName(personName);
             user.setPostalAddresses(postalAddresses);
             user.setEmailAddresses(emailAddresses);
             user.setTelephoneNumbers(numbers);
             organization.setPrimaryContact(user);
 
             orgs.add(organization);
 
             //Now save the Organization along with a Service, ServiceBinding and Classification
             BulkResponse br = blm.saveOrganizations(orgs);
             
             // Get registry service and business query manager
              rs = connection.getRegistryService();
             bqm = rs.getBusinessQueryManager();
             System.out.println("We have the Business Query Manager");
             Printerter.html#Printer">Printer printer = new Printer();
              finder = new Finder(bqm, uddiversion);
 
              orgs = finder.findOrganizationsByName(this.getClass().getName());
             if (orgs == null) {
                 fail("Only Expecting 1 Organization");
             } else {
                 assertTrue(orgs.size()>=1);
                 // then step through them
                 for (Iterator orgIter = orgs.iterator(); orgIter.hasNext();)
                 {
                     Organization org = (Organization) orgIter.next();
                     System.out.println("Org name: " + printer.getName(org));
                     System.out.println("Org description: " + printer.getDescription(org));
                     System.out.println("Org key id: " + printer.getKey(org));
 
                     printer.printUser(org);
                     printer.printServices(org);
                     printer.printClassifications(org);
                 }
             }//end else
         } catch (JAXRException e) {
             e.printStackTrace();
             fail(e.getMessage());
         } 
     }
     
     @Test
     public void deleteOrganization()
     {
         login();
         try
         {
             RegistryService rs = connection.getRegistryService();
             blm = rs.getBusinessLifeCycleManager();
     //      Get registry service and business query manager
             bqm = rs.getBusinessQueryManager();
             System.out.println("We have the Business Query Manager");
             Findernder.html#Finder">Finder finder = new Finder(bqm, uddiversion);
             Removerver.html#Remover">Remover remover = new Remover(blm);
             Collection orgs = finder.findOrganizationsByName(this.getClass().getName());
             for (Iterator orgIter = orgs.iterator(); orgIter.hasNext();)
             {
                 Organization org = (Organization) orgIter.next();
                 remover.removeOrganization(org);
             }
             
         } catch (Exception e) {
             e.printStackTrace();
             fail(e.getMessage());
         }
     }
     
     @Test
     public void deleteClassificationScheme()
     {
         login();
         try {
             RegistryService rs = connection.getRegistryService();
             bqm = rs.getBusinessQueryManager();
             blm = rs.getBusinessLifeCycleManager();
             System.out.println("We have the Business Query Manager");
             Findernder.html#Finder">Finder finder = new Finder(bqm, uddiversion);
             Removerver.html#Remover">Remover remover = new Remover(blm);
             Collection schemes = finder.findClassificationSchemesByName(this.getClass().getName());
             for (Iterator iter = schemes.iterator(); iter.hasNext();)
             {
                 ClassificationScheme scheme = (ClassificationScheme) iter.next();
                 remover.removeClassificationScheme(scheme);
             }
             
         } catch (Exception e) {
             e.printStackTrace();
             fail(e.getMessage());
         }
     }
     
     public static junit.framework.Test suite() {
         return new JUnit4TestAdapter(JAXR010OrganizationTest.class);
     }
 }