/*************************************************************************
 *                                                                       *
 *  EJBCA Community: The OpenSource Certificate Authority                *
 *                                                                       *
 *  This software is free software; you can redistribute it and/or       *
 *  modify it under the terms of the GNU Lesser General Public           *
 *  License as published by the Free Software Foundation; either         *
 *  version 2.1 of the License, or any later version.                    *
 *                                                                       *
 *  See terms of license at gnu.org.                                     *
 *                                                                       *
 *************************************************************************/
package org.ejbca.util.dn;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNull;

import java.util.HashMap;
import java.util.Map;

import javax.naming.ldap.Rdn;

import org.cesecore.certificates.util.DnComponents;
import org.junit.Before;
import org.junit.Test;

/** Tests for DistinguishedName class.
 * 
 * @version $Id: DistinguishedNameTest.java 23703 2016-06-21 09:26:03Z anatom $
 */
public class DistinguishedNameTest {
    
    private static final String DN = "cn=David Galichet,o=Fimasys,email=dgalichet@fimasys.fr," 
        + "g=M,email=david.galichet@fimasys.fr";
    private static final String OTHER_DN = "o=Linagora,email=dgalichet@linagora.fr,ou=Linagora Secu," 
        + "l=Paris,email=david.galichet@linagora.com,email=dgalichet@linagora.com";
    DistinguishedName dn = null;
    DistinguishedName otherDn = null;
  
    private static final String SUBJECT_ALT_NAME = "RFC822NAME=vkn@linagora.com,IPADDRESS=208.77.188.166";
    private static final String OTHER_SUBJECT_ALT_NAME = "RFC822NAME=linagora.mail@linagora.com,IPADDRESS=777.77.777.777,UNIFORMRESOURCEID=other.uri";
    DistinguishedName subjectAltName = null;
    DistinguishedName otherSubjectAltName = null;

    @Before
    public void setUp() throws Exception {
        otherDn = new DistinguishedName(OTHER_DN);
        otherSubjectAltName = new DistinguishedName(OTHER_SUBJECT_ALT_NAME);
    }

    @Test
    public void testGetRdn() throws Exception {
        dn = createNewDN();
        assertEquals(dn.getRdn("cn"), new Rdn("cn", "David Galichet"));
        assertEquals(dn.getRdn("email", 1), new Rdn("email", "dgalichet@fimasys.fr"));
        assertEquals(dn.getRdn("email", 0), new Rdn("email", "david.galichet@fimasys.fr"));
        assertNull(dn.getRdn("email", 2));
    }

    /**
     * Test of mergeDN method, of class DistinguishedName.
     * This version tests the merge without override.
     */
    @Test
    public void testMergeDnWithoutOverride() throws Exception {

        final String EXPECTED_DN = "cn=David Galichet,o=Fimasys,email=dgalichet@fimasys.fr,"
            + "g=M,email=david.galichet@fimasys.fr,ou=Linagora Secu,email=dgalichet@linagora.com,l=Paris";
        final String EXPECTED_DN_JDK8 = "cn=David Galichet,o=Fimasys,email=dgalichet@fimasys.fr,"
                + "g=M,email=david.galichet@fimasys.fr,ou=Linagora Secu,l=Paris,email=dgalichet@linagora.com";
        dn = createNewDN();
        DistinguishedName newDn = dn.mergeDN(otherDn, false, null);

        // This returns slightly different between JDK 7 and JDK 8
        String[] javaVersionElements = System.getProperty("java.version").split("\\.");
        int major = Integer.parseInt(javaVersionElements[1]);
        if (major > 7) {
            assertEquals(EXPECTED_DN_JDK8, newDn.toString());
        } else {
            assertEquals(EXPECTED_DN, newDn.toString());
        }        
    }

    /**
     * Test of mergeDN method, of class DistinguishedName.
     * This version tests the merge with override.
     */
    @Test
   public void testMergeDnWithOverride() throws Exception {

        final String EXPECTED_DN = "cn=David Galichet,o=Linagora,email=dgalichet@linagora.fr,"
            + "g=M,email=david.galichet@linagora.com,ou=Linagora Secu,email=dgalichet@linagora.com,l=Paris";
        final String EXPECTED_DN_JDK8 = "cn=David Galichet,o=Linagora,email=dgalichet@linagora.fr,"
                + "g=M,email=david.galichet@linagora.com,ou=Linagora Secu,l=Paris,email=dgalichet@linagora.com";

        dn = createNewDN();
        DistinguishedName newDn = dn.mergeDN(otherDn, true, null);

        // This returns slightly different between JDK 7 and JDK 8
        String[] javaVersionElements = System.getProperty("java.version").split("\\.");
        int major = Integer.parseInt(javaVersionElements[1]);
        if (major > 7) {
            assertEquals(EXPECTED_DN_JDK8, newDn.toString());
        } else {
            assertEquals(EXPECTED_DN, newDn.toString());
        }        
    }

    /**
     * Test of mergeDN method, of class DistinguishedName.
     * This version tests the merge without override.
     */
    @Test
    public void testMergeSubjectAltNameWithoutOverrideNotUsingEntityEmail() throws Exception {

        final String EXPECTED = "RFC822NAME=vkn@linagora.com,IPADDRESS=208.77.188.166,UNIFORMRESOURCEID=other.uri";
        subjectAltName = createNewSubjectAltName();
        Map<String, String> dnMap = new HashMap<String, String>();
        dnMap.put(DnComponents.RFC822NAME, "entitymail@linagora.com");
        DistinguishedName altName = subjectAltName.mergeDN(otherSubjectAltName, false, dnMap);

        assertEquals(3, altName.size());

        assertEquals(EXPECTED, altName.toString());
    }

    /**
     * Test of mergeDN method, of class DistinguishedName.
     * This version tests the merge without override.
     */
    @Test
    public void testMergeSubjectAltNameWithoutOverrideUsingEntityEmail() throws Exception {

    	final String EXPECTED = "RFC822NAME=vkn@linagora.com,IPADDRESS=208.77.188.166,UNIFORMRESOURCEID=other.uri";
        subjectAltName = createNewSubjectAltName();
        Map<String, String> dnMap = new HashMap<String, String>();
        dnMap.put(DnComponents.RFC822NAME, "entitymail@linagora.com");
        DistinguishedName altName = subjectAltName.mergeDN(otherSubjectAltName, false, dnMap);

        assertEquals(EXPECTED, altName.toString());
    }
    /**
     * Test of mergeDN method, of class DistinguishedName.
     * This version tests the merge with override.
     */
    @Test
    public void testMergeSubjectAltNameWithOverrideNotUsingEntityEmail() throws Exception {

    	final String EXPECTED = "RFC822NAME=linagora.mail@linagora.com,IPADDRESS=777.77.777.777,UNIFORMRESOURCEID=other.uri";
        subjectAltName = createNewSubjectAltName();
        Map<String, String> dnMap = new HashMap<String, String>();
        DistinguishedName altName = subjectAltName.mergeDN(otherSubjectAltName, true, dnMap);

        assertEquals(EXPECTED, altName.toString());
    }
    /**
     * Test of mergeDN method, of class DistinguishedName.
     * This version tests the merge with override.
     */
    @Test
    public void testMergeSubjectAltNameWithOverrideUsingEntityEmail() throws Exception {
        final String _OTHER_SUBJECT_ALT_NAME = "IPADDRESS=777.77.777.777,UNIFORMRESOURCEID=other.uri";

    	final String EXPECTED = "RFC822NAME=entitymail@linagora.com,IPADDRESS=777.77.777.777,UNIFORMRESOURCEID=other.uri";
        DistinguishedName san = new DistinguishedName(_OTHER_SUBJECT_ALT_NAME);
        subjectAltName = createNewSubjectAltName();
        Map<String, String> dnMap = new HashMap<String, String>();
        dnMap.put(DnComponents.RFC822NAME, "entitymail@linagora.com");
        DistinguishedName altName = subjectAltName.mergeDN(san, true, dnMap);

        assertEquals(EXPECTED, altName.toString());
    }

    private DistinguishedName createNewDN() throws Exception {
        return new DistinguishedName(DN);
    }

    private DistinguishedName createNewSubjectAltName() throws Exception {
        return new DistinguishedName(SUBJECT_ALT_NAME);
    }
}