tests/org.eclipse.wst.wsdl.tests/src/org/eclipse/wst/wsdl/tests/LoadAndSerializationTest.java

/*******************************************************************************
 * Copyright (c) 2001, 2004 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;

import java.io.File;
import java.io.FileFilter;
import java.util.Vector;

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

import org.eclipse.emf.ecore.resource.Resource;
import org.eclipse.wst.wsdl.Definition;
import org.eclipse.wst.wsdl.WSDLPackage;
import org.eclipse.wst.wsdl.internal.impl.DefinitionImpl;
import org.eclipse.wst.wsdl.internal.util.WSDLResourceFactoryImpl;
import org.eclipse.wst.wsdl.tests.util.DefinitionLoader;
import org.eclipse.wst.wsdl.tests.util.XMLDiff;
import org.eclipse.wst.wsdl.util.WSDLResourceImpl;
import org.eclipse.xsd.XSDPackage;
import org.eclipse.xsd.util.XSDResourceFactoryImpl;
import org.w3c.dom.Element;

/**
 * @author Kihup Boo
 */
public class LoadAndSerializationTest extends TestCase
{ 
  {	    
    Resource.Factory.Registry.INSTANCE.getExtensionToFactoryMap().put("wsdl", new WSDLResourceFactoryImpl());
    WSDLPackage pkg = WSDLPackage.eINSTANCE;
	    
    // We need this for XSD <import>.
    Resource.Factory.Registry.INSTANCE.getExtensionToFactoryMap().put("xsd", new XSDResourceFactoryImpl());
    XSDPackage xsdpkg = XSDPackage.eINSTANCE; 
  }
  
  private String PLUGIN_ABSOLUTE_PATH = WSDLTestsPlugin.getInstallURL();
  private String TEST_DATA_DIR;
  
  //static private File[] wsdls;
  private Vector wsdlFiles = new Vector();
  static private Definition definition = null;
	
  /**
   * Executes a stand-alone test.
   * @param objects an array of Strings from the command line.
   * @see #run
   */
  public static void main(String args[]) 
  {
    junit.textui.TestRunner.run(suite());
  }
  
  public LoadAndSerializationTest(String name) 
  {
    super(name);
  }
  
  public static Test suite() 
  {
    TestSuite suite = new TestSuite();
    suite.addTest
    (new LoadAndSerializationTest("LoadAndStore") 
      {
        protected void runTest() 
        {
          testLoadAndStore();
        }
      }
    );
    suite.addTest // wtp bug 79326
    (new LoadAndSerializationTest("Compare") 
      {
        protected void runTest() 
        {
          testCompare();
        }
      }
    );
    return suite;
  } 

  /**
   * Load from the WSDL definitions file and store back to a different file.
   */
  public void testLoadAndStore()
  {
	String TEST_DATA_DIR = System.getProperty("testDataDir");
	//Assert.assertNotNull(testDataDir);
	File dir = null;
	if (TEST_DATA_DIR != null)
		dir = new File(TEST_DATA_DIR);
	else	
		// KB: if you are here, fix text.xml
        dir = new File(PLUGIN_ABSOLUTE_PATH + "samples"); // fallback
	
    if (dir.exists() && dir.isDirectory())
    {
      listDataFiles(dir);

      try 
      {
        File myFile = null;
        for (int i = 0; i < wsdlFiles.size(); i++)
        {
          myFile = (File)wsdlFiles.elementAt(i);
          System.out.println(myFile.toURL().toString());
          load(myFile.toString());
          print(myFile.toString() + ".out");
        }
      }
      catch (Exception e)
      {
        Assert.fail("Test failed due to an exception: " + e.getLocalizedMessage());
      }
    }
    else
      fail(dir.toString());

  }
  
	private void listDataFiles(File src)  
	{
		if (!src.isDirectory())
			return; // Assertion failed
		
		filterWSDLFiles(src); // Add WSDL files in the src directory
		
		File[] children = src.listFiles();
		File myFile;
		for (int i = 0; i < children.length; i++) 
		{
			myFile = children[i];			
			if (myFile.isDirectory()) 
			{
				if ("CVS".equals(myFile.getName()))
				  continue;
				
				listDataFiles(myFile); // Visit sub-directories recursively
			} 
		}
	}
	
	private void filterWSDLFiles(File dir)
	{
		File[] wsdls = dir.listFiles
	      (
	        new FileFilter()
	        {
	          public boolean accept(File pathname)
	          {
	            return pathname.getName().endsWith(".wsdl");
	          }
	        }
	      );
		
		for (int j=0; j<wsdls.length; j++)
		{
			wsdlFiles.add(wsdls[j]);
		}
	}
  
  /*
   * Load from the WSDL definitions file.
   */
  private void load(String filename)
  {	
    try
    {
      definition = DefinitionLoader.load(filename);
	  Assert.assertNotNull(definition);
      Assert.assertTrue(definition.eResource() instanceof WSDLResourceImpl);
	  
      WSDLResourceImpl wsdlResource = (WSDLResourceImpl)definition.eResource();
	  Assert.assertNotNull(wsdlResource);

	  Element element = definition.getElement();
      Assert.assertNotNull(definition.getElement());
	  
      WSDLResourceImpl.serialize(System.out, element);
    }
    catch (Exception e)
    {
      Assert.fail("Test failed due to an exception: " + e.getLocalizedMessage());
    }  
  }
  
  /*
   * Store the WSDL definitions to a file.
   */  
  private void print(String filename)
  {
    try
    {
      // This removes the associated DOM element, creates a new associated DOM element, and then prints it.
      // This is a good test for how well serialization works for a model created "bottom up".
      //
      definition.setDocument(null);
      definition.setElement(null);
      ((DefinitionImpl)definition).updateElement();

	  DefinitionLoader.store(definition,filename);
    }
    catch (Exception e)
    {
      Assert.fail("Test failed due to an exception: " + e.getLocalizedMessage());
    }   
  }
  
  /**
   * Compare the output WSDL file to the original.
   */  
  public void testCompare()
  {
    XMLDiff xmldiff = new XMLDiff();
    try
    {
      File myFile = null;
      for (int i = 0; i < wsdlFiles.size(); i++)
      {
    	myFile = (File)wsdlFiles.elementAt(i);
        Assert.assertTrue(xmldiff.diff(myFile.toString(),myFile.toString() + ".out")); 
      }
    }
    catch (Exception e)
    {
      Assert.fail("Test failed due to an exception: " + e.getLocalizedMessage());
    }
  }      
}