src_unitTests/org/apache/xml/security/test/utils/IdResolverTest.java

package org.apache.xml.security.test.utils;

import java.io.ByteArrayInputStream;

import javax.xml.parsers.DocumentBuilderFactory;

import org.apache.xml.security.utils.IdResolver;
import org.w3c.dom.Document;
import org.w3c.dom.Element;

import junit.framework.Test;
import junit.framework.TestCase;
import junit.framework.TestSuite;

public class IdResolverTest extends TestCase {
	public static Test suite() {
	      return new TestSuite(IdResolverTest.class);
	   }
	public void testIdSoap() throws Exception {
		String s="<env:Envelope xmlns:SOAP-SEC=\"http://schemas.xmlsoap.org/soap/security/2000-12\" xmlns:env=\"http://www.w3.org/2001/12/soap-envelope\" actor=\"some-uri\" mustUnderstand=\"1\">\r\n" + 
				"<env:Header><SOAP-SEC:Signature>xxxx</SOAP-SEC:Signature></env:Header>\r\n" + 
				"<env:Body SOAP-SEC:id=\"Body\">This is signed together with it\'s Body ancestor</env:Body>\r\n" + 
				"</env:Envelope>";
		DocumentBuilderFactory dbf=DocumentBuilderFactory.newInstance();
		dbf.setNamespaceAware(true);
		Document doc=dbf.newDocumentBuilder().parse(new ByteArrayInputStream(s.getBytes()));
		Element el=IdResolver.getElementById(doc, "Body");
		assertNotNull(el);
		assertEquals("Body",el.getLocalName());
	}
	public void testIdWithOtherIdSoap() throws Exception {
		String s="<env:Envelope xmlns:SOAP-SEC=\"http://schemas.xmlsoap.org/soap/security/2000-12\" xmlns:env=\"http://www.w3.org/2001/12/soap-envelope\" actor=\"some-uri\" mustUnderstand=\"1\">\r\n" + 
				"<env:Header><SOAP-SEC:Signature>xxxx</SOAP-SEC:Signature></env:Header>\r\n" + 
				"<a id=\"Body\"/><env:Body SOAP-SEC:id=\"Body\">This is signed together with it\'s Body ancestor</env:Body>\r\n" + 
				"</env:Envelope>";
		DocumentBuilderFactory dbf=DocumentBuilderFactory.newInstance();
		dbf.setNamespaceAware(true);
		Document doc=dbf.newDocumentBuilder().parse(new ByteArrayInputStream(s.getBytes()));
		Element el=IdResolver.getElementById(doc, "Body");
		assertNotNull(el);
		assertEquals("Body",el.getLocalName());
	}
	public void testANoId() throws Exception {
		String s="<env:Envelope xmlns:SOAP-SEC=\"http://schemas.xmlsoap.org/soap/security/2000-12\" xmlns:env=\"http://www.w3.org/2001/12/soap-envelope\" actor=\"some-uri\" mustUnderstand=\"1\">\r\n" + 
				"<env:Header><SOAP-SEC:Signature>xxxx</SOAP-SEC:Signature></env:Header>\r\n" + 
				"<a id=\"Body\"/>" + 
				"</env:Envelope>";
		DocumentBuilderFactory dbf=DocumentBuilderFactory.newInstance();
		dbf.setNamespaceAware(true);
		Document doc=dbf.newDocumentBuilder().parse(new ByteArrayInputStream(s.getBytes()));
		Element el=IdResolver.getElementById(doc, "Body");
		assertNotNull(el);
		assertEquals("a",el.getLocalName());
	}
}