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


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.wst.wsdl.Definition;
import org.eclipse.wst.wsdl.internal.impl.DefinitionImpl;
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.w3c.dom.Element;


/**
 * @author Kihup Boo
 */
public class LoadAndSerializationTest extends TestCase
{
  private String PLUGIN_ABSOLUTE_PATH = WSDLTestsPlugin.getInstallURL();

  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/LoadStoreCompare"); // 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());
    }
  }
}