tests/org.eclipse.wst.wsdl.tests/src/org/eclipse/wst/wsdl/tests/extensions/HTTPExtensionsTest.java

/*******************************************************************************
 * Copyright (c) 2007 IBM Corporation and others.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 * 
 * Contributors:
 *     IBM Corporation - initial API and implementation
 *******************************************************************************/

package org.eclipse.wst.wsdl.tests.extensions;


import java.util.List;

import javax.xml.namespace.QName;

import junit.framework.Assert;
import junit.framework.Test;
import junit.framework.TestSuite;

import org.eclipse.emf.common.util.URI;
import org.eclipse.emf.ecore.resource.Resource;
import org.eclipse.emf.ecore.resource.ResourceSet;
import org.eclipse.emf.ecore.resource.impl.ResourceSetImpl;
import org.eclipse.wst.wsdl.Binding;
import org.eclipse.wst.wsdl.BindingInput;
import org.eclipse.wst.wsdl.BindingOperation;
import org.eclipse.wst.wsdl.Definition;
import org.eclipse.wst.wsdl.ExtensibilityElement;
import org.eclipse.wst.wsdl.Port;
import org.eclipse.wst.wsdl.PortType;
import org.eclipse.wst.wsdl.Service;
import org.eclipse.wst.wsdl.binding.http.HTTPAddress;
import org.eclipse.wst.wsdl.binding.http.HTTPBinding;
import org.eclipse.wst.wsdl.binding.http.HTTPFactory;
import org.eclipse.wst.wsdl.binding.http.HTTPOperation;
import org.eclipse.wst.wsdl.binding.http.HTTPPackage;
import org.eclipse.wst.wsdl.binding.http.HTTPUrlEncoded;
import org.eclipse.wst.wsdl.binding.http.HTTPUrlReplacement;
import org.eclipse.wst.wsdl.binding.http.internal.generator.HTTPContentGenerator;
import org.eclipse.wst.wsdl.binding.http.internal.util.HTTPConstants;
import org.eclipse.wst.wsdl.internal.generator.extension.ContentGeneratorExtensionFactoryRegistry;
import org.eclipse.wst.wsdl.tests.WSDLTestsPlugin;
import org.eclipse.wst.wsdl.tests.util.DefinitionLoader;


/**
 * Tests the HTTP binding extensions. 
 */
public class HTTPExtensionsTest extends WSDLExtensionsTest
{
  private static final String ADDRESS_LOCATION_URI = "http://www.example.org/"; //$NON-NLS-1$

  private static final HTTPFactory HTTP_FACTORY = HTTPFactory.eINSTANCE;

  private static final String HTTP_LOCATION_URI = "http://www.example.org/HTTPTest/NewOperation";

  private static final String PLUGIN_ABSOLUTE_PATH = WSDLTestsPlugin.getInstallURL();

  private static final String TARGET_NAMESPACE = "http://www.example.org/HTTPTest/"; //$NON-NLS-1$

  private static final String VERB_GET = "GET"; //$NON-NLS-1$

  private static final String VERB_PUT = "PUT"; //$NON-NLS-1$

  public static void main(String[] args)
  {
    junit.textui.TestRunner.run(suite());
  }

  public static Test suite()
  {
    TestSuite suite = new TestSuite();

    suite.addTest(new HTTPExtensionsTest("HTTPExtensionsCreation") //$NON-NLS-1$
      {
        protected void runTest()
        {
          testHTTPExtensionsCreation();
        }
      });

    suite.addTest(new HTTPExtensionsTest("EMFSerialization") //$NON-NLS-1$
      {
        protected void runTest()
        {
          testEMFSerialization();
        }
      });

    suite.addTest(new HTTPExtensionsTest("HTTPExtensionsReconciliation") //$NON-NLS-1$
      {
        protected void runTest()
        {
          testHTTPExtensionsReconciliation();
        }
      });

    suite.addTest(new HTTPExtensionsTest("HTTPContentGenerator") //$NON-NLS-1$
      {
        protected void runTest()
        {
          testHTTPContentGenerator();
        }
      });

    return suite;
  }

  public HTTPExtensionsTest(String name)
  {
    super(name);
  }

  public void testHTTPExtensionsCreation()
  {
    try
    {
      Definition definition = DefinitionLoader.load(PLUGIN_ABSOLUTE_PATH + "samples/Extensions/HTTP/HTTPTest.wsdl"); //$NON-NLS-1$

      String bindingName = "HTTPTestHTTP"; //$NON-NLS-1$
      QName bindingQName = new QName(TARGET_NAMESPACE, bindingName);
      Binding binding = (Binding)definition.getBinding(bindingQName);

      addHTTPBinding(binding);

      BindingOperation bindingOperation = (BindingOperation)binding.getBindingOperation("NewOperation", null, null); //$NON-NLS-1$

      addHTTPOperation(bindingOperation);

      BindingInput bindingInput = bindingOperation.getEBindingInput();

      addHTTPBindingInput(bindingInput);

      addHTTPAddress(definition);
    }
    catch (Exception e)
    {
      Assert.fail("Test failed due to an exception: " + e.getLocalizedMessage()); //$NON-NLS-1$
    }
  }

  public void testHTTPExtensionsReconciliation()
  {
    try
    {
      Definition definition = DefinitionLoader.load(PLUGIN_ABSOLUTE_PATH + "samples/Extensions/HTTP/HTTPExample.wsdl", true); //$NON-NLS-1$

      String serviceName = "HTTPExample"; //$NON-NLS-1$
      String targetNamespace = "http://www.example.org/HTTPExample/"; //$NON-NLS-1$
      QName serviceQName = new QName(targetNamespace, serviceName);
      Service service = (Service)definition.getService(serviceQName);
      service.toString();

      Port port = (Port)service.getPort("HTTPExampleHTTP"); //$NON-NLS-1$
      List extensibilityElements = port.getExtensibilityElements();
      assertEquals(1, extensibilityElements.size());
      HTTPAddress httpAddress = (HTTPAddress)extensibilityElements.get(0);

      String locationURI = httpAddress.getLocationURI();
      assertEquals("http://www.example.org/", locationURI);

      checkStringAttributeReconciliation(
        httpAddress,
        HTTPConstants.LOCATION_URI_ATTRIBUTE,
        "test",
        HTTPPackage.Literals.HTTP_ADDRESS__LOCATION_URI);

      String bindingName = "HTTPExampleHTTP"; //$NON-NLS-1$
      QName bindingQName = new QName(targetNamespace, bindingName);
      Binding binding = (Binding)definition.getBinding(bindingQName);

      extensibilityElements = binding.getExtensibilityElements();
      assertEquals(1, extensibilityElements.size());
      HTTPBinding httpBinding = (HTTPBinding)extensibilityElements.get(0);

      String verb = httpBinding.getVerb();
      assertEquals(VERB_GET, verb);

      checkStringAttributeReconciliation(httpBinding, HTTPConstants.VERB_ATTRIBUTE, VERB_PUT, HTTPPackage.Literals.HTTP_BINDING__VERB);
    }
    catch (Exception e)
    {
      Assert.fail("Test failed due to an exception: " + e.getLocalizedMessage()); //$NON-NLS-1$
    }
  }

  /**
   * This method loads a WSDL model then saves it using the default EMF serialization
   * instead of the WSDL XML format, then loads it again. The intent is to exercise
   * the EMF e* methods to aid in identifying real code coverage issues.  
   */
  public void testEMFSerialization()
  {
    try
    {
      Definition definition = DefinitionLoader.load(PLUGIN_ABSOLUTE_PATH + "samples/Extensions/HTTP/HTTPExample.wsdl", true); //$NON-NLS-1$

      ResourceSet resourceSet = new ResourceSetImpl();
      URI fileURI = URI.createFileURI(PLUGIN_ABSOLUTE_PATH + "samples/generated/HTTPExample.xml");
      Resource resource = resourceSet.createResource(fileURI);
      resource.getContents().add(definition);
      resource.save(null);
      resourceSet = new ResourceSetImpl();
      resource = resourceSet.getResource(fileURI, true);
    }
    catch (Exception e)
    {
      Assert.fail("Test failed due to an exception: " + e.getLocalizedMessage()); //$NON-NLS-1$
    }
  }

  public void testHTTPContentGenerator()
  {
    try
    {
      Definition definition = DefinitionLoader.load(PLUGIN_ABSOLUTE_PATH + "samples/Extensions/HTTP/HTTPTest.wsdl"); //$NON-NLS-1$

      HTTPContentGenerator contentGenerator = (HTTPContentGenerator)ContentGeneratorExtensionFactoryRegistry.getInstance().getGeneratorClassFromName(
        "HTTP");
      String locationURI = "http://test.org/example"; //$NON-NLS-1$
      contentGenerator.setAddressLocation(locationURI);
      contentGenerator.setVerb(HTTPContentGenerator.VERB_GET);

      String serviceName = "HTTPTest"; //$NON-NLS-1$
      QName serviceQName = new QName(TARGET_NAMESPACE, serviceName);
      Service service = (Service)definition.getService(serviceQName);

      Port port = (Port)service.getPort("HTTPTestHTTP"); //$NON-NLS-1$
      List extensibilityElements = port.getExtensibilityElements();
      assertEquals(0, extensibilityElements.size());
      contentGenerator.generatePortContent(port);

      extensibilityElements = port.getExtensibilityElements();
      assertEquals(1, extensibilityElements.size());
      ExtensibilityElement extensibilityElement = (ExtensibilityElement)extensibilityElements.get(0);
      assertTrue(extensibilityElement instanceof HTTPAddress);
      HTTPAddress httpAddress = (HTTPAddress)extensibilityElement;
      assertEquals(locationURI, httpAddress.getLocationURI());

      QName bindingQName = new QName(TARGET_NAMESPACE, "HTTPTestHTTP"); //$NON-NLS-1$
      Binding binding = (Binding)definition.getBinding(bindingQName);

      QName portTypeQName = new QName(TARGET_NAMESPACE, "HTTPTest"); //$NON-NLS-1$
      PortType portType = (PortType)definition.getPortType(portTypeQName);

      contentGenerator.generateBindingContent(binding, portType);

      // TODO Complete this test.
    }
    catch (Exception e)
    {
      Assert.fail("Test failed due to an exception: " + e.getLocalizedMessage()); //$NON-NLS-1$
    }

  }

  private void addHTTPAddress(Definition definition)
  {
    HTTPAddress httpAddress = HTTP_FACTORY.createHTTPAddress();
    httpAddress.setLocationURI(ADDRESS_LOCATION_URI);

    httpAddress.toString();

    String serviceName = "HTTPTest"; //$NON-NLS-1$
    QName serviceQName = new QName(TARGET_NAMESPACE, serviceName);
    Service service = (Service)definition.getService(serviceQName);
    service.toString();

    Port port = (Port)service.getPort("HTTPTestHTTP"); //$NON-NLS-1$
    port.addExtensibilityElement(httpAddress);
    port.toString();
  }

  private void addHTTPBinding(Binding binding)
  {
    binding.toString();

    HTTPBinding httpBinding = HTTP_FACTORY.createHTTPBinding();
    binding.addExtensibilityElement(httpBinding);

    httpBinding.setVerb(VERB_GET);

    httpBinding.toString();
  }

  private void addHTTPBindingInput(BindingInput bindingInput)
  {
    bindingInput.toString();

    HTTPUrlEncoded httpURLEncoded = HTTP_FACTORY.createHTTPUrlEncoded();
    bindingInput.addExtensibilityElement(httpURLEncoded);

    bindingInput.getExtensibilityElements().clear();

    HTTPUrlReplacement httpURLReplacement = HTTP_FACTORY.createHTTPUrlReplacement();
    bindingInput.addExtensibilityElement(httpURLReplacement);
  }

  private void addHTTPOperation(BindingOperation bindingOperation)
  {
    bindingOperation.toString();

    HTTPOperation httpOperation = HTTP_FACTORY.createHTTPOperation();

    httpOperation.setLocationURI(HTTP_LOCATION_URI);

    httpOperation.toString();

    bindingOperation.addExtensibilityElement(httpOperation);
  }
}