package org.apache.maven.scm.tck.command.checkin;

 /*

  * 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 org.apache.maven.scm.ScmFile;

 import org.apache.maven.scm.ScmFileSet;

 import org.apache.maven.scm.ScmFileStatus;

 import org.apache.maven.scm.ScmTckTestCase;

 import org.apache.maven.scm.command.add.AddScmResult;

 import org.apache.maven.scm.command.checkin.CheckInScmResult;

 import org.apache.maven.scm.command.checkout.CheckOutScmResult;

 import org.codehaus.plexus.util.FileUtils;

 import org.codehaus.plexus.util.IOUtil;

 import java.io.File;

 import java.io.FileWriter;

 import java.io.PrintWriter;

 import java.util.List;

 import java.util.Map;

 /**

  * This test tests the check out command.

  *

  * @author <a href="mailto:brett@apache.org">Brett Porter</a>

  * @version $Id: CheckInCommandTckTest.java 1306856 2012-03-29 13:40:58Z olamy $

  */

 public abstract class CheckInCommandTckTest

     extends ScmTckTestCase

 {

     public void testCheckInCommandTest()

         throws Exception

     {

         // Make sure that the correct files was checked out

         File fooJava = new File( getWorkingCopy(), "src/main/java/Foo.java" );

         File barJava = new File( getWorkingCopy(), "src/main/java/Bar.java" );

         File readmeTxt = new File( getWorkingCopy(), "readme.txt" );

         assertFalse( "check Foo.java doesn't yet exist", fooJava.canRead() );

         assertFalse( "check Bar.java doesn't yet exist", barJava.canRead() );

         assertTrue( "check can read readme.txt", readmeTxt.canRead() );

         // Change the files

         createFooJava( fooJava );

         createBarJava( barJava );

         changeReadmeTxt( readmeTxt );

         AddScmResult addResult = getScmManager().add( getScmRepository(),

                                                       new ScmFileSet( getWorkingCopy(), "src/main/java/Foo.java",

                                                                       null ) );

         assertResultIsSuccess( addResult );

         CheckInScmResult result =

             getScmManager().checkIn( getScmRepository(), new ScmFileSet( getWorkingCopy() ), "Commit message" );

         assertResultIsSuccess( result );

         List<ScmFile> files = result.getCheckedInFiles();

         assertNotNull( files );

         assertEquals( 2, files.size() );

         Map<String, ScmFile> fileMap = mapFilesByPath( files );

         ScmFile file1 = fileMap.get( "src/main/java/Foo.java" );

         assertNotNull( file1 );

         assertEquals( ScmFileStatus.CHECKED_IN, file1.getStatus() );

         ScmFile file2 = fileMap.get( "readme.txt" );

         assertNotNull( file2 );

         assertEquals( ScmFileStatus.CHECKED_IN, file2.getStatus() );

         CheckOutScmResult checkoutResult =

             getScmManager().checkOut( getScmRepository(), new ScmFileSet( getAssertionCopy() ) );

         assertResultIsSuccess( checkoutResult );

         fooJava = new File( getAssertionCopy(), "src/main/java/Foo.java" );

         barJava = new File( getAssertionCopy(), "src/main/java/Bar.java" );

         readmeTxt = new File( getAssertionCopy(), "readme.txt" );

         assertTrue( "check can read Foo.java", fooJava.canRead() );

         assertFalse( "check Bar.java doesn't exist", barJava.canRead() );

         assertTrue( "check can read readme.txt", readmeTxt.canRead() );

         assertEquals( "check readme.txt contents", "changed file", FileUtils.fileRead( readmeTxt ) );

     }

     public void testCheckInCommandPartialFileset()

         throws Exception

     {

         // Make sure that the correct files was checked out

         File fooJava = new File( getWorkingCopy(), "src/main/java/Foo.java" );

         File barJava = new File( getWorkingCopy(), "src/main/java/Bar.java" );

         File readmeTxt = new File( getWorkingCopy(), "readme.txt" );

         assertFalse( "check Foo.java doesn't yet exist", fooJava.canRead() );

         assertFalse( "check Bar.java doesn't yet exist", barJava.canRead() );

         assertTrue( "check can read readme.txt", readmeTxt.canRead() );

         // Change the files

         createFooJava( fooJava );

         createBarJava( barJava );

         changeReadmeTxt( readmeTxt );

         AddScmResult addResult = getScmManager().getProviderByUrl( getScmUrl() ).add( getScmRepository(),

                                                                                       new ScmFileSet( getWorkingCopy(),

                                                                                                       "src/main/java/Foo.java",

                                                                                                       null ) );

         assertResultIsSuccess( addResult );

         CheckInScmResult result =

             getScmManager().checkIn( getScmRepository(), new ScmFileSet( getWorkingCopy(), "**/Foo.java", null ),

                                      "Commit message" );

         assertResultIsSuccess( result );

         List<ScmFile> files = result.getCheckedInFiles();

         assertNotNull( files );

         assertEquals( 1, files.size() );

         ScmFile file1 = files.get( 0 );

         assertEquals( ScmFileStatus.CHECKED_IN, file1.getStatus() );

         assertPath( "/test-repo/check-in/Foo.java", file1.getPath() );

         CheckOutScmResult checkoutResult =

             getScmManager().checkOut( getScmRepository(), new ScmFileSet( getAssertionCopy() ) );

         assertResultIsSuccess( checkoutResult );

         fooJava = new File( getAssertionCopy(), "src/main/java/Foo.java" );

         barJava = new File( getAssertionCopy(), "src/main/java/Bar.java" );

         readmeTxt = new File( getAssertionCopy(), "readme.txt" );

         assertTrue( "check can read Foo.java", fooJava.canRead() );

         assertFalse( "check Bar.java doesn't exist", barJava.canRead() );

         assertTrue( "check can read readme.txt", readmeTxt.canRead() );

         assertEquals( "check readme.txt contents", "/readme.txt", FileUtils.fileRead( readmeTxt ) );

     }

     private void createFooJava( File fooJava )

         throws Exception

     {

         FileWriter output = new FileWriter( fooJava );

         PrintWriter printer = new PrintWriter( output );

         try

         {

             printer.println( "public class Foo" );

             printer.println( "{" );

             printer.println( "    public void foo()" );

             printer.println( "    {" );

             printer.println( "        int i = 10;" );

             printer.println( "    }" );

             printer.println( "}" );

         }

         finally

         {

             IOUtil.close( output );

             IOUtil.close( printer );

         }

     }

     private void createBarJava( File barJava )

         throws Exception

     {

         FileWriter output = new FileWriter( barJava );

         PrintWriter printer = new PrintWriter( output );

         printer.println( "public class Bar" );

         printer.println( "{" );

         printer.println( "    public int bar()" );

         printer.println( "    {" );

         printer.println( "        return 20;" );

         printer.println( "    }" );

         printer.println( "}" );

         printer.close();

         output.close();

     }

     private void changeReadmeTxt( File readmeTxt )

         throws Exception

     {

         FileWriter output = new FileWriter( readmeTxt );

         output.write( "changed file" );

         output.close();

     }

 }