package org.apache.maven.cli;
   
   /*
    * 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 static org.junit.Assert.assertEquals;
  import static org.junit.Assert.assertFalse;
  import static org.junit.Assert.assertTrue;
  import static org.junit.Assert.fail;
  import static org.junit.Assume.assumeTrue;
  
  import java.io.File;
  
  import org.apache.commons.cli.ParseException;
  import org.apache.maven.shared.utils.logging.MessageUtils;
  import org.junit.After;
  import org.junit.Before;
  import org.junit.Test;
  
  public class MavenCliTest
  {
      private MavenCli cli;
  
      private String origBasedir;
  
      @Before
      public void setUp()
      {
          cli = new MavenCli();
          origBasedir = System.getProperty( MavenCli.MULTIMODULE_PROJECT_DIRECTORY );
      }
  
      @After
      public void tearDown()
          throws Exception
      {
          if ( origBasedir != null )
          {
              System.setProperty( MavenCli.MULTIMODULE_PROJECT_DIRECTORY, origBasedir );
          }
          else
          {
              System.getProperties().remove( MavenCli.MULTIMODULE_PROJECT_DIRECTORY );
          }
      }
  
      @Test
      public void testCalculateDegreeOfConcurrencyWithCoreMultiplier()
      {
          int cores = Runtime.getRuntime().availableProcessors();
          // -T2.2C
          assertEquals( (int) ( cores * 2.2 ), cli.calculateDegreeOfConcurrencyWithCoreMultiplier( "C2.2" ) );
          // -TC2.2
          assertEquals( (int) ( cores * 2.2 ), cli.calculateDegreeOfConcurrencyWithCoreMultiplier( "2.2C" ) );
  
          try
          {
              cli.calculateDegreeOfConcurrencyWithCoreMultiplier( "CXXX" );
              fail( "Should have failed with a NumberFormatException" );
          }
          catch ( NumberFormatException e )
          {
              // carry on
          }
      }
  
      @Test
      public void testMavenConfig()
          throws Exception
      {
          System.setProperty( MavenCli.MULTIMODULE_PROJECT_DIRECTORY,
                              new File( "src/test/projects/config" ).getCanonicalPath() );
          CliRequest request = new CliRequest( new String[0], null );
  
          // read .mvn/maven.config
          cli.initialize( request );
          cli.cli( request );
          assertEquals( "multithreaded", request.commandLine.getOptionValue( CLIManager.BUILDER ) );
          assertEquals( "8", request.commandLine.getOptionValue( CLIManager.THREADS ) );
  
          // override from command line
          request = new CliRequest( new String[]{ "--builder", "foobar" }, null );
          cli.cli( request );
         assertEquals( "foobar", request.commandLine.getOptionValue( "builder" ) );
     }
 
     @Test
     public void testMavenConfigInvalid()
         throws Exception
     {
         System.setProperty( MavenCli.MULTIMODULE_PROJECT_DIRECTORY,
                             new File( "src/test/projects/config-illegal" ).getCanonicalPath() );
         CliRequest request = new CliRequest( new String[0], null );
 
         cli.initialize( request );
         try
         {
             cli.cli( request );
             fail();
         }
         catch ( ParseException expected )
         {
 
         }
     }
 
     /**
      * Read .mvn/maven.config with the following definitions:
      * <pre>
      *   -T 3
      *   -Drevision=1.3.0
      * </pre>
      * and check if the {@code -T 3} option can be overwritten via command line
      * argument.
      *
      * @throws Exception in case of failure.
      */
     @Test
     public void testMVNConfigurationThreadCanBeOverwrittenViaCommandLine()
         throws Exception
     {
         System.setProperty( MavenCli.MULTIMODULE_PROJECT_DIRECTORY,
                             new File( "src/test/projects/mavenConfigProperties" ).getCanonicalPath() );
         CliRequest request = new CliRequest( new String[]{ "-T", "5" }, null );
 
         cli.initialize( request );
         // read .mvn/maven.config
         cli.cli( request );
 
         assertEquals( "5", request.commandLine.getOptionValue( CLIManager.THREADS ) );
     }
 
     /**
      * Read .mvn/maven.config with the following definitions:
      * <pre>
      *   -T 3
      *   -Drevision=1.3.0
      * </pre>
      * and check if the {@code -Drevision-1.3.0} option can be overwritten via command line
      * argument.
      *
      * @throws Exception
      */
     @Test
     public void testMVNConfigurationDefinedPropertiesCanBeOverwrittenViaCommandLine()
         throws Exception
     {
         System.setProperty( MavenCli.MULTIMODULE_PROJECT_DIRECTORY,
                             new File( "src/test/projects/mavenConfigProperties" ).getCanonicalPath() );
         CliRequest request = new CliRequest( new String[]{ "-Drevision=8.1.0" }, null );
 
         cli.initialize( request );
         // read .mvn/maven.config
         cli.cli( request );
         cli.properties( request );
 
         String revision = System.getProperty( "revision" );
         assertEquals( "8.1.0", revision );
     }
 
     /**
      * Read .mvn/maven.config with the following definitions:
      * <pre>
      *   -T 3
      *   -Drevision=1.3.0
      * </pre>
      * and check if the {@code -Drevision-1.3.0} option can be overwritten via command line
      * argument.
      *
      * @throws Exception
      */
     @Test
     public void testMVNConfigurationCLIRepeatedPropertiesLastWins()
         throws Exception
     {
         System.setProperty( MavenCli.MULTIMODULE_PROJECT_DIRECTORY,
                             new File( "src/test/projects/mavenConfigProperties" ).getCanonicalPath() );
         CliRequest request = new CliRequest( new String[]{ "-Drevision=8.1.0", "-Drevision=8.2.0" }, null );
 
         cli.initialize( request );
         // read .mvn/maven.config
         cli.cli( request );
         cli.properties( request );
 
         String revision = System.getProperty( "revision" );
         assertEquals( "8.2.0", revision );
     }
 
     /**
      * Read .mvn/maven.config with the following definitions:
      * <pre>
      *   -T 3
      *   -Drevision=1.3.0
      * </pre>
      * and check if the {@code -Drevision-1.3.0} option can be overwritten via command line argument when there are
      * funky arguments present.
      *
      * @throws Exception
      */
     @Test
     public void testMVNConfigurationFunkyArguments()
         throws Exception
     {
         System.setProperty( MavenCli.MULTIMODULE_PROJECT_DIRECTORY,
                             new File( "src/test/projects/mavenConfigProperties" ).getCanonicalPath() );
         CliRequest request = new CliRequest(
             new String[]{ "-Drevision=8.1.0", "--file=-Dpom.xml", "\"-Dfoo=bar ", "\"-Dfoo2=bar two\"",
                 "-Drevision=8.2.0" }, null );
 
         cli.initialize( request );
         // read .mvn/maven.config
         cli.cli( request );
         cli.properties( request );
 
         String revision = System.getProperty( "revision" );
         assertEquals( "8.2.0", revision );
 
         assertEquals( "bar ", request.getSystemProperties().getProperty( "foo" ) );
         assertEquals( "bar two", request.getSystemProperties().getProperty( "foo2" ) );
 
         assertEquals( "-Dpom.xml", request.getCommandLine().getOptionValue( CLIManager.ALTERNATE_POM_FILE ) );
     }
 
     @Test
     public void testStyleColors()
         throws Exception
     {
         assumeTrue( "ANSI not supported", MessageUtils.isColorEnabled() );
         CliRequest request;
 
         MessageUtils.setColorEnabled( true );
         request = new CliRequest( new String[] { "-B" }, null );
         cli.cli( request );
         cli.properties( request );
         cli.logging( request );
         assertFalse( MessageUtils.isColorEnabled() );
 
         MessageUtils.setColorEnabled( true );
         request = new CliRequest( new String[] { "-l", "target/temp/mvn.log" }, null );
         cli.cli( request );
         cli.properties( request );
         cli.logging( request );
         assertFalse( MessageUtils.isColorEnabled() );
 
         MessageUtils.setColorEnabled( false );
         request = new CliRequest( new String[] { "-Dstyle.color=always" }, null );
         cli.cli( request );
         cli.properties( request );
         cli.logging( request );
         assertTrue( MessageUtils.isColorEnabled() );
 
         MessageUtils.setColorEnabled( true );
         request = new CliRequest( new String[] { "-Dstyle.color=never" }, null );
         cli.cli( request );
         cli.properties( request );
         cli.logging( request );
         assertFalse( MessageUtils.isColorEnabled() );
 
         MessageUtils.setColorEnabled( false );
         request = new CliRequest( new String[] { "-Dstyle.color=always", "-B", "-l", "target/temp/mvn.log" }, null );
         cli.cli( request );
         cli.properties( request );
         cli.logging( request );
         assertTrue( MessageUtils.isColorEnabled() );
 
         try
         {
             MessageUtils.setColorEnabled( false );
             request = new CliRequest( new String[] { "-Dstyle.color=maybe", "-B", "-l", "target/temp/mvn.log" }, null );
             cli.cli( request );
             cli.properties( request );
             cli.logging( request );
             fail( "maybe is not a valid option" );
         }
         catch ( IllegalArgumentException e )
         {
             // noop
         }
     }
 }