package org.apache.maven.wagon;
/*
* 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.wagon.observers.ChecksumObserver;
import org.apache.maven.wagon.resource.Resource;
import org.codehaus.plexus.util.FileUtils;
import org.codehaus.plexus.util.IOUtil;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.InputStream;
import java.io.OutputStream;
import java.text.SimpleDateFormat;
/**
* @author <a href="mailto:brett@apache.org">Brett Porter</a>
* @version $Id: WagonTestCase.java 660462 2008-05-27 10:52:46Z brett $
*/
public abstract class StreamingWagonTestCase
extends WagonTestCase
{
public void testStreamingWagon()
throws Exception
{
if ( supportsGetIfNewer() )
{
setupRepositories();
setupWagonTestingFixtures();
streamRoundTripTesting();
tearDownWagonTestingFixtures();
}
}
public void testFailedGetToStream()
throws Exception
{
setupRepositories();
setupWagonTestingFixtures();
message( "Getting test artifact from test repository " + testRepository );
StreamingWagon wagon = (StreamingWagon) getWagon();
wagon.addTransferListener( checksumObserver );
wagon.connect( testRepository, getAuthInfo() );
destFile = FileTestUtils.createUniqueFile( getName(), getName() );
destFile.deleteOnExit();
OutputStream stream = null;
try
{
stream = new FileOutputStream( destFile );
wagon.getToStream( "fubar.txt", stream );
fail( "File was found when it shouldn't have been" );
}
catch ( ResourceDoesNotExistException e )
{
// expected
assertTrue( true );
}
finally
{
wagon.removeTransferListener( checksumObserver );
wagon.disconnect();
IOUtil.close( stream );
tearDownWagonTestingFixtures();
}
}
public void testWagonGetIfNewerToStreamIsNewer()
throws Exception
{
if ( supportsGetIfNewer() )
{
setupRepositories();
setupWagonTestingFixtures();
int expectedSize = putFile();
getIfNewerToStream( getExpectedLastModifiedOnGet( testRepository, new Resource( resource ) ) + 30000, false,
expectedSize );
}
}
public void testWagonGetIfNewerToStreamIsOlder()
throws Exception
{
if ( supportsGetIfNewer() )
{
setupRepositories();
setupWagonTestingFixtures();
int expectedSize = putFile();
getIfNewerToStream( new SimpleDateFormat( "yyyy-MM-dd" ).parse( "2006-01-01" ).getTime(), true,
expectedSize );
}
}
public void testWagonGetIfNewerToStreamIsSame()
throws Exception
{
if ( supportsGetIfNewer() )
{
setupRepositories();
setupWagonTestingFixtures();
int expectedSize = putFile();
getIfNewerToStream( getExpectedLastModifiedOnGet( testRepository, new Resource( resource ) ), false,
expectedSize );
}
}
private void getIfNewerToStream( long timestamp, boolean expectedResult, int expectedSize )
throws Exception
{
StreamingWagon wagon = (StreamingWagon) getWagon();
ProgressArgumentMatcher progressArgumentMatcher = setupGetIfNewerTest( wagon, expectedResult, expectedSize );
connectWagon( wagon );
OutputStream stream = new LazyFileOutputStream( destFile );
try
{
boolean result = wagon.getIfNewerToStream( this.resource, stream, timestamp );
assertEquals( expectedResult, result );
}
finally
{
IOUtil.close( stream );
}
disconnectWagon( wagon );
assertGetIfNewerTest( progressArgumentMatcher, expectedResult, expectedSize );
tearDownWagonTestingFixtures();
}
public void testFailedGetIfNewerToStream()
throws Exception
{
if ( supportsGetIfNewer() )
{
setupRepositories();
setupWagonTestingFixtures();
message( "Getting test artifact from test repository " + testRepository );
StreamingWagon wagon = (StreamingWagon) getWagon();
wagon.addTransferListener( checksumObserver );
wagon.connect( testRepository, getAuthInfo() );
destFile = FileTestUtils.createUniqueFile( getName(), getName() );
destFile.deleteOnExit();
OutputStream stream = null;
try
{
stream = new FileOutputStream( destFile );
wagon.getIfNewerToStream( "fubar.txt", stream, 0 );
fail( "File was found when it shouldn't have been" );
}
catch ( ResourceDoesNotExistException e )
{
// expected
assertTrue( true );
}
finally
{
wagon.removeTransferListener( checksumObserver );
wagon.disconnect();
IOUtil.close( stream );
tearDownWagonTestingFixtures();
}
}
}
protected void streamRoundTripTesting()
throws Exception
{
message( "Stream round trip testing ..." );
int expectedSize = putStream();
assertNotNull( "check checksum is not null", checksumObserver.getActualChecksum() );
assertEquals( "compare checksums", "6b144b7285ffd6b0bc8300da162120b9", checksumObserver.getActualChecksum() );
checksumObserver = new ChecksumObserver();
getStream( expectedSize );
assertNotNull( "check checksum is not null", checksumObserver.getActualChecksum() );
assertEquals( "compare checksums", "6b144b7285ffd6b0bc8300da162120b9", checksumObserver.getActualChecksum() );
// Now compare the conents of the artifact that was placed in
// the repository with the contents of the artifact that was
// retrieved from the repository.
String sourceContent = FileUtils.fileRead( sourceFile );
String destContent = FileUtils.fileRead( destFile );
assertEquals( sourceContent, destContent );
}
private int putStream()
throws Exception
{
String content = "test-resource.txt\n";
sourceFile = new File( FileTestUtils.getTestOutputDir(), "test-resource" );
sourceFile.getParentFile().mkdirs();
FileUtils.fileWrite( sourceFile.getAbsolutePath(), content );
StreamingWagon wagon = (StreamingWagon) getWagon();
ProgressArgumentMatcher progressArgumentMatcher = replayMockForPut( resource, content, wagon );
message( "Putting test artifact: " + resource + " into test repository " + testRepository );
connectWagon( wagon );
InputStream stream = null;
try
{
stream = new FileInputStream( sourceFile );
wagon.putFromStream( stream, resource, sourceFile.length(), sourceFile.lastModified() );
}
catch ( Exception e )
{
logger.error( "error while putting resources to the FTP Server", e );
}
finally
{
IOUtil.close( stream );
}
disconnectWagon( wagon );
verifyMock( progressArgumentMatcher, content.length() );
return content.length();
}
private void getStream( int expectedSize )
throws Exception
{
destFile = FileTestUtils.createUniqueFile( getName(), getName() );
destFile.deleteOnExit();
StreamingWagon wagon = (StreamingWagon) getWagon();
ProgressArgumentMatcher progressArgumentMatcher = replaceMockForGet( wagon, expectedSize );
message( "Getting test artifact from test repository " + testRepository );
connectWagon( wagon );
OutputStream stream = null;
try
{
stream = new FileOutputStream( destFile );
wagon.getToStream( this.resource, stream );
}
catch ( Exception e )
{
logger.error( "error while reading resources from the FTP Server", e );
}
finally
{
IOUtil.close( stream );
}
disconnectWagon( wagon );
verifyMock( progressArgumentMatcher, expectedSize );
}
}