/*
 *  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 Vera Y. Petrashkova
* @version $Revision$
*/

package org.apache.harmony.security.tests.java.security;

import dalvik.annotation.TestTargetClass;
import dalvik.annotation.TestTargetNew;
import dalvik.annotation.TestTargets;
import dalvik.annotation.TestLevel;

import java.security.UnrecoverableKeyException;

import junit.framework.TestCase;
@TestTargetClass(UnrecoverableKeyException.class)
/**
 * Tests for <code>UnrecoverableKeyException</code> class constructors and
 * methods.
 *
 */
public class UnrecoverableKeyExceptionTest extends TestCase {

    static String[] msgs = {
            "",
            "Check new message",
            "Check new message Check new message Check new message Check new message Check new message" };

    static Throwable tCause = new Throwable("Throwable for exception");

    /**
     * Test for <code>UnrecoverableKeyException()</code> constructor
     * Assertion: constructs UnrecoverableKeyException with no detail message
     */
    @TestTargetNew(
        level = TestLevel.COMPLETE,
        notes = "",
        method = "UnrecoverableKeyException",
        args = {}
    )
    public void testUnrecoverableKeyException01() {
        UnrecoverableKeyException tE = new UnrecoverableKeyException();
        assertNull("getMessage() must return null.", tE.getMessage());
        assertNull("getCause() must return null", tE.getCause());
    }

    /**
     * Test for <code>UnrecoverableKeyException(String)</code> constructor
     * Assertion: constructs UnrecoverableKeyException with detail message msg.
     * Parameter <code>msg</code> is not null.
     */
    @TestTargetNew(
        level = TestLevel.PARTIAL_COMPLETE,
        notes = "",
        method = "UnrecoverableKeyException",
        args = {java.lang.String.class}
    )
    public void testUnrecoverableKeyException02() {
        UnrecoverableKeyException tE;
        for (int i = 0; i < msgs.length; i++) {
            tE = new UnrecoverableKeyException(msgs[i]);
            assertEquals("getMessage() must return: ".concat(msgs[i]), tE
                    .getMessage(), msgs[i]);
            assertNull("getCause() must return null", tE.getCause());
        }
    }

    /**
     * Test for <code>UnrecoverableKeyException(String)</code> constructor
     * Assertion: constructs UnrecoverableKeyException when <code>msg</code>
     * is null
     */
    @TestTargetNew(
        level = TestLevel.PARTIAL_COMPLETE,
        notes = "",
        method = "UnrecoverableKeyException",
        args = {java.lang.String.class}
    )
    public void testUnrecoverableKeyException03() {
        String msg = null;
        UnrecoverableKeyException tE = new UnrecoverableKeyException(msg);
        assertNull("getMessage() must return null.", tE.getMessage());
        assertNull("getCause() must return null", tE.getCause());
    }

}
