wagon-provider-api/src/test/java/org/apache/maven/wagon/MockWagon.java

package org.apache.maven.wagon;

import org.apache.maven.wagon.artifact.Artifact;

import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.InputStream;
import java.io.OutputStream;

/**
 * @author <a href="mailto:jason@maven.org">Jason van Zyl</a>
 * @version $Id$
 */
public class MockWagon
    extends StreamWagon
{
    private boolean errorInputStream;

    public MockWagon()
    {
    }

    public MockWagon( boolean errorInputStream )
    {
        this.errorInputStream = errorInputStream;
    }

    public String[] getSupportedProtocols()
    {
        return new String[]{"mock"};
    }

    public InputStream getInputStream( Artifact artifact )
        throws Exception
    {
        if ( errorInputStream )
        {
            MockInputStream is = new MockInputStream();

            is.setForcedError( true );

            return is;
        }

        byte[] buffer = new byte[1024 * 4 * 5];

        ByteArrayInputStream is = new ByteArrayInputStream( buffer );

        return is;
    }

    public OutputStream getOutputStream( Artifact artifact )
        throws Exception
    {
        if ( errorInputStream )
        {
            MockOutputStream os = new MockOutputStream();

            os.setForcedError( true );

            return os;
        }

        ByteArrayOutputStream os = new ByteArrayOutputStream();

        return os;
    }


    public void openConnection()
    {
    }

    public void closeConnection()
    {
    }
}