/*
 * Copyright 2006-2009 The Apache Software Foundation.
 *
 *  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.
 *
 */
/*
 * Copyright 2005 Sun Microsystems, Inc. All rights reserved.
 */
package javax.xml.crypto.test.dsig;

import javax.xml.crypto.*;
import javax.xml.crypto.dsig.dom.DOMValidateContext;
import javax.xml.crypto.dsig.*;
import java.security.*;
import javax.crypto.SecretKey;
import javax.crypto.spec.SecretKeySpec;

import junit.framework.*;

/**
 * Unit test for javax.xml.crypto.dsig.XMLValidateContext
 *
 * @version $Id$
 * @author Valerie Peng
 */
public class XMLValidateContextTest extends TestCase {

    private XMLValidateContext defContext;
    private Key[] KEYS;
    private KeySelector[] KEY_SELECTORS;

    public XMLValidateContextTest() {
	super("XMLValidateContextTest");
    }

    public XMLValidateContextTest(String name) {
	super(name);
    }

    public void setUp() throws Exception {

	// set up the validatingKeys
	KEYS = new Key[3];
	KEYS[0] = TestUtils.getPublicKey("DSA");
	KEYS[1] = TestUtils.getPublicKey("RSA");
	KEYS[2] = new SecretKeySpec(new byte[16], "HmacSHA1");

	// set up the default XMLValidateContext
	SecretKey sk = new SecretKeySpec(new byte[8], "DES");
	defContext = new DOMValidateContext(sk, TestUtils.newDocument());

	// set up the key selectors
	KEY_SELECTORS = new KeySelector[1];
        KEY_SELECTORS[0] = KeySelector.singletonKeySelector(sk);
    }

    public void tearDown() { }

    public void testsetngetKeySelector() throws Exception {
	defContext.setKeySelector(null);
	assertNull(defContext.getKeySelector());

	for (int i = 0; i < KEY_SELECTORS.length; i++) {
	    defContext.setKeySelector(KEY_SELECTORS[i]);
	    assertEquals(defContext.getKeySelector(), KEY_SELECTORS[i]);
	}
    }

    public void testsetngetBaseURI() throws Exception {
	assertNull(defContext.getBaseURI());

	String uri = "http://www.w3.org/2000/09/xmldsig#";
	defContext.setBaseURI(uri);
	assertEquals(defContext.getBaseURI(), uri);
	defContext.setBaseURI(null);
	assertNull(defContext.getBaseURI());
    }

    public void testsetngetProperty() throws Exception {
	String name = "key";
	assertNull(defContext.getProperty(name));
	try {
	    defContext.setProperty(null, null);
	    fail("Should raise a NPE with a null name");
	} catch (NullPointerException npe) {
	} catch (Exception ex) {
	    fail("Should raise a NPE instead of " + ex);
	}
	String value1 = "value#1";
	String value2 = "value#2";
	assertNull(defContext.setProperty(name, value1));
	assertEquals(defContext.getProperty(name), value1);
	assertEquals(defContext.setProperty(name, value2), value1);
	assertEquals(defContext.getProperty(name), value2);
	assertEquals(defContext.setProperty(name, null), value2);
	assertNull(defContext.getProperty(name));
    }

    public void testsetngetURIDereferencer() throws Exception {
	assertNull(defContext.getURIDereferencer());
	byte[] data = "simpleDereferencer".getBytes();
	URIDereferencer deref = new TestUtils.OctetStreamURIDereferencer(data);

	defContext.setURIDereferencer(deref);
	assertEquals(defContext.getURIDereferencer(), deref);
	defContext.setURIDereferencer(null);
	assertNull(defContext.getURIDereferencer());
    }
}