package org.codehaus.plexus.digest;

/*
 * Copyright 2001-2006 The Codehaus.
 *
 * Licensed 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.codehaus.plexus.PlexusTestCase;

import java.io.File;
import java.io.IOException;
import java.security.NoSuchAlgorithmException;

/**
 * Test the digester.
 *
 * @author <a href="mailto:brett@apache.org">Brett Porter</a>
 */
public class DigesterTest
    extends PlexusTestCase
{
    private static final String MD5 = "adbc688ce77fa2aece4bb72cad9f98ba";

    private static final String SHA1 = "2a7b459938e12a2dc35d1bf6cff35e9c2b592fa9";

    private static final String WRONG_SHA1 = "4d8703779816556cdb8be7f6bb5c954f4b5730e2";

    private Digester sha1Digest;

    private Digester md5Digest;

    protected void setUp()
        throws Exception
    {
        super.setUp();

        sha1Digest = (Digester) lookup( Digester.ROLE, "sha1" );
        md5Digest = (Digester) lookup( Digester.ROLE, "md5" );
    }

    public void testAlgorithm()
    {
        assertEquals( "SHA-1", sha1Digest.getAlgorithm() );
        assertEquals( "MD5", md5Digest.getAlgorithm() );
    }

    public void testBareDigestFormat()
        throws DigesterException, IOException
    {
        File file = new File( getClass().getResource( "/test-file.txt" ).getPath() );

        try
        {
            md5Digest.verify( file, MD5 );
        }
        catch ( DigesterException e )
        {
            fail( "Bare format MD5 must not throw exception" );
        }

        try
        {
            sha1Digest.verify( file, SHA1 );
        }
        catch ( DigesterException e )
        {
            fail( "Bare format SHA1 must not throw exception" );
        }

        try
        {
            sha1Digest.verify( file, WRONG_SHA1 );
            fail( "wrong checksum must throw an exception" );
        }
        catch ( DigesterException e )
        {
            //expected
        }
    }

    public void testOpensslDigestFormat()
        throws IOException, DigesterException
    {
        File file = new File( getClass().getResource( "/test-file.txt" ).getPath() );

        try
        {
            md5Digest.verify( file, "MD5(test-file.txt)= " + MD5 );
        }
        catch ( DigesterException e )
        {
            fail( "OpenSSL MD5 format must not cause exception" );
        }

        try
        {
            md5Digest.verify( file, "MD5 (test-file.txt) = " + MD5 );
        }
        catch ( DigesterException e )
        {
            fail( "FreeBSD MD5 format must not cause exception" );
        }

        try
        {
            sha1Digest.verify( file, "SHA1(test-file.txt)= " + SHA1 );
        }
        catch ( DigesterException e )
        {
            fail( "OpenSSL SHA1 format must not cause exception" );
        }

        try
        {
            sha1Digest.verify( file, "SHA1 (test-file.txt) = " + SHA1 );
        }
        catch ( DigesterException e )
        {
            fail( "FreeBSD SHA1 format must not cause exception" );
        }

        try
        {
            sha1Digest.verify( file, "SHA1 (FOO) = " + SHA1 );
            fail( "Wrong filename should cause an exception" );
        }
        catch ( DigesterException e )
        {
            //expected
        }

        try
        {
            sha1Digest.verify( file, "SHA1 (test-file.txt) = " + WRONG_SHA1 );
            fail( "Wrong sha1 should cause an exception" );
        }
        catch ( DigesterException e )
        {
            //expected
        }
    }

    public void testGnuDigestFormat()
        throws NoSuchAlgorithmException, IOException, DigesterException
    {
        File file = new File( getClass().getResource( "/test-file.txt" ).getPath() );

        try
        {
            md5Digest.verify( file, MD5 + " *test-file.txt" );
        }
        catch ( DigesterException e )
        {
            fail( "GNU format MD5 must not cause exception" );
        }

        try
        {
            md5Digest.verify( file, MD5 + " test-file.txt" );
        }
        catch ( DigesterException e )
        {
            fail( "GNU text format MD5 must not cause exception" );
        }

        try
        {
            sha1Digest.verify( file, SHA1 + " *test-file.txt" );
        }
        catch ( DigesterException e )
        {
            fail( "GNU format SHA1 must not cause exception" );
        }

        try
        {
            sha1Digest.verify( file, SHA1 + " test-file.txt" );
        }
        catch ( DigesterException e )
        {
            fail( "GNU text format SHA1 must not cause exception" );
        }

        try
        {
            sha1Digest.verify( file, SHA1 + " FOO" );
            fail( "Wrong filename cause an exception" );
        }
        catch ( DigesterException e )
        {
            //expected
        }

        try
        {
            sha1Digest.verify( file, WRONG_SHA1 + " test-file.txt" );
            fail( "Wrong SHA1 cause an exception" );
        }
        catch ( DigesterException e )
        {
            //expected
        }
    }

    public void testUntrimmedContent()
        throws NoSuchAlgorithmException, IOException
    {
        File file = new File( getClass().getResource( "/test-file.txt" ).getPath() );
        try
        {
            sha1Digest.verify( file, SHA1 + " *test-file.txt \n" );
        }
        catch ( DigesterException e )
        {
            fail( "GNU untrimmed SHA1 must not cause exception" );
        }
    }
}