package eu.dnetlib.enabling.aas.utils;

import org.opensaml.saml2.core.Assertion;
import org.opensaml.saml2.core.Issuer;
import org.opensaml.saml2.core.impl.AssertionBuilder;
import org.opensaml.saml2.core.impl.IssuerBuilder;

/**
 * Assertion test helper class.
 * @author mhorst
 *
 */
public class AssertionTestHelper {
	
    /**
     * Returns basic assertion for testing purposes.
     * @return basic assertion
     */
    public static Assertion prepareBasicAssertion() {
    	AssertionBuilder assertionBuilder = new AssertionBuilder();
    	Assertion assertion = assertionBuilder.buildObject();
    	assertion.setID("testAssertion-" + System.currentTimeMillis());
    	IssuerBuilder issuerBuilder = new IssuerBuilder();
    	Issuer issuer = issuerBuilder.buildObject();
    	issuer.setValue("YAAS");
    	assertion.setIssuer(issuer);
    	return assertion;
    }
    
    /**
     * Returns basic assertion for testing purposes.
     * @param issuerAddress
     * @return basic assertion
     */
    public static Assertion prepareBasicAssertion(String issuerAddress) {
    	AssertionBuilder assertionBuilder = new AssertionBuilder();
    	Assertion assertion = assertionBuilder.buildObject();
    	assertion.setID("testAssertion-" + System.currentTimeMillis());
    	IssuerBuilder issuerBuilder = new IssuerBuilder();
    	Issuer issuer = issuerBuilder.buildObject();
    	issuer.setValue(issuerAddress);
    	assertion.setIssuer(issuer);
    	return assertion;
    }
}