File: KeySelectorTest.java

package info (click to toggle)
libxml-security-java 1.4.3-2%2Bdeb6u1
  • links: PTS, VCS
  • area: main
  • in suites: squeeze-lts
  • size: 14,184 kB
  • ctags: 5,406
  • sloc: java: 41,126; xml: 22,042; sh: 196; makefile: 17
file content (106 lines) | stat: -rw-r--r-- 3,101 bytes parent folder | download | duplicates (4)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
/*
 * Copyright 2006 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;

import java.util.*;
import java.security.Key;
import java.security.cert.X509Certificate;
import javax.xml.crypto.*;
import javax.xml.crypto.dsig.keyinfo.*;
import javax.xml.crypto.dsig.*;

import junit.framework.*;

/**
 * Unit test for javax.xml.crypto.KeySelector
 *
 * @version $Id$
 * @author Valerie Peng
 */
public class KeySelectorTest extends TestCase {
    private Key key;
    private KeySelector selector1;

    private class MyOwnKey implements Key {
	private String algo;
	private byte[] val;
	MyOwnKey(String algorithm, byte[] value) {
	    algo = algorithm;
	    val = (byte[]) value.clone();
	}
	
	public String getAlgorithm() {
	    return algo;
	}
	public byte[] getEncoded() {
	    return val;
	}
	public String getFormat() {
	    return "RAW";
	}
    }

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

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

    public void setUp() throws Exception {
	// selector1: singletonKeySelector
	key = new MyOwnKey("test", new byte[16]);
	selector1 = KeySelector.singletonKeySelector(key);
    }

    public void tearDown() { }

    public void testselect() throws Exception {
	KeyInfoFactory factory = KeyInfoFactory.getInstance
	    ("DOM", new org.jcp.xml.dsig.internal.dom.XMLDSigRI());
	X509Data obj = factory.newX509Data(Collections.singletonList("CN=foo"));
	KeyInfo info = factory.newKeyInfo(Collections.singletonList(obj));
	//@@@@@what about other types of X509Data, i.e. subject name String,
	// X509IssuerSerial objects, etc?
	XMLSignatureFactory dsigFac = XMLSignatureFactory.getInstance
	    ("DOM", new org.jcp.xml.dsig.internal.dom.XMLDSigRI());
	SignatureMethod sm1 = 
	    dsigFac.newSignatureMethod(SignatureMethod.DSA_SHA1, null);
	SignatureMethod sm2 = 
	    dsigFac.newSignatureMethod(SignatureMethod.RSA_SHA1, null);

	assertTrue(compareKey(key, selector1.select
	    (info, KeySelector.Purpose.VERIFY, sm1, null).getKey()));
	assertTrue(compareKey(key, selector1.select
	    (info, KeySelector.Purpose.VERIFY, sm2, null).getKey()));
    }
    
    private static boolean compareKey(Object answer, Key key) {
	boolean result = false;
	if (answer instanceof MyOwnKey) {
	    result = ((MyOwnKey) answer == key);
	} else if (answer instanceof X509Certificate) {
	    result =
	        ((X509Certificate)answer).getPublicKey().equals(key);
	}
	return result;
    }
}