modules/security/src/test/impl/java.injected/java/security/Signature_Impl2Test.java

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

/**
* @author Boris V. Kuznetsov
*/

package java.security;

import org.apache.harmony.security.tests.support.MySignature1;
import org.apache.harmony.security.tests.support.MySignature2;

import junit.framework.TestCase;

/**
 * Tests for <code>Signature</code> constructor and methods
 * 
 */
public class Signature_Impl2Test extends TestCase {

	/**
	 * Provider
	 */
	Provider p;
	
	/*
	 * @see TestCase#setUp()
	 */
	protected void setUp() throws Exception {
		super.setUp();
		p = new MyProvider();
		Security.insertProviderAt(p, 1);
	}

	/*
	 * @see TestCase#tearDown()
	 */
	protected void tearDown() throws Exception {
		super.tearDown();
		Security.removeProvider(p.getName());
	}

	/*
	 * Class under test for Signature getInstance(String)
	 */
	public void testGetInstanceString1() throws Exception {
		Signature sig = Signature.getInstance("ABC");		
		checkSig1(sig, p);
	}
	
	/*
	 * Class under test for Signature getInstance(String)
	 */
	public void testGetInstanceString2() throws Exception {
		Signature sig = Signature.getInstance("CBA");
		checkSig2(sig, p);
	}

	/*
	 * Class under test for Signature getInstance(String, String)
	 */
	public void testGetInstanceStringString1() throws Exception {
		Signature sig = Signature.getInstance("ABC", "MyProvider");		
		checkSig1(sig, p);
	}
	
	/*
	 * Class under test for Signature getInstance(String, String)
	 */
	public void testGetInstanceStringString2() throws Exception {
		Signature sig = Signature.getInstance("CBA", "MyProvider");
		checkSig2(sig, p);
	}


	/*
	 * Class under test for Signature getInstance(String, Provider)
	 */
	public void testGetInstanceStringProvider1() throws Exception {
		Provider p1 = new MyProvider();
		Signature sig = Signature.getInstance("ABC", p1);		
		checkSig1(sig, p1);
	}
	
	/*
	 * Class under test for Signature getInstance(String, Provider)
	 */
	public void testGetInstanceStringProvider2() throws Exception {
		Provider p2 = new MyProvider();
		Signature sig = Signature.getInstance("CBA", p2);
		checkSig2(sig, p2);
	}

	private void checkSig1(Signature s, Provider p) throws Exception {
        byte[] b = { 1, 2, 3, 4 };
        assertTrue("getInstance() failed", s instanceof MySignature1);
        assertEquals("getProvider() failed", p, s.getProvider());
        assertEquals("getAlgorithm() failed", "ABC", s.getAlgorithm());

        try {
            s.sign();
            fail("No expected SignatureException");
        } catch (SignatureException e) {
        }

        s.initVerify(new MyPublicKey());

        try {
            s.sign();
            fail("No expected SignatureException");
        } catch (SignatureException e) {
        }

        s.initSign(new MyPrivateKey());
        s.sign();
        assertEquals("Incorrect state", Signature.SIGN, ((MySignature1) s)
                .getState());
        assertTrue("sign() failed", ((MySignature1) s).runEngineSign);

        s.initVerify(new MyPublicKey());
        s.update((byte) 1);

        s.initSign(new MyPrivateKey());
        s.update((byte) 1);

        assertEquals("Incorrect state", Signature.SIGN, ((MySignature1) s)
                .getState());
        assertTrue("sign() failed", ((MySignature1) s).runEngineUpdate1);

        s.initSign(new MyPrivateKey());

        try {
            s.verify(b);
            fail("No expected SignatureException");
        } catch (SignatureException e) {
        }
        s.initVerify(new MyPublicKey());
        s.verify(b);
        assertEquals("Incorrect state", Signature.VERIFY, ((MySignature1) s)
                .getState());
        assertTrue("verify() failed", ((MySignature1) s).runEngineVerify);
    }
	
	private void checkSig2(Signature s, Provider p) throws Exception {
        byte[] b = { 1, 2, 3, 4 };

        assertEquals("getProvider() failed", p, s.getProvider());
        assertEquals("getAlgorithm() failed", "CBA", s.getAlgorithm());

        s.initVerify(new MyCertificate());

        try {
            s.sign(b, 0, 5);
            fail("No expected IllegalArgumentException 1");
        } catch (IllegalArgumentException e) {
        }

        s.initSign(new MyPrivateKey());
        s.sign(b, 0, 3);

        assertTrue("sign() failed", MySignature2.runEngineSign);
        s.update(b);
        s.initSign(new MyPrivateKey());
        s.update(b);
        assertTrue("update() failed", MySignature2.runEngineUpdate2);

        s.initSign(new MyPrivateKey());
        try {
            s.verify(b, 0, 3);
            fail("No expected SignatureException");
        } catch (SignatureException e) {
        }
        s.initVerify(new MyPublicKey());

        try {
            s.verify(b, 0, 5);
            fail("No expected IllegalArgumentException");
        } catch (IllegalArgumentException e) {
        } catch (SignatureException e) {
        }

        s.verify(b, 0, 3);
        assertTrue("verify() failed", MySignature2.runEngineVerify);
    }
	
	private class MyProvider extends Provider {
		MyProvider() {
			super("MyProvider", 1.0, "Provider for testing");
			put("Signature.ABC", "org.apache.harmony.security.tests.support.MySignature1");
			put("Signature.CBA", "org.apache.harmony.security.tests.support.MySignature2");
		}
		
		MyProvider(String name, double version, String info) {
			super(name, version, info);
		}
	}
	private class MyKey implements Key {
		public String getFormat() {
			return "123";
		}
		public byte[] getEncoded() {
			return null;
		}
		public String getAlgorithm() {
			return "aaa";
		}		
	}
	
	private class MyPublicKey extends MyKey implements PublicKey {}

	private class MyPrivateKey extends MyKey implements PrivateKey {}
	
	private class MyCertificate extends java.security.cert.Certificate {	
		public  MyCertificate() {
			super("MyCertificateType");
		}
		
		public PublicKey getPublicKey() {
			return new MyPublicKey();
		}
		
		public byte[] getEncoded() {
			return null;
		}
		public void verify(PublicKey key) {}
		
		public void verify(PublicKey key, String sigProvider) {}
		
		public String toString() {
			return "MyCertificate";
		}
	}
}