* 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.

*/

* Test the proxy handler.

*

* @author Brett Porter

*/

    extends PlexusTestCase

    private ProxyRequestHandler requestHandler;

    private List proxiedRepositories;

    private List legacyProxiedRepositories;

    private ArtifactRepository defaultManagedRepository;

    private ArtifactRepository legacyManagedRepository;

    private ArtifactRepository proxiedRepository1;

    private ArtifactRepository proxiedRepository2;

    private ArtifactRepository legacyProxiedRepository;

    private ArtifactRepositoryLayout defaultLayout;

    private ArtifactRepositoryFactory factory;

    private MockControl wagonMockControl;

    private Wagon wagonMock;

    private static final ArtifactRepositoryPolicy DEFAULT_POLICY =

        new ArtifactRepositoryPolicy( true, ArtifactRepositoryPolicy.UPDATE_POLICY_NEVER, null );

    private static final ArtifactRepositoryPolicy ALWAYS_UPDATE_POLICY =

        new ArtifactRepositoryPolicy( true, ArtifactRepositoryPolicy.UPDATE_POLICY_ALWAYS, null );

    private static final TimeZone UTC_TIMEZONE = TimeZone.getTimeZone( "UTC" );

    protected void setUp()

        throws Exception

    {

        super.setUp();

        requestHandler = (ProxyRequestHandler) lookup( ProxyRequestHandler.ROLE );

        factory = (ArtifactRepositoryFactory) lookup( ArtifactRepositoryFactory.ROLE );

        factory.setGlobalChecksumPolicy( ArtifactRepositoryPolicy.CHECKSUM_POLICY_FAIL );

       

        File repoLocation = getTestFile( "target/test-repository/managed" );

        // faster only to delete this one before copying, the others are done case by case

        FileUtils.deleteDirectory( new File( repoLocation, "org/apache/maven/test/get-merged-metadata" ) );

        copyDirectoryStructure( getTestFile( "src/test/repositories/managed" ), repoLocation );

        defaultLayout = (ArtifactRepositoryLayout) lookup( ArtifactRepositoryLayout.ROLE, "default" );

        defaultManagedRepository = createRepository( "managed-repository", repoLocation );

        repoLocation = getTestFile( "target/test-repository/legacy-managed" );

        FileUtils.deleteDirectory( repoLocation );

        copyDirectoryStructure( getTestFile( "src/test/repositories/legacy-managed" ), repoLocation );

        ArtifactRepositoryLayout legacyLayout =

            (ArtifactRepositoryLayout) lookup( ArtifactRepositoryLayout.ROLE, "legacy" );

        legacyManagedRepository = createRepository( "managed-repository", repoLocation, legacyLayout );

        File location = getTestFile( "src/test/repositories/proxied1" );

        proxiedRepository1 = createRepository( "proxied1", location );

        location = getTestFile( "src/test/repositories/proxied2" );

        proxiedRepository2 = createRepository( "proxied2", location );

        proxiedRepositories = new ArrayList( 2 );

        proxiedRepositories.add( createProxiedRepository( proxiedRepository1 ) );

        proxiedRepositories.add( createProxiedRepository( proxiedRepository2 ) );

        location = getTestFile( "src/test/repositories/legacy-proxied" );

        legacyProxiedRepository = createRepository( "legacy-proxied", location, legacyLayout );

        legacyProxiedRepositories = Collections.singletonList( createProxiedRepository( legacyProxiedRepository ) );

        wagonMockControl = MockControl.createNiceControl( Wagon.class );

        wagonMock = (Wagon) wagonMockControl.getMock();

        WagonDelegate delegate = (WagonDelegate) lookup( Wagon.ROLE, "test" );

        delegate.setDelegate( wagonMock );

    }

    public void testGetDefaultLayoutNotPresent()

        throws ResourceDoesNotExistException, ProxyException, IOException

    {

        String path = "org/apache/maven/test/get-default-layout/1.0/get-default-layout-1.0.jar";

        File expectedFile = new File( defaultManagedRepository.getBasedir(), path );

        expectedFile.delete();

        assertFalse( expectedFile.exists() );

        ProxiedArtifact proxied = requestHandler.get( path, proxiedRepositories, defaultManagedRepository );

        assertEquals( "Check file matches", expectedFile, proxied.getFile() );

        assertTrue( "Check file created", proxied.getFile().exists() );

        File proxiedFile = new File( proxiedRepository1.getBasedir(), path );

        String expectedContents = FileUtils.readFileToString( proxiedFile, null );

        assertEquals( "Check file contents", expectedContents, FileUtils.readFileToString( proxied.getFile(), null ) );

        // TODO: timestamp preservation requires support for that in wagon

//        assertEquals( "Check file timestamp", proxiedFile.lastModified(), proxied.getFile().lastModified() );

    }

    public void testGetDefaultLayoutAlreadyPresent()

        throws ResourceDoesNotExistException, ProxyException, IOException

    {

        String path = "org/apache/maven/test/get-default-layout-present/1.0/get-default-layout-present-1.0.jar";

        File expectedFile = new File( defaultManagedRepository.getBasedir(), path );

        String expectedContents = FileUtils.readFileToString( expectedFile, null );

        long originalModificationTime = expectedFile.lastModified();

        assertTrue( expectedFile.exists() );

        ProxiedArtifact proxied = requestHandler.get( path, proxiedRepositories, defaultManagedRepository );

        assertEquals( "Check file matches", expectedFile, proxied.getFile() );

        assertTrue( "Check file created", proxied.getFile().exists() );

        assertEquals( "Check file contents", expectedContents, FileUtils.readFileToString( proxied.getFile(), null ) );

        File proxiedFile = new File( proxiedRepository1.getBasedir(), path );

        String unexpectedContents = FileUtils.readFileToString( proxiedFile, null );

        assertFalse( "Check file contents", unexpectedContents.equals( FileUtils.readFileToString( proxied.getFile(), null ) ) );

        assertFalse( "Check file timestamp is not that of proxy", proxiedFile.lastModified() == proxied.getFile().lastModified() );

        assertEquals( "Check file timestamp is that of original managed file", originalModificationTime,

                      proxied.getFile().lastModified() );

    }

    public void testGetDefaultLayoutRemoteUpdate()

        throws ResourceDoesNotExistException, ProxyException, IOException, ParseException

    {

        String path = "org/apache/maven/test/get-default-layout-present/1.0/get-default-layout-present-1.0.jar";

        File expectedFile = new File( defaultManagedRepository.getBasedir(), path );

        String expectedContents = FileUtils.readFileToString( expectedFile, null );

        assertTrue( expectedFile.exists() );

        expectedFile.setLastModified( getPastDate().getTime() );

        ProxiedArtifact proxied = requestHandler.get( path, proxiedRepositories, defaultManagedRepository );

        assertEquals( "Check file matches", expectedFile, proxied.getFile() );

        assertTrue( "Check file created", proxied.getFile().exists() );

        assertEquals( "Check file contents", expectedContents, FileUtils.readFileToString( proxied.getFile(), null ) );

        File proxiedFile = new File( proxiedRepository1.getBasedir(), path );

        String unexpectedContents = FileUtils.readFileToString( proxiedFile, null );

        assertFalse( "Check file contents", unexpectedContents.equals( FileUtils.readFileToString( proxied.getFile(), null ) ) );

    }

    public void testGetWhenInBothProxiedRepos()

        throws ResourceDoesNotExistException, ProxyException, IOException

    {

        String path = "org/apache/maven/test/get-in-both-proxies/1.0/get-in-both-proxies-1.0.jar";

        File expectedFile = new File( defaultManagedRepository.getBasedir(), path );

        expectedFile.delete();

        assertFalse( expectedFile.exists() );

        ProxiedArtifact proxied = requestHandler.get( path, proxiedRepositories, defaultManagedRepository );

        assertEquals( "Check file matches", expectedFile, proxied.getFile() );

        assertTrue( "Check file created", proxied.getFile().exists() );

        File proxiedFile = new File( proxiedRepository1.getBasedir(), path );

        String expectedContents = FileUtils.readFileToString( proxiedFile, null );

        assertEquals( "Check file contents", expectedContents, FileUtils.readFileToString( proxied.getFile(), null ) );

        proxiedFile = new File( proxiedRepository2.getBasedir(), path );

        String unexpectedContents = FileUtils.readFileToString( proxiedFile, null );

        assertFalse( "Check file contents", unexpectedContents.equals( FileUtils.readFileToString( proxied.getFile(), null ) ) );

    }

    public void testGetInSecondProxiedRepo()

        throws ResourceDoesNotExistException, ProxyException, IOException

    {

        String path = "org/apache/maven/test/get-in-second-proxy/1.0/get-in-second-proxy-1.0.jar";

        File expectedFile = new File( defaultManagedRepository.getBasedir(), path );

        expectedFile.delete();

        assertFalse( expectedFile.exists() );

        ProxiedArtifact proxied = requestHandler.get( path, proxiedRepositories, defaultManagedRepository );

        assertEquals( "Check file matches", expectedFile, proxied.getFile() );

        assertTrue( "Check file created", proxied.getFile().exists() );

        File proxiedFile = new File( proxiedRepository2.getBasedir(), path );

        String expectedContents = FileUtils.readFileToString( proxiedFile, null );

        assertEquals( "Check file contents", expectedContents, FileUtils.readFileToString( proxied.getFile(), null ) );

    }

    public void testNotFoundInAnyProxies()

        throws ResourceDoesNotExistException, ProxyException, IOException

    {

        String path = "org/apache/maven/test/does-not-exist/1.0/does-not-exist-1.0.jar";

        File expectedFile = new File( defaultManagedRepository.getBasedir(), path );

        assertFalse( expectedFile.exists() );

        try

        {

            ProxiedArtifact proxied = requestHandler.get( path, proxiedRepositories, defaultManagedRepository );

            fail( "File returned was: " + proxied.getFile() + "; should have got a not found exception" );

        }

        catch ( ResourceDoesNotExistException e )

        {

            // expected, but check file was not created

            assertFalse( expectedFile.exists() );

        }

    }

    public void testGetInSecondProxiedRepoFirstFails()

        throws ResourceDoesNotExistException, ProxyException, IOException, TransferFailedException,

        AuthorizationException

    {

        String path = "org/apache/maven/test/get-in-second-proxy/1.0/get-in-second-proxy-1.0.jar";

        File expectedFile = new File( defaultManagedRepository.getBasedir(), path ).getAbsoluteFile();

        expectedFile.delete();

        assertFalse( expectedFile.exists() );

        proxiedRepository1 = createRepository( "proxied1", "test://..." );

        proxiedRepositories.clear();

        ProxiedArtifactRepository proxiedArtifactRepository = createProxiedRepository( proxiedRepository1 );

        proxiedRepositories.add( proxiedArtifactRepository );

        proxiedRepositories.add( createProxiedRepository( proxiedRepository2 ) );

        wagonMock.get( path, new File( expectedFile.getParentFile(), expectedFile.getName() + ".tmp" ) );

        wagonMockControl.setThrowable( new TransferFailedException( "transfer failed" ) );

        wagonMockControl.replay();

        ProxiedArtifact proxied = requestHandler.get( path, proxiedRepositories, defaultManagedRepository );

        wagonMockControl.verify();

        assertEquals( "Check file matches", expectedFile, proxied.getFile() );

        assertTrue( "Check file created", proxied.getFile().exists() );

        File proxiedFile = new File( proxiedRepository2.getBasedir(), path );

        String expectedContents = FileUtils.readFileToString( proxiedFile, null );

        assertEquals( "Check file contents", expectedContents, FileUtils.readFileToString( proxied.getFile(), null ) );

        assertTrue( "Check failure", proxiedArtifactRepository.isCachedFailure( path ) );

    }

    public void testGetButAllRepositoriesFail()

        throws ResourceDoesNotExistException, ProxyException, IOException, TransferFailedException,

        AuthorizationException

    {

        String path = "org/apache/maven/test/get-in-second-proxy/1.0/get-in-second-proxy-1.0.jar";

        File expectedFile = new File( defaultManagedRepository.getBasedir(), path ).getAbsoluteFile();

        expectedFile.delete();

        assertFalse( expectedFile.exists() );

        proxiedRepository1 = createRepository( "proxied1", "test://..." );

        proxiedRepository2 = createRepository( "proxied2", "test://..." );

        proxiedRepositories.clear();

        ProxiedArtifactRepository proxiedArtifactRepository1 = createProxiedRepository( proxiedRepository1 );

        proxiedRepositories.add( proxiedArtifactRepository1 );

        ProxiedArtifactRepository proxiedArtifactRepository2 = createProxiedRepository( proxiedRepository2 );

        proxiedRepositories.add( proxiedArtifactRepository2 );

        wagonMock.get( path, new File( expectedFile.getParentFile(), expectedFile.getName() + ".tmp" ) );

        wagonMockControl.setThrowable( new TransferFailedException( "transfer failed" ) );

        wagonMock.get( path, new File( expectedFile.getParentFile(), expectedFile.getName() + ".tmp" ) );

        wagonMockControl.setThrowable( new TransferFailedException( "transfer failed" ) );

        wagonMockControl.replay();

        try

        {

            ProxiedArtifact proxied = requestHandler.get( path, proxiedRepositories, defaultManagedRepository );

            fail( "Found file: " + proxied.getFile() + "; but was expecting a failure" );

        }

        catch ( ResourceDoesNotExistException e )

        {

            // as expected

            wagonMockControl.verify();

            assertTrue( "Check failure", proxiedArtifactRepository1.isCachedFailure( path ) );

            assertTrue( "Check failure", proxiedArtifactRepository2.isCachedFailure( path ) );

            // TODO: do not want failures to present as a not found!

            // TODO: How much information on each failure should we pass back to the user vs. logging in the proxy?

        }

    }

    public void testGetInSecondProxiedRepoFirstHardFails()

        throws ResourceDoesNotExistException, ProxyException, IOException, TransferFailedException,

        AuthorizationException

    {

        String path = "org/apache/maven/test/get-in-second-proxy/1.0/get-in-second-proxy-1.0.jar";

        File expectedFile = new File( defaultManagedRepository.getBasedir(), path ).getAbsoluteFile();

        expectedFile.delete();

        assertFalse( expectedFile.exists() );

        proxiedRepository1 = createRepository( "proxied1", "test://..." );

        proxiedRepositories.clear();

        ProxiedArtifactRepository proxiedArtifactRepository = createHardFailProxiedRepository( proxiedRepository1 );

        proxiedRepositories.add( proxiedArtifactRepository );

        proxiedRepositories.add( createProxiedRepository( proxiedRepository2 ) );

        wagonMock.get( path, new File( expectedFile.getParentFile(), expectedFile.getName() + ".tmp" ) );

        TransferFailedException failedException = new TransferFailedException( "transfer failed" );

        wagonMockControl.setThrowable( failedException );

        wagonMockControl.replay();

        try

        {

            ProxiedArtifact proxied = requestHandler.get( path, proxiedRepositories, defaultManagedRepository );

            fail( "Found file: " + proxied.getFile() + "; but was expecting a failure" );

        }

        catch ( ProxyException e )

        {

            // expect a failure

            wagonMockControl.verify();

            assertEquals( "Check cause", failedException, e.getCause() );

            assertTrue( "Check failure", proxiedArtifactRepository.isCachedFailure( path ) );

        }

    }

    public void testGetInSecondProxiedRepoFirstFailsFromCache()

        throws ResourceDoesNotExistException, ProxyException, IOException, TransferFailedException,

        AuthorizationException

    {

        // fail from the cache, even though it is in the first repo now

        String path = "org/apache/maven/test/get-in-both-proxies/1.0/get-in-both-proxies-1.0.jar";

        File expectedFile = new File( defaultManagedRepository.getBasedir(), path );

        expectedFile.delete();

        assertFalse( expectedFile.exists() );

        proxiedRepositories.clear();

        ProxiedArtifactRepository proxiedArtifactRepository = createProxiedRepository( proxiedRepository1 );

        proxiedArtifactRepository.addFailure( path, DEFAULT_POLICY );

        proxiedRepositories.add( proxiedArtifactRepository );

        proxiedRepositories.add( createProxiedRepository( proxiedRepository2 ) );

        ProxiedArtifact proxied = requestHandler.get( path, proxiedRepositories, defaultManagedRepository );

        assertEquals( "Check file matches", expectedFile, proxied.getFile() );

        assertTrue( "Check file created", proxied.getFile().exists() );

        File proxiedFile = new File( proxiedRepository2.getBasedir(), path );

        String expectedContents = FileUtils.readFileToString( proxiedFile, null );

        assertEquals( "Check file contents", expectedContents, FileUtils.readFileToString( proxied.getFile(), null ) );

        proxiedFile = new File( proxiedRepository1.getBasedir(), path );

        String unexpectedContents = FileUtils.readFileToString( proxiedFile, null );

        assertFalse( "Check file contents", unexpectedContents.equals( FileUtils.readFileToString( proxied.getFile(), null ) ) );

    }

    public void testGetInSecondProxiedRepoFirstHardFailsFromCache()

        throws ResourceDoesNotExistException, ProxyException, IOException, TransferFailedException,

        AuthorizationException

    {

        // fail from the cache, even though it is in the first repo now

        String path = "org/apache/maven/test/get-in-both-proxies/1.0/get-in-both-proxies-1.0.jar";

        File expectedFile = new File( defaultManagedRepository.getBasedir(), path );

        expectedFile.delete();

        assertFalse( expectedFile.exists() );

        proxiedRepositories.clear();

        ProxiedArtifactRepository proxiedArtifactRepository = createHardFailProxiedRepository( proxiedRepository1 );

        proxiedArtifactRepository.addFailure( path, DEFAULT_POLICY );

        proxiedRepositories.add( proxiedArtifactRepository );

        proxiedRepositories.add( createProxiedRepository( proxiedRepository2 ) );

        try

        {

            ProxiedArtifact proxied = requestHandler.get( path, proxiedRepositories, defaultManagedRepository );

            fail( "Found file: " + proxied.getFile() + "; but was expecting a failure" );

        }

        catch ( ProxyException e )

        {

            // expect a failure

            assertTrue( "Check failure", proxiedArtifactRepository.isCachedFailure( path ) );

        }

    }

    public void testGetInSecondProxiedRepoFirstFailsDisabledCacheFailure()

        throws ResourceDoesNotExistException, ProxyException, IOException, TransferFailedException,

        AuthorizationException

    {

        String path = "org/apache/maven/test/get-in-second-proxy/1.0/get-in-second-proxy-1.0.jar";

        File expectedFile = new File( defaultManagedRepository.getBasedir(), path ).getAbsoluteFile();

        assertFalse( expectedFile.exists() );

        proxiedRepository1 = createRepository( "proxied1", "test://..." );

        proxiedRepositories.clear();

        ProxiedArtifactRepository proxiedArtifactRepository = createProxiedRepository( proxiedRepository1 );

        proxiedArtifactRepository.addFailure( path, DEFAULT_POLICY );

        proxiedArtifactRepository.setCacheFailures( false );

        proxiedRepositories.add( proxiedArtifactRepository );

        proxiedRepositories.add( createProxiedRepository( proxiedRepository2 ) );

        wagonMock.get( path, new File( expectedFile.getParentFile(), expectedFile.getName() + ".tmp" ) );

        wagonMockControl.setThrowable( new TransferFailedException( "transfer failed" ) );

        wagonMockControl.replay();

        ProxiedArtifact proxied = requestHandler.get( path, proxiedRepositories, defaultManagedRepository );

        wagonMockControl.verify();

        assertEquals( "Check file matches", expectedFile, proxied.getFile() );

        assertTrue( "Check file created", proxied.getFile().exists() );

        File proxiedFile = new File( proxiedRepository2.getBasedir(), path );

        String expectedContents = FileUtils.readFileToString( proxiedFile, null );

        assertEquals( "Check file contents", expectedContents, FileUtils.readFileToString( proxied.getFile(), null ) );

        assertFalse( "Check failure", proxiedArtifactRepository.isCachedFailure( path ) );

    }

    public void testGetWhenInBothProxiedReposFirstHasExpiredCacheFailure()

        throws ResourceDoesNotExistException, ProxyException, IOException, ParseException

    {

        String path = "org/apache/maven/test/get-in-both-proxies/1.0/get-in-both-proxies-1.0.jar";

        File expectedFile = new File( defaultManagedRepository.getBasedir(), path );

        assertFalse( expectedFile.exists() );

        proxiedRepositories.clear();

        ProxiedArtifactRepository proxiedArtifactRepository = createProxiedRepository( proxiedRepository1 );

        proxiedArtifactRepository.addFailure( path, ALWAYS_UPDATE_POLICY );

        proxiedRepositories.add( proxiedArtifactRepository );

        proxiedRepositories.add( createProxiedRepository( proxiedRepository2 ) );

        ProxiedArtifact proxied = requestHandler.get( path, proxiedRepositories, defaultManagedRepository );

        assertEquals( "Check file matches", expectedFile, proxied.getFile() );

        assertTrue( "Check file created", proxied.getFile().exists() );

        File proxiedFile = new File( proxiedRepository1.getBasedir(), path );

        String expectedContents = FileUtils.readFileToString( proxiedFile, null );

        assertEquals( "Check file contents", expectedContents, FileUtils.readFileToString( proxied.getFile(), null ) );

        proxiedFile = new File( proxiedRepository2.getBasedir(), path );

        String unexpectedContents = FileUtils.readFileToString( proxiedFile, null );

        assertFalse( "Check file contents", unexpectedContents.equals( FileUtils.readFileToString( proxied.getFile(), null ) ) );

        assertFalse( "Check failure", proxiedArtifactRepository.isCachedFailure( path ) );

    }

    public void testGetAlwaysAlreadyPresent()

        throws ResourceDoesNotExistException, ProxyException, IOException

    {

        String path = "org/apache/maven/test/get-default-layout-present/1.0/get-default-layout-present-1.0.jar";

        File expectedFile = new File( defaultManagedRepository.getBasedir(), path );

        String unexpectedContents = FileUtils.readFileToString( expectedFile, null );

        assertTrue( expectedFile.exists() );

        ProxiedArtifact proxied = requestHandler.getAlways( path, proxiedRepositories, defaultManagedRepository );

        assertEquals( "Check file matches", expectedFile, proxied.getFile() );

        assertTrue( "Check file created", proxied.getFile().exists() );

        File proxiedFile = new File( proxiedRepository1.getBasedir(), path );

        String expectedContents = FileUtils.readFileToString( proxiedFile, null );

        assertEquals( "Check file contents", expectedContents, FileUtils.readFileToString( proxied.getFile(), null ) );

        assertFalse( "Check file contents", unexpectedContents.equals( FileUtils.readFileToString( proxied.getFile(), null ) ) );

    }

    public void testGetAlwaysAlreadyPresentRemovedFromProxies()

        throws ResourceDoesNotExistException, ProxyException, IOException

    {

        String path = "org/apache/maven/test/get-removed-from-proxies/1.0/get-removed-from-proxies-1.0.jar";

        File expectedFile = new File( defaultManagedRepository.getBasedir(), path );

        String expectedContents = FileUtils.readFileToString( expectedFile, null );

        assertTrue( expectedFile.exists() );

        ProxiedArtifact proxied = requestHandler.getAlways( path, proxiedRepositories, defaultManagedRepository );

        assertEquals( "Check file matches", expectedFile, proxied.getFile() );

        assertTrue( "Check file created", proxied.getFile().exists() );

        assertEquals( "Check file contents", expectedContents, FileUtils.readFileToString( proxied.getFile(), null ) );

    }

    public void testGetAlwaysWithCachedFailure()

        throws ResourceDoesNotExistException, ProxyException, IOException

    {

        String path = "org/apache/maven/test/get-default-layout-present/1.0/get-default-layout-present-1.0.jar";

        File expectedFile = new File( defaultManagedRepository.getBasedir(), path );

        String unexpectedContents = FileUtils.readFileToString( expectedFile, null );

        assertTrue( expectedFile.exists() );

        proxiedRepositories.clear();

        ProxiedArtifactRepository proxiedArtifactRepository = createProxiedRepository( proxiedRepository1 );

        proxiedArtifactRepository.addFailure( path, DEFAULT_POLICY );

        proxiedRepositories.add( proxiedArtifactRepository );

        proxiedRepositories.add( createProxiedRepository( proxiedRepository2 ) );

        ProxiedArtifact proxied = requestHandler.getAlways( path, proxiedRepositories, defaultManagedRepository );

        assertEquals( "Check file matches", expectedFile, proxied.getFile() );

        assertTrue( "Check file created", proxied.getFile().exists() );

        File proxiedFile = new File( proxiedRepository1.getBasedir(), path );

        String expectedContents = FileUtils.readFileToString( proxiedFile, null );

        assertEquals( "Check file contents", expectedContents, FileUtils.readFileToString( proxied.getFile(), null ) );

        assertFalse( "Check file contents", unexpectedContents.equals( FileUtils.readFileToString( proxied.getFile(), null ) ) );

    }

    public void testGetRemovesTemporaryFileOnSuccess()

        throws ResourceDoesNotExistException, ProxyException, IOException

    {

        String path = "org/apache/maven/test/get-default-layout/1.0/get-default-layout-1.0.jar";

        File expectedFile = new File( defaultManagedRepository.getBasedir(), path );

        expectedFile.delete();

        assertFalse( expectedFile.exists() );

        ProxiedArtifact proxied = requestHandler.get( path, proxiedRepositories, defaultManagedRepository );

        assertEquals( "Check file matches", expectedFile, proxied.getFile() );

        assertTrue( "Check file created", proxied.getFile().exists() );

        File tempFile = new File( proxied.getFile().getParentFile(), proxied.getFile().getName() + ".tmp" );

        assertFalse( "Check temporary file removed", tempFile.exists() );

    }

    public void testGetRemovesTemporaryFileOnError()

        throws ResourceDoesNotExistException, ProxyException, IOException, TransferFailedException,

        AuthorizationException

    {

        String path = "org/apache/maven/test/get-default-layout/1.0/get-default-layout-1.0.jar";

        File expectedFile = new File( defaultManagedRepository.getBasedir(), path );

        expectedFile.delete();

        assertFalse( expectedFile.exists() );

        proxiedRepository1 = createRepository( "proxied1", "test://..." );

        proxiedRepositories.clear();

        ProxiedArtifactRepository proxiedArtifactRepository1 = createProxiedRepository( proxiedRepository1 );

        proxiedRepositories.add( proxiedArtifactRepository1 );

        wagonMock.get( path, new File( expectedFile.getParentFile(), expectedFile.getName() + ".tmp" ) );

        wagonMockControl.setThrowable( new TransferFailedException( "transfer failed" ) );

        wagonMockControl.replay();

        try

        {

            ProxiedArtifact proxied = requestHandler.get( path, proxiedRepositories, defaultManagedRepository );

            fail( "Found file: " + proxied.getFile() + "; but was expecting a failure" );

        }

        catch ( ResourceDoesNotExistException e )

        {

            // as expected

            wagonMockControl.verify();

            File tempFile = new File( expectedFile.getParentFile(), expectedFile.getName() + ".tmp" );

            assertFalse( "Check temporary file removed", tempFile.exists() );

        }

    }

    public void testGetRemovesTemporaryChecksumFileOnSuccess()

        throws ResourceDoesNotExistException, ProxyException, IOException

    {

        String path = "org/apache/maven/test/get-checksum-sha1-only/1.0/get-checksum-sha1-only-1.0.jar";

        File expectedFile = new File( defaultManagedRepository.getBasedir(), path );

        assertFalse( expectedFile.exists() );

        ProxiedArtifact proxied = requestHandler.get( path, proxiedRepositories, defaultManagedRepository );

        assertEquals( "Check file matches", expectedFile, proxied.getFile() );

        assertTrue( "Check file created", proxied.getFile().exists() );

        File tempFile = new File( proxied.getFile().getParentFile(), proxied.getFile().getName() + ".sha1.tmp" );

        assertFalse( "Check temporary file removed", tempFile.exists() );

    }

    public void testGetRemovesTemporaryChecksumFileOnError()

        throws ResourceDoesNotExistException, ProxyException, IOException, TransferFailedException,

        AuthorizationException

    {

        String path = "org/apache/maven/test/get-checksum-sha1-only/1.0/get-checksum-sha1-only-1.0.jar";

        File expectedFile = new File( defaultManagedRepository.getBasedir(), path );

        FileUtils.deleteDirectory( expectedFile.getParentFile() );

        assertFalse( expectedFile.exists() );

        proxiedRepository1 = createRepository( "proxied1", "test://..." );

        proxiedRepositories.clear();

        ProxiedArtifactRepository proxiedArtifactRepository1 = createProxiedRepository( proxiedRepository1 );

        proxiedRepositories.add( proxiedArtifactRepository1 );

        wagonMock.get( path, new File( expectedFile.getParentFile(), expectedFile.getName() + ".tmp" ) );

        mockFailedChecksums( path, expectedFile );

        wagonMockControl.replay();

        try

        {

            ProxiedArtifact proxied = requestHandler.get( path, proxiedRepositories, defaultManagedRepository );

            fail( "Found file: " + proxied.getFile() + "; but was expecting a failure" );

        }

        catch ( ResourceDoesNotExistException e )

        {

            // as expected

            wagonMockControl.verify();

            File tempFile = new File( expectedFile.getParentFile(), expectedFile.getName() + ".tmp" );

            assertFalse( "Check temporary file removed", tempFile.exists() );

            tempFile = new File( expectedFile.getParentFile(), expectedFile.getName() + ".sha1.tmp" );

            assertFalse( "Check temporary file removed", tempFile.exists() );

        }

    }

    public void testGetChecksumBothCorrect()

        throws ResourceDoesNotExistException, ProxyException, IOException

    {

        String path = "org/apache/maven/test/get-checksum-both-right/1.0/get-checksum-both-right-1.0.jar";

        File expectedFile = new File( defaultManagedRepository.getBasedir(), path );

        FileUtils.deleteDirectory( expectedFile.getParentFile() );

        assertFalse( expectedFile.exists() );

        ProxiedArtifact proxied = requestHandler.get( path, proxiedRepositories, defaultManagedRepository );

        assertEquals( "Check file matches", expectedFile, proxied.getFile() );

        assertTrue( "Check file created", proxied.getFile().exists() );

        File checksumFile = getChecksumFile( proxied.getFile(), "sha1" );

        assertTrue( "Check file created", checksumFile.exists() );

        assertEquals( "Check checksum", "066d76e459f7782c312c31e8a11b3c0f1e3e43a7 *get-checksum-both-right-1.0.jar",

                      FileUtils.readFileToString( checksumFile, null ).trim() );

        assertFalse( "Check file not created", getChecksumFile( proxied.getFile(), "md5" ).exists() );

    }

    public void testGetCorrectSha1NoMd5()

        throws ResourceDoesNotExistException, ProxyException, IOException

    {

        String path = "org/apache/maven/test/get-checksum-sha1-only/1.0/get-checksum-sha1-only-1.0.jar";

        File expectedFile = new File( defaultManagedRepository.getBasedir(), path );

        FileUtils.deleteDirectory( expectedFile.getParentFile() );

        assertFalse( expectedFile.exists() );

        ProxiedArtifact proxied = requestHandler.get( path, proxiedRepositories, defaultManagedRepository );

        assertEquals( "Check file matches", expectedFile, proxied.getFile() );

        assertTrue( "Check file created", proxied.getFile().exists() );

        File checksumFile = getChecksumFile( proxied.getFile(), "sha1" );

        assertTrue( "Check file created", checksumFile.exists() );

        assertEquals( "Check checksum", "748a3a013bf5eacf2bbb40a2ac7d37889b728837 *get-checksum-sha1-only-1.0.jar",

                      FileUtils.readFileToString( checksumFile, null ).trim() );

        assertFalse( "Check file not created", getChecksumFile( proxied.getFile(), "md5" ).exists() );

    }

    public void testGetCorrectSha1BadMd5()

        throws ResourceDoesNotExistException, ProxyException, IOException

    {

        String path = "org/apache/maven/test/get-checksum-sha1-bad-md5/1.0/get-checksum-sha1-bad-md5-1.0.jar";

        File expectedFile = new File( defaultManagedRepository.getBasedir(), path );

        FileUtils.deleteDirectory( expectedFile.getParentFile() );

        assertFalse( expectedFile.exists() );

        ProxiedArtifact proxied = requestHandler.get( path, proxiedRepositories, defaultManagedRepository );

        assertEquals( "Check file matches", expectedFile, proxied.getFile() );

        assertTrue( "Check file created", proxied.getFile().exists() );

        File checksumFile = getChecksumFile( proxied.getFile(), "sha1" );

        assertTrue( "Check file created", checksumFile.exists() );

        assertEquals( "Check checksum", "3dd1a3a57b807d3ef3fbc6013d926c891cbb8670 *get-checksum-sha1-bad-md5-1.0.jar",

                      FileUtils.readFileToString( checksumFile, null ).trim() );

        assertFalse( "Check file not created", getChecksumFile( proxied.getFile(), "md5" ).exists() );

    }

    public void testGetCorrectMd5NoSha1()

        throws ResourceDoesNotExistException, ProxyException, IOException

    {

        String path = "org/apache/maven/test/get-checksum-md5-only/1.0/get-checksum-md5-only-1.0.jar";

        File expectedFile = new File( defaultManagedRepository.getBasedir(), path );

        FileUtils.deleteDirectory( expectedFile.getParentFile() );

        assertFalse( expectedFile.exists() );

        ProxiedArtifact proxied = requestHandler.get( path, proxiedRepositories, defaultManagedRepository );

        assertEquals( "Check file matches", expectedFile, proxied.getFile() );

        assertTrue( "Check file created", proxied.getFile().exists() );

        File checksumFile = getChecksumFile( proxied.getFile(), "md5" );

        assertTrue( "Check file created", checksumFile.exists() );

        assertEquals( "Check checksum", "f3af5201bf8da801da37db8842846e1c *get-checksum-md5-only-1.0.jar",

                      FileUtils.readFileToString( checksumFile, null ).trim() );

        assertFalse( "Check file not created", getChecksumFile( proxied.getFile(), "sha1" ).exists() );

    }

    public void testGetCorrectMd5BadSha1()

        throws ResourceDoesNotExistException, ProxyException, IOException

    {

        String path = "org/apache/maven/test/get-checksum-md5-bad-sha1/1.0/get-checksum-md5-bad-sha1-1.0.jar";

        File expectedFile = new File( defaultManagedRepository.getBasedir(), path );

        FileUtils.deleteDirectory( expectedFile.getParentFile() );

        assertFalse( expectedFile.exists() );

        ProxiedArtifact proxied = requestHandler.get( path, proxiedRepositories, defaultManagedRepository );

        assertEquals( "Check file matches", expectedFile, proxied.getFile() );

        assertTrue( "Check file created", proxied.getFile().exists() );

        File checksumFile = getChecksumFile( proxied.getFile(), "md5" );

        assertTrue( "Check file created", checksumFile.exists() );

        assertEquals( "Check checksum", "8a02aa67549d27b2a03cd4547439c6d3 *get-checksum-md5-bad-sha1-1.0.jar",

                      FileUtils.readFileToString( checksumFile, null ).trim() );

        assertFalse( "Check file not created", getChecksumFile( proxied.getFile(), "sha1" ).exists() );

    }

    public void testGetWithNoChecksums()

        throws ResourceDoesNotExistException, ProxyException, IOException

    {

        String path = "org/apache/maven/test/get-default-layout/1.0/get-default-layout-1.0.jar";

        File expectedFile = new File( defaultManagedRepository.getBasedir(), path );

        expectedFile.delete();

        assertFalse( expectedFile.exists() );

        ProxiedArtifact proxied = requestHandler.get( path, proxiedRepositories, defaultManagedRepository );

        assertEquals( "Check file matches", expectedFile, proxied.getFile() );

        assertTrue( "Check file created", proxied.getFile().exists() );

        assertFalse( "Check file not created", getChecksumFile( proxied.getFile(), "md5" ).exists() );

        assertFalse( "Check file not created", getChecksumFile( proxied.getFile(), "sha1" ).exists() );

    }

    public void testGetBadMd5BadSha1()

        throws ResourceDoesNotExistException, ProxyException, IOException

    {

        String path = "org/apache/maven/test/get-checksum-both-bad/1.0/get-checksum-both-bad-1.0.jar";

        File expectedFile = new File( defaultManagedRepository.getBasedir(), path );

        FileUtils.deleteDirectory( expectedFile.getParentFile() );

        assertFalse( expectedFile.exists() );

        try

        {

            ProxiedArtifact proxied = requestHandler.get( path, proxiedRepositories, defaultManagedRepository );

            fail( "Found file: " + proxied.getFile() + "; but was expecting a failure" );

        }

        catch ( ResourceDoesNotExistException e )

        {

            // expect a failure

            assertFalse( "Check file not created", expectedFile.exists() );

            assertFalse( "Check file not created", getChecksumFile( expectedFile, "md5" ).exists() );

            assertFalse( "Check file not created", getChecksumFile( expectedFile, "sha1" ).exists() );

        }

    }

    public void testGetChecksumTransferFailed()

        throws ResourceDoesNotExistException, ProxyException, IOException, TransferFailedException,

        AuthorizationException

    {

        String path = "org/apache/maven/test/get-checksum-sha1-only/1.0/get-checksum-sha1-only-1.0.jar";

        File expectedFile = new File( defaultManagedRepository.getBasedir(), path );

        FileUtils.deleteDirectory( expectedFile.getParentFile() );

        assertFalse( expectedFile.exists() );

        proxiedRepository1 = createRepository( "proxied1", "test://..." );

        proxiedRepositories.clear();

        ProxiedArtifactRepository proxiedArtifactRepository1 = createProxiedRepository( proxiedRepository1 );

        proxiedRepositories.add( proxiedArtifactRepository1 );

        wagonMock.get( path, new File( expectedFile.getParentFile(), expectedFile.getName() + ".tmp" ) );

        mockFailedChecksums( path, expectedFile );

        wagonMockControl.replay();

        try

        {

            ProxiedArtifact proxied = requestHandler.get( path, proxiedRepositories, defaultManagedRepository );

            fail( "Found file: " + proxied.getFile() + "; but was expecting a failure" );

        }

        catch ( ResourceDoesNotExistException e )

        {

            // as expected

            wagonMockControl.verify();

            assertFalse( "Check file not created", expectedFile.exists() );

            assertFalse( "Check file not created", getChecksumFile( expectedFile, "md5" ).exists() );

            assertFalse( "Check file not created", getChecksumFile( expectedFile, "sha1" ).exists() );

        }

    }

    public void testGetAlwaysBadChecksumPresentLocallyAbsentRemote()

        throws ResourceDoesNotExistException, ProxyException, IOException

    {

        String path = "org/apache/maven/test/get-bad-local-checksum/1.0/get-bad-local-checksum-1.0.jar";

        File expectedFile = new File( defaultManagedRepository.getBasedir(), path );

        String unexpectedContents = FileUtils.readFileToString( expectedFile, null );

        assertTrue( expectedFile.exists() );

        ProxiedArtifact proxied = requestHandler.getAlways( path, proxiedRepositories, defaultManagedRepository );

        assertEquals( "Check file matches", expectedFile, proxied.getFile() );

        assertTrue( "Check file created", proxied.getFile().exists() );

        File proxiedFile = new File( proxiedRepository1.getBasedir(), path );

        String expectedContents = FileUtils.readFileToString( proxiedFile, null );

        assertEquals( "Check file contents", expectedContents, FileUtils.readFileToString( proxied.getFile(), null ) );

        assertFalse( "Check file contents", unexpectedContents.equals( FileUtils.readFileToString( proxied.getFile(), null ) ) );

        assertFalse( "Check checksum removed", new File( proxied.getFile().getParentFile(), proxied.getFile().getName() + ".sha1" ).exists() );

        assertFalse( "Check checksum removed", new File( proxied.getFile().getParentFile(), proxied.getFile().getName() + ".md5" ).exists() );

    }

    public void testGetChecksumPresentInManagedRepo()

        throws ResourceDoesNotExistException, ProxyException, IOException

    {

        String path =

            "org/apache/maven/test/get-checksum-from-managed-repo/1.0/get-checksum-from-managed-repo-1.0.jar.sha1";

        File expectedFile = new File( defaultManagedRepository.getBasedir(), path );

        String expectedContents = FileUtils.readFileToString( expectedFile, null );

        assertTrue( expectedFile.exists() );

        ProxiedArtifact proxied = requestHandler.get( path, proxiedRepositories, defaultManagedRepository );

        assertEquals( "Check file matches", expectedFile, proxied.getFile() );

        assertTrue( "Check file created", proxied.getFile().exists() );

        File proxiedFile = new File( proxiedRepository1.getBasedir(), path );

        String unexpectedContents = FileUtils.readFileToString( proxiedFile, null );

        assertEquals( "Check file contents", expectedContents, FileUtils.readFileToString( proxied.getFile(), null ) );