package org.apache.velocity.test;
   
   /*
    * Licensed to the Apache Software Foundation (ASF) under one
    * or more contributor license agreements.  See the NOTICE file
    * distributed with this work for additional information
    * regarding copyright ownership.  The ASF licenses this file
    * to you 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.    
   */
  
  import java.io.BufferedWriter;
  import java.io.FileOutputStream;
  import java.io.OutputStreamWriter;
  import java.io.Writer;
  
  import junit.framework.Test;
  import junit.framework.TestSuite;
  
  import org.apache.velocity.Template;
  import org.apache.velocity.VelocityContext;
  import org.apache.velocity.app.Velocity;
  import org.apache.velocity.runtime.log.NullLogChute;
  
  /**
   * Tests input encoding handling.  The input target is UTF-8, having
   * chinese and and a spanish enyay (n-twiddle)
   *
   *  Thanks to Kent Johnson for the example input file.
   *
   *
   * @author <a href="mailto:geirm@optonline.net">Geir Magnusson Jr.</a>
   * @version $Id: EncodingTestCase.java 477002 2006-11-20 01:07:43Z henning $
   */
  public class EncodingTestCase extends BaseTestCase implements TemplateTestBase
  {
      public EncodingTestCase(String name)
      {
          super(name);
      }
  
      public void setUp()
              throws Exception
      {
          Velocity.setProperty(
                  Velocity.FILE_RESOURCE_LOADER_PATH, FILE_RESOURCE_LOADER_PATH);
  
          Velocity.setProperty( Velocity.INPUT_ENCODING, "UTF-8" );
  
          Velocity.setProperty(
                  Velocity.RUNTIME_LOG_LOGSYSTEM_CLASS, NullLogChute.class.getName());
  
          Velocity.init();
      }
  
      public static Test suite()
      {
          return new TestSuite(EncodingTestCase.class);
      }
  
      /**
       * Runs the test.
       */
      public void testChineseEncoding ()
              throws Exception
      {
          VelocityContext context = new VelocityContext();
  
          assureResultsDirectoryExists(RESULT_DIR);
  
          /*
           *  get the template and the output
           */
  
          /*
           *  Chinese and spanish
           */
  
          Template template = Velocity.getTemplate(
              getFileName(null, "encodingtest", TMPL_FILE_EXT), "UTF-8");
  
          FileOutputStream fos =
              new FileOutputStream (
                  getFileName(RESULT_DIR, "encodingtest", RESULT_FILE_EXT));
  
          Writer writer = new BufferedWriter(new OutputStreamWriter(fos, "UTF-8"));
  
          template.merge(context, writer);
          writer.flush();
         writer.close();
 
         if (!isMatch(RESULT_DIR,COMPARE_DIR,"encodingtest",
                 RESULT_FILE_EXT,CMP_FILE_EXT) )
         {
             fail("Output 1 incorrect.");
         }
     }
 
     public void testHighByteChinese()
             throws Exception
     {
         VelocityContext context = new VelocityContext();
 
         assureResultsDirectoryExists(RESULT_DIR);
 
         /*
          *  a 'high-byte' chinese example from Michael Zhou
          */
 
         Template template = Velocity.getTemplate(
               getFileName( null, "encodingtest2", TMPL_FILE_EXT), "UTF-8");
 
         FileOutputStream fos =
             new FileOutputStream (
                 getFileName(RESULT_DIR, "encodingtest2", RESULT_FILE_EXT));
 
         Writer writer = new BufferedWriter(new OutputStreamWriter(fos, "UTF-8"));
 
         template.merge(context, writer);
         writer.flush();
         writer.close();
 
         if (!isMatch(RESULT_DIR,COMPARE_DIR,"encodingtest2",
                 RESULT_FILE_EXT,CMP_FILE_EXT) )
         {
             fail("Output 2 incorrect.");
         }
 
     }
 
     public void testHighByteChinese2()
             throws Exception
     {
         VelocityContext context = new VelocityContext();
 
         assureResultsDirectoryExists(RESULT_DIR);
 
         /*
          *  a 'high-byte' chinese from Ilkka
          */
 
         Template template = Velocity.getTemplate(
               getFileName( null, "encodingtest3", TMPL_FILE_EXT), "GBK");
 
         FileOutputStream fos =
             new FileOutputStream (
                 getFileName(RESULT_DIR, "encodingtest3", RESULT_FILE_EXT));
 
         Writer writer = new BufferedWriter(new OutputStreamWriter(fos, "GBK"));
 
         template.merge(context, writer);
         writer.flush();
         writer.close();
 
         if (!isMatch(RESULT_DIR,COMPARE_DIR,"encodingtest3",
                 RESULT_FILE_EXT,CMP_FILE_EXT) )
         {
             fail("Output 3 incorrect.");
         }
     }
 
     public void testRussian()
             throws Exception
     {
         VelocityContext context = new VelocityContext();
 
         assureResultsDirectoryExists(RESULT_DIR);
 
         /*
          *  Russian example from Vitaly Repetenko
          */
 
         Template template = Velocity.getTemplate(
               getFileName( null, "encodingtest_KOI8-R", TMPL_FILE_EXT), "KOI8-R");
 
         FileOutputStream fos =
             new FileOutputStream (
                 getFileName(RESULT_DIR, "encodingtest_KOI8-R", RESULT_FILE_EXT));
 
         Writer writer = new BufferedWriter(new OutputStreamWriter(fos, "KOI8-R"));
 
         template.merge(context, writer);
         writer.flush();
         writer.close();
 
         if (!isMatch(RESULT_DIR,COMPARE_DIR,"encodingtest_KOI8-R",
                 RESULT_FILE_EXT,CMP_FILE_EXT) )
         {
             fail("Output 4 incorrect.");
         }
     }
 }