modules/security/src/test/api/java.injected/java/security/SignatureTest.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 junit.framework.TestCase;

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

	/*
	 * Class under test for Object clone()
	 */
	public void testClone() {
		MySignature1 s = new MySignature1("ABC");
		try {
			s.clone();
			fail("No expected CloneNotSupportedException");
		} catch (CloneNotSupportedException e) {	
		}	
	}

	public void testGetProvider() {
		MySignature1 s = new MySignature1("ABC");
		
        assertEquals("state", Signature.UNINITIALIZED, s.getState());
        assertNull("provider", s.getProvider());
	}

	public void testGetAlgorithm() {
		MySignature1 s = new MySignature1("ABC");

        assertEquals("state", Signature.UNINITIALIZED, s.getState());
        assertEquals("algorithm", "ABC", s.getAlgorithm());
	}

	/*
	 * Class under test for void initVerify(PublicKey)
	 */
	public void testInitVerifyPublicKey() throws InvalidKeyException {
		MySignature1 s = new MySignature1("ABC");

        s.initVerify(new MyPublicKey());
        assertEquals("state", Signature.VERIFY, s.getState());
        assertTrue("initVerify() failed", s.runEngineInitVerify);
	}

	/*
	 * Class under test for void initVerify(Certificate)
	 */
	public void testInitVerifyCertificate() throws InvalidKeyException {
		MySignature1 s = new MySignature1("ABC");

        s.initVerify(new MyCertificate());
        assertEquals("state", Signature.VERIFY, s.getState());
        assertTrue("initVerify() failed", s.runEngineInitVerify);
	}

	/*
	 * Class under test for void initSign(PrivateKey)
	 */
	public void testInitSignPrivateKey() throws InvalidKeyException {
		MySignature1 s = new MySignature1("ABC");

        s.initSign(new MyPrivateKey());
        assertEquals("state", Signature.SIGN, s.getState());
        assertTrue("initSign() failed", s.runEngineInitSign);
	}

	/*
	 * Class under test for void initSign(PrivateKey, SecureRandom)
	 */
	public void testInitSignPrivateKeySecureRandom() throws InvalidKeyException {
		MySignature1 s = new MySignature1("ABC");

        s.initSign(new MyPrivateKey(), new SecureRandom());
        assertEquals("state", Signature.SIGN, s.getState());
        assertTrue("initSign() failed", s.runEngineInitSign);
	}

	/*
	 * Class under test for byte[] sign()
	 */
	public void testSign() throws Exception {
		MySignature1 s = new MySignature1("ABC");
		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("state", Signature.SIGN, s.getState());
        assertTrue("sign() failed", s.runEngineSign);
	}

	/*
	 * Class under test for boolean verify(byte[])
	 */
	public void testVerifybyteArray() throws Exception {
		MySignature1 s = new MySignature1("ABC");
		byte[] b = {1, 2, 3, 4};
		try {
			s.verify(b);
			fail("No expected SignatureException");
		} catch (SignatureException e) {		
		}

		s.initSign(new MyPrivateKey());
		try {
			s.verify(b);
			fail("No expected SignatureException");
		} catch (SignatureException e) {		
		}
		
		s.initVerify(new MyPublicKey());
        s.verify(b);
        assertEquals("state", Signature.VERIFY, s.getState());
        assertTrue("verify() failed", s.runEngineVerify);
	}

	/*
	 * Class under test for boolean verify(byte[], int, int)
	 */
	public void testVerifybyteArrayintint() throws Exception {
		MySignature1 s = new MySignature1("ABC");
		byte[] b = {1, 2, 3, 4};
		try {
			s.verify(b, 0, 3);
			fail("No expected SignatureException");
		} catch (SignatureException e) {		
		}

		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) {		
		}
		
		s.verify(b, 0, 3);
        assertEquals("state", Signature.VERIFY, s.getState());
        assertTrue("verify() failed", s.runEngineVerify);
	}

	/*
	 * Class under test for void update(byte)
	 */
	public void testUpdatebyte() throws Exception {
		MySignature1 s = new MySignature1("ABC");
		try {
			s.update((byte)1);
			fail("No expected SignatureException");
		} catch (SignatureException e) {		
		}

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

        assertEquals("state", Signature.SIGN, s.getState());
        assertTrue("update() failed", s.runEngineUpdate1);
	}

	/*
	 * Class under test for void update(byte[])
	 */
	public void testUpdatebyteArray() throws Exception {
		MySignature1 s = new MySignature1("ABC");
		byte[] b = {1, 2, 3, 4};
		try {
			s.update(b);
			fail("No expected SignatureException");
		} catch (SignatureException e) {		
		}

		s.initVerify(new MyPublicKey());
        s.update(b);
        s.initSign(new MyPrivateKey());
        s.update(b);

        assertEquals("state", Signature.SIGN, s.getState());
        assertTrue("update() failed", s.runEngineUpdate2);
	}

	/*
	 * Class under test for void update(byte[], int, int)
	 */
	public void testUpdatebyteArrayintint() throws Exception {
		MySignature1 s = new MySignature1("ABC");
		byte[] b = {1, 2, 3, 4};
		try {
			s.update(b, 0, 3);
			fail("No expected SignatureException");
		} catch (SignatureException e) {		
		}

		s.initVerify(new MyPublicKey());
        s.update(b, 0, 3);
        s.initSign(new MyPrivateKey());
        s.update(b, 0, 3);

        assertEquals("state", Signature.SIGN, s.getState());
        assertTrue("update() failed", s.runEngineUpdate2);
	}

	/*
	 * Class under test for void setParameter(String, Object)
	 */
	public void testSetParameterStringObject() {
		MySignature1 s = new MySignature1("ABC");
		s.setParameter("aaa", new Object());
	}

	/*
	 * Class under test for void setParameter(AlgorithmParameterSpec)
	 */
	public void testSetParameterAlgorithmParameterSpec() throws InvalidAlgorithmParameterException {
		MySignature1 s = new MySignature1("ABC");
		try {
			s.setParameter((java.security.spec.AlgorithmParameterSpec)null);
			fail("No expected UnsupportedOperationException");
		} catch (UnsupportedOperationException e){	
		}
	}

	public void testGetParameter() {
		MySignature1 s = new MySignature1("ABC");
        s.getParameter("aaa");
	}
	
	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";
		}
	}
}