TokenHandlerTest.java

/*
 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.
 *
 * Copyright © 2011 ForgeRock AS. All rights reserved.
 *
 * The contents of this file are subject to the terms
 * of the Common Development and Distribution License
 * (the License). You may not use this file except in
 * compliance with the License.
 *
 * You can obtain a copy of the License at
 * http://forgerock.org/license/CDDLv1.0.html
 * See the License for the specific language governing
 * permission and limitations under the License.
 *
 * When distributing Covered Code, include this CDDL
 * Header Notice in each file and include the License file
 * at http://forgerock.org/license/CDDLv1.0.html
 * If applicable, add the following below the CDDL Header,
 * with the fields enclosed by brackets [] replaced by
 * your own identifying information:
 * "Portions Copyrighted [year] [name of copyright owner]"
 */
package org.forgerock.openidm.repo.orientdb.impl.query;

import java.util.Arrays;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import org.forgerock.json.resource.BadRequestException;

import org.testng.annotations.Test;

import static org.testng.Assert.assertEquals;

public class TokenHandlerTest {

    TokenHandler tokenHandler = null;
    
    @Test
    public void initTokenHandler() {
        tokenHandler = new TokenHandler();
    }
    
    @Test(dependsOnMethods = {"initTokenHandler"})
    public void replaceTokensWithValues() throws BadRequestException {
        String queryString = "select ${unquoted:_fields} from ${unquoted:_resource} where firstname = ${firstname} and lastname like '${unquoted:lastname}%'";
        Map params = new HashMap();
        params.put("_fields", "*");
        params.put("_resource", "managed/user");
        params.put("firstname", "John");
        params.put("lastname", "D");
        String result = tokenHandler.replaceTokensWithValues(queryString, params);
        assertEquals(result, "select * from managed/user where firstname = 'John' and lastname like 'D%'");
    }
    
    // Disabled as CREST 2.x supports param values only
    //@Test(dependsOnMethods = {"initTokenHandler"})
    public void replaceTokensWithListValues() throws BadRequestException {
        String queryString = "select ${unquoted:_fields} from ${unquoted:_resource} where firstname = ${firstname} and lastname like '${unquoted:lastname}%'";

        List fieldList = Arrays.asList(new String[] {"firstname", "lastname", "email"});
        
        Map params = new HashMap();
        params.put("_fields", fieldList);
        params.put("_resource", "managed/user");
        params.put("firstname", "John");
        params.put("lastname", "D");
        String result = tokenHandler.replaceTokensWithValues(queryString, params);
        assertEquals(result, "select firstname,lastname,email from managed/user where firstname = 'John' and lastname like 'D%'");
    }

    @Test(dependsOnMethods = {"initTokenHandler"})
    public void replaceTokenWithDotNotationAbsolute() throws BadRequestException {
        String queryString = "select ${dotnotation:jsonpath} from ${unquoted:_resource} where firstname = ${firstname} and lastname like '${unquoted:lastname}%'";
        Map params = new HashMap();
        params.put("jsonpath", "/sunset/date");
        params.put("_resource", "managed/user");
        params.put("firstname", "John");
        params.put("lastname", "D");
        String result = tokenHandler.replaceTokensWithValues(queryString, params);
        assertEquals(result, "select sunset.date from managed/user where firstname = 'John' and lastname like 'D%'");
    }

    @Test(dependsOnMethods = {"initTokenHandler"})
    public void replaceTokenWithDotNotationRelative() throws BadRequestException {
        String queryString = "select ${dotnotation:jsonpath} from ${unquoted:_resource} where firstname = ${firstname} and lastname like '${unquoted:lastname}%'";
        Map params = new HashMap();
        params.put("jsonpath", "sunset/date");
        params.put("_resource", "managed/user");
        params.put("firstname", "John");
        params.put("lastname", "D");
        String result = tokenHandler.replaceTokensWithValues(queryString, params);
        assertEquals(result, "select sunset.date from managed/user where firstname = 'John' and lastname like 'D%'");
    }

    @Test(dependsOnMethods = {"initTokenHandler"}, expectedExceptions = BadRequestException.class )
    public void valueReplaceMissingToken() throws BadRequestException {
        String queryString = "select ${unquoted:_fields} from ${unquoted:_resource} where firstname = ${firstname} and lastname like '${unquoted:lastname}%'";
        Map params = new HashMap();
        params.put("_fields", "*");
        params.put("_resource", "managed/user");
        // don't define firstname, should fail
        params.put("lastname", "D");
        String result = tokenHandler.replaceTokensWithValues(queryString, params);
        assertEquals(result, "select * from managed/user where firstname = 'John' and lastname like 'D%'");
    }
    
    @Test(dependsOnMethods = {"initTokenHandler"})
    public void replaceTokensWithOrientToken() throws PrepareNotSupported {
        String queryString = "select ${unquoted:_fields} from ${unquoted:_resource} where firstname = ${firstname} and lastname like ${lastname}";
        String result = tokenHandler.replaceTokensWithOrientToken(queryString);
        assertEquals(result, "select :_fields from :_resource where firstname = :firstname and lastname like :lastname");
    }
}