/*
    * 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.uddi.api_v3;
  
  import java.io.StringReader;
  import java.io.StringWriter;
  
  import javax.xml.bind.JAXBContext;
  import javax.xml.bind.JAXBElement;
  import javax.xml.bind.JAXBException;
  import javax.xml.bind.Marshaller;
  import javax.xml.bind.Unmarshaller;
  import javax.xml.namespace.QName;
  import javax.xml.transform.stream.StreamSource;
  
  import static junit.framework.Assert.fail;
  import static junit.framework.Assert.assertEquals;
  import static junit.framework.Assert.assertTrue;
  
  import org.junit.Test;
  import org.uddi.api_v3.AuthToken;
  import org.uddi.api_v3.ObjectFactory;
  
  /**
   * Testing marshalling functionality, making sure UTF-8 is handled correctly.
   * 
   * @author <a href="mailto:kstam@apache.org">Kurt T Stam</a>
   */
  public class AuthInfoTest {
  
  	private final static String EXPECTED_XML_FRAGMENT1 = "<fragment xmlns:ns2=\"http://www.w3.org/2000/09/xmldsig#\" xmlns:ns3=\"urn:uddi-org:api_v3\">\n"
                                                         +"    <ns3:authInfo>AuthInfo String</ns3:authInfo>\n"
                                                         +"</fragment>";
  	private final static String EXPECTED_XML_FRAGMENT2 = "<fragment xmlns:ns3=\"urn:uddi-org:api_v3\" xmlns:ns2=\"http://www.w3.org/2000/09/xmldsig#\">\n"
          +"    <ns3:authInfo>AuthInfo String</ns3:authInfo>\n"
          +"</fragment>";
  	private final static String UTF8_WORD = "メインページ";
  
  	private final static String EXPECTED_UTF8_XML_FRAGMENT1 = "<fragment xmlns:ns2=\"http://www.w3.org/2000/09/xmldsig#\" xmlns:ns3=\"urn:uddi-org:api_v3\">\n"
          +"    <ns3:authInfo>" + UTF8_WORD + "</ns3:authInfo>\n"
          +"</fragment>";
  	private final static String EXPECTED_UTF8_XML_FRAGMENT2 = "<fragment xmlns:ns3=\"urn:uddi-org:api_v3\" xmlns:ns2=\"http://www.w3.org/2000/09/xmldsig#\">\n"
          +"    <ns3:authInfo>" + UTF8_WORD + "</ns3:authInfo>\n"
          +"</fragment>";
  	/**
  	 * Testing going from object to XML using JAXB using a XML Fragment.
  	 */
  	@Test 
  	public void marshall()
  	{
  		try {
  			JAXBContext jaxbContext=JAXBContext.newInstance("org.uddi.api_v3");
  			Marshaller marshaller = jaxbContext.createMarshaller();
  			marshaller.setProperty(Marshaller.JAXB_FORMATTED_OUTPUT, Boolean.TRUE);
  			marshaller.setProperty(Marshaller.JAXB_FRAGMENT, Boolean.TRUE);
  			marshaller.setProperty(Marshaller.JAXB_ENCODING, "UTF-8");
  			ObjectFactory factory = new ObjectFactory();
  			AuthToken authToken = factory.createAuthToken();
  			authToken.setAuthInfo("AuthInfo String");
  			StringWriter writer = new StringWriter();
  			JAXBElement<AuthToken> element = new JAXBElement<AuthToken>(new QName("","fragment"),AuthToken.class,authToken);
  			marshaller.marshal(element,writer);
  			String actualXml=writer.toString();
  			
  			boolean assertion = EXPECTED_XML_FRAGMENT1.equals(actualXml) || EXPECTED_XML_FRAGMENT2.equals(actualXml);
  			assertTrue(assertion);
  		} catch (JAXBException jaxbe) {
  			jaxbe.printStackTrace();
  			
  			fail("No exception should be thrown");
  		}
  	}
  	/**
  	 * Unmarshall an xml fragment.
  	 */
  	@Test 
  	public void unmarshall()
  	{
  		try {
  			JAXBContext jaxbContext=JAXBContext.newInstance("org.uddi.api_v3");
  			Unmarshaller unMarshaller = jaxbContext.createUnmarshaller();
  			StringReader reader = new StringReader(EXPECTED_XML_FRAGMENT1);
  			JAXBElement<AuthToken> element = unMarshaller.unmarshal(new StreamSource(reader),AuthToken.class);
  			String infoString = element.getValue().getAuthInfo();
  			assertEquals("AuthInfo String", infoString);
  		} catch (JAXBException jaxbe) {
  			fail("No exception should be thrown");
 		}
 	}
 	/**
 	 * Test handling of utf8 characters
 	 */
 	@Test
 	public void marshallUTF8()
 	{
 		try {
 			JAXBContext jaxbContext=JAXBContext.newInstance("org.uddi.api_v3");
 			Marshaller marshaller = jaxbContext.createMarshaller();
 			marshaller.setProperty(Marshaller.JAXB_FORMATTED_OUTPUT, Boolean.TRUE);
 			marshaller.setProperty(Marshaller.JAXB_FRAGMENT, Boolean.TRUE);
 			marshaller.setProperty(Marshaller.JAXB_ENCODING, "UTF-8");
 			ObjectFactory factory = new ObjectFactory();
 			AuthToken authToken = factory.createAuthToken();
 			authToken.setAuthInfo(UTF8_WORD);
 			StringWriter writer = new StringWriter();
 			JAXBElement<AuthToken> element = new JAXBElement<AuthToken>(new QName("","fragment"),AuthToken.class,authToken);
 			marshaller.marshal(element,writer);
 			String actualXml=writer.toString();
 
 			boolean assertion = EXPECTED_UTF8_XML_FRAGMENT1.equals(actualXml) || EXPECTED_UTF8_XML_FRAGMENT2.equals(actualXml);
 			assertTrue(assertion);
 		} catch (JAXBException jaxbe) {
 			fail("No exception should be thrown");
 		}
 	}
 	
 	/**
 	 * Test handling of utf8 characters
 	 */
 	@Test
 	public void unmarshallUTF8()
 	{
 		try {
 			JAXBContext jaxbContext=JAXBContext.newInstance("org.uddi.api_v3");
 			Unmarshaller unMarshaller = jaxbContext.createUnmarshaller();
 			StringReader reader = new StringReader(EXPECTED_UTF8_XML_FRAGMENT1);
 			JAXBElement<AuthToken> utf8Element = unMarshaller.unmarshal(new StreamSource(reader),AuthToken.class);
 			String infoString = utf8Element.getValue().getAuthInfo();
 			assertEquals(UTF8_WORD, infoString);
 		} catch (JAXBException jaxbe) {
 			fail("No exception should be thrown");
 		}
 	}
 }