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


import java.io.File;
import java.security.Security;

import javax.xml.crypto.test.KeySelectors;

import static org.junit.Assert.assertTrue;

/**
 * This is a testcase to validate the "merlin-xpath-filter2-three" testcases
 * under data/vectors/ie/baltimore/merlin-examples directory
 *
 */
public class BaltimoreXPathFilter2ThreeTest {

    private SignatureValidator validator;

    static {
        Security.insertProviderAt
            (new org.apache.jcp.xml.dsig.internal.dom.XMLDSigRI(), 1);
    }

    public BaltimoreXPathFilter2ThreeTest() {
        String fs = System.getProperty("file.separator");
        String base = System.getProperty("basedir") == null ? "./": System.getProperty("basedir");
        base += fs + "src/test/resources"
            + fs + "interop" + fs + "xfilter2";
        validator = new SignatureValidator(new File
            (base, "merlin-xpath-filter2-three"));
    }

    @org.junit.Test
    public void testSignSpec() throws Exception {
        String file = "sign-spec.xml";

        boolean coreValidity = validator.validate(file,
                    new KeySelectors.KeyValueKeySelector());
        assertTrue("Signature failed core validation#1", coreValidity);

        coreValidity = validator.validate(file,
                    new KeySelectors.RawX509KeySelector());
        assertTrue("Signature failed core validation#2", coreValidity);
    }

    @org.junit.Test
    public void testSignXfdl() throws Exception {
        String file = "sign-xfdl.xml";

        boolean coreValidity = validator.validate(file,
                    new KeySelectors.KeyValueKeySelector());
        assertTrue("Signature failed core validation#1", coreValidity);

        coreValidity = validator.validate(file,
                    new KeySelectors.RawX509KeySelector());
        assertTrue("Signature failed core validation#2", coreValidity);
    }

}