ldap/adapters/OAuthAdapterTest.java

	OAuthAdapterTest.java revision c951f676fd0230347f0e66f8c6d090d0d46ae650
1746e8b8399da2a7a8da4aace186f66055ccfec1Jakub Hrozek/**
1746e8b8399da2a7a8da4aace186f66055ccfec1Jakub Hrozek * Copyright 2013 ForgeRock, Inc.
1746e8b8399da2a7a8da4aace186f66055ccfec1Jakub Hrozek *
1746e8b8399da2a7a8da4aace186f66055ccfec1Jakub Hrozek * The contents of this file are subject to the terms of the Common Development and
1746e8b8399da2a7a8da4aace186f66055ccfec1Jakub Hrozek * Distribution License (the License). You may not use this file except in compliance with the
1746e8b8399da2a7a8da4aace186f66055ccfec1Jakub Hrozek * License.
1746e8b8399da2a7a8da4aace186f66055ccfec1Jakub Hrozek *
1746e8b8399da2a7a8da4aace186f66055ccfec1Jakub Hrozek * You can obtain a copy of the License at legal/CDDLv1.0.txt. See the License for the
1746e8b8399da2a7a8da4aace186f66055ccfec1Jakub Hrozek * specific language governing permission and limitations under the License.
1746e8b8399da2a7a8da4aace186f66055ccfec1Jakub Hrozek *
1746e8b8399da2a7a8da4aace186f66055ccfec1Jakub Hrozek * When distributing Covered Software, include this CDDL Header Notice in each file and include
1746e8b8399da2a7a8da4aace186f66055ccfec1Jakub Hrozek * the License file at legal/CDDLv1.0.txt. If applicable, add the following below the CDDL
1746e8b8399da2a7a8da4aace186f66055ccfec1Jakub Hrozek * Header, with the fields enclosed by brackets [] replaced by your own identifying
1746e8b8399da2a7a8da4aace186f66055ccfec1Jakub Hrozek * information: "Portions copyright [year] [name of copyright owner]".
1746e8b8399da2a7a8da4aace186f66055ccfec1Jakub Hrozek */
1746e8b8399da2a7a8da4aace186f66055ccfec1Jakub Hrozekpackage com.sun.identity.sm.ldap.adapters;
1746e8b8399da2a7a8da4aace186f66055ccfec1Jakub Hrozek
1746e8b8399da2a7a8da4aace186f66055ccfec1Jakub Hrozekimport com.sun.identity.shared.debug.Debug;
1746e8b8399da2a7a8da4aace186f66055ccfec1Jakub Hrozekimport com.sun.identity.sm.ldap.api.TokenType;
1746e8b8399da2a7a8da4aace186f66055ccfec1Jakub Hrozekimport com.sun.identity.sm.ldap.api.fields.OAuthTokenField;
1746e8b8399da2a7a8da4aace186f66055ccfec1Jakub Hrozekimport com.sun.identity.sm.ldap.api.tokens.Token;
1746e8b8399da2a7a8da4aace186f66055ccfec1Jakub Hrozekimport com.sun.identity.sm.ldap.api.tokens.TokenIdFactory;
1746e8b8399da2a7a8da4aace186f66055ccfec1Jakub Hrozekimport com.sun.identity.sm.ldap.utils.JSONSerialisation;
aea1d5c0ca9bb1470759b024c8b97b6c1f577193Pavel Březinaimport com.sun.identity.sm.ldap.utils.KeyConversion;
1746e8b8399da2a7a8da4aace186f66055ccfec1Jakub Hrozekimport org.forgerock.json.fluent.JsonValue;
1746e8b8399da2a7a8da4aace186f66055ccfec1Jakub Hrozekimport org.testng.annotations.Test;
1746e8b8399da2a7a8da4aace186f66055ccfec1Jakub Hrozek
1746e8b8399da2a7a8da4aace186f66055ccfec1Jakub Hrozekimport java.util.ArrayList;
1746e8b8399da2a7a8da4aace186f66055ccfec1Jakub Hrozekimport java.util.Arrays;
aea1d5c0ca9bb1470759b024c8b97b6c1f577193Pavel Březinaimport java.util.Calendar;
1746e8b8399da2a7a8da4aace186f66055ccfec1Jakub Hrozekimport java.util.HashMap;
1746e8b8399da2a7a8da4aace186f66055ccfec1Jakub Hrozekimport java.util.HashSet;
1746e8b8399da2a7a8da4aace186f66055ccfec1Jakub Hrozekimport java.util.List;
1746e8b8399da2a7a8da4aace186f66055ccfec1Jakub Hrozekimport java.util.Map;
1746e8b8399da2a7a8da4aace186f66055ccfec1Jakub Hrozekimport java.util.Set;
1746e8b8399da2a7a8da4aace186f66055ccfec1Jakub Hrozekimport java.util.TimeZone;
1746e8b8399da2a7a8da4aace186f66055ccfec1Jakub Hrozek
import static org.mockito.BDDMockito.given;
import static org.mockito.BDDMockito.mock;
import static org.testng.Assert.assertEquals;
import static org.testng.Assert.assertNotNull;
import static org.testng.AssertJUnit.assertTrue;

/**
 * @author robert.wapshott@forgerock.com
 */
public class OAuthAdapterTest {
    @Test
    public void shouldSerialiseSimpleString() {
        // Given
        OAuthAdapter adapter = generateOAuthAdapter();

        Set<String> text = new HashSet<String>();
        text.add("badger");
        OAuthTokenField field = OAuthTokenField.PARENT;

        Map<String, Object> values = new HashMap<String, Object>();
        values.put(field.getOAuthField(), text);
        JsonValue jsonValue = makeDefaultJsonValue(values);

        // When
        Token result = adapter.toToken(jsonValue);

        // Then
        assert(result.getValue(field.getField()).toString().contains("badger"));
    }

    @Test
    public void shouldSerialiseCollectionOfStrings() {
        // Given
        OAuthAdapter adapter = generateOAuthAdapter();

        String one = "badger";
        String two = "weasel";
        String three = "ferret";

        OAuthTokenField field = OAuthTokenField.SCOPE;

        Map<String, Object> values = new HashMap<String, Object>();
        values.put(field.getOAuthField(), Arrays.asList(new String[]{one, two, three}));
        JsonValue jsonValue = makeDefaultJsonValue(values);

        // When
        String result = adapter.toToken(jsonValue).getValue(field.getField());

        // Then
        assertTrue(result.contains(one));
        assertTrue(result.contains(two));
        assertTrue(result.contains(three));
    }

    @Test (expectedExceptions = IllegalStateException.class)
    public void shouldNotAllowASingleDate() {
        // Given
        OAuthAdapter adapter = generateOAuthAdapter();

        String text = "1234567890";
        OAuthTokenField field = OAuthTokenField.EXPIRY_TIME;

        Map<String, Object> values = new HashMap<String, Object>();
        values.put(field.getOAuthField(), text);
        JsonValue jsonValue = makeDefaultJsonValue(values);

        // When
        adapter.toToken(jsonValue);
    }

    @Test
    public void shouldSerialiseACollectionOfTimestamps() {
        // Given
        OAuthAdapter adapter = generateOAuthAdapter();

        String timestamp = "1370425721197";
        OAuthTokenField field = OAuthTokenField.EXPIRY_TIME;

        Map<String, Object> values = new HashMap<String, Object>();
        values.put(field.getOAuthField(), Arrays.asList(new String[]{timestamp}));
        JsonValue jsonValue = makeDefaultJsonValue(values);

        // When
        Calendar result = adapter.toToken(jsonValue).getExpiryTimestamp();

        // Then
        // The result timezone is set to local time.
        // Convert this timestamp to a known timezone.
        Calendar c = Calendar.getInstance(TimeZone.getTimeZone("Europe/London"));
        c.setTimeInMillis(result.getTimeInMillis());

        // Wed, 05 Jun 2013 10:48:41 BST
        assertEquals(c.get(Calendar.MONTH), 5);
        assertEquals(c.get(Calendar.YEAR), 2013);
        assertEquals(c.get(Calendar.HOUR_OF_DAY), 10);
        assertEquals(c.get(Calendar.MINUTE), 48);
    }

    @Test
    public void shouldDeserialiseSerialisedToken() {
        // Given
        String[] id = {"badger"};
        List<String> list = new ArrayList<String>(Arrays.asList(id));
        OAuthTokenField field = OAuthTokenField.ID;

        JSONSerialisation serialisation = new JSONSerialisation(mock(Debug.class));
        OAuthAdapter adapter = generateOAuthAdapter();

        // Populate a map for serialisation.
        Map<String, Object> values = new HashMap<String, Object>();
        values.put(field.getOAuthField(), list);
        //Map<String, Object> map = new HashMap<String, Object>();
        //map.put(OAuthAdapter.VALUE, values);
        String serialisedObject = serialisation.serialise(values);

        Token token = new Token(id[0], TokenType.OAUTH);
        // Set the serialised binary data
        token.setBlob(serialisedObject.getBytes());

        // When
        JsonValue result = adapter.fromToken(token);

        // Then
        assertNotNull(result);
        assert(result.asMap().get(field.getOAuthField()).toString().contains(id[0]));
    }

    @Test (expectedExceptions = IllegalArgumentException.class)
    public void shouldVerifyThatObjectIsAMap() {
        // Given
        OAuthAdapter adapter = generateOAuthAdapter();

        JsonValue mockValue = mock(JsonValue.class);
        given(mockValue.getObject()).willReturn("badger");

        // When/Then
        adapter.toToken(mockValue);
    }

    @Test (expectedExceptions = IllegalArgumentException.class)
    public void shouldNotDeserialiseATokenWhichDoesntContainAMap() {
        // Given
        JSONSerialisation serialisation = new JSONSerialisation(mock(Debug.class));
        OAuthAdapter adapter = generateOAuthAdapter();

        Token token = new Token("", TokenType.OAUTH);
        token.setBlob(serialisation.serialise("badger").getBytes());

        // When/Then
        adapter.fromToken(token);
    }


    /**
     * @return Makes a standard OAuthAdapter with real dependencies.
     */
    private OAuthAdapter generateOAuthAdapter() {
        JSONSerialisation serialisation = new JSONSerialisation(mock(Debug.class));
        KeyConversion keyConversion = new KeyConversion();
        OAuthValues oAuthValues = new OAuthValues();
        return new OAuthAdapter(new TokenIdFactory(keyConversion), serialisation, oAuthValues);
    }

    /**
     * @param values Generate a JsonValue based on the values in this map.
     * @return A non null JsonValue.
     */
    private JsonValue makeDefaultJsonValue(Map<String, Object> values) {
        return new JsonValue(values);
    }
}