org.apache.lens.server.user.TestUserConfigLoader.java Source code

Introduction

Here is the source code for org.apache.lens.server.user.TestUserConfigLoader.java

Source

/**
 * 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.
 */
package org.apache.lens.server.user;

import java.io.PrintWriter;
import java.sql.SQLException;
import java.util.HashMap;

import org.apache.lens.server.LensServerConf;
import org.apache.lens.server.api.LensConfConstants;
import org.apache.lens.server.api.error.LensException;
import org.apache.lens.server.api.user.UserConfigLoader;
import org.apache.lens.server.util.UtilityMethods;

import org.apache.commons.dbcp.BasicDataSource;
import org.apache.hadoop.hive.conf.HiveConf;

import org.hsqldb.server.Server;
import org.testng.Assert;
import org.testng.annotations.*;

import liquibase.Liquibase;
import liquibase.database.jvm.HsqlConnection;
import liquibase.exception.LiquibaseException;
import liquibase.resource.FileSystemResourceAccessor;

/**
 * The Class TestUserConfigLoader.
 */
@Test(groups = "unit-test")
public class TestUserConfigLoader {

    /** The conf. */
    private HiveConf conf;

    /**
     * Inits the.
     */
    @BeforeClass(alwaysRun = true)
    public void init() {
        conf = new HiveConf(LensServerConf.getHiveConf());
    }

    /**
     * Reset factory.
     */
    @AfterClass
    public void resetFactory() {
        init();
        UserConfigLoaderFactory.init(conf);
    }

    /**
     * Test fixed.
     *
     * @throws LensException the lens exception
     */
    @Test
    public void testFixed() throws LensException {
        conf.addResource(TestUserConfigLoader.class.getResourceAsStream("/user/fixed.xml"));
        UserConfigLoaderFactory.init(conf);
        HashMap<String, String> expected = new HashMap<String, String>() {
            {
                put(LensConfConstants.SESSION_CLUSTER_USER, "lensuser");
            }
        };
        Assert.assertEquals(UserConfigLoaderFactory.getUserConfig("user1"), expected);
    }

    /**
     * Test property based.
     *
     * @throws LensException the lens exception
     */
    @Test
    public void testPropertyBased() throws LensException {
        conf.addResource(TestUserConfigLoader.class.getResourceAsStream("/user/propertybased.xml"));
        conf.set(LensConfConstants.USER_RESOLVER_PROPERTYBASED_FILENAME,
                TestUserConfigLoader.class.getResource("/user/propertybased.data").getPath());
        UserConfigLoaderFactory.init(conf);
        Assert.assertEquals(UserConfigLoaderFactory.getUserConfig("user1"), new HashMap<String, String>() {
            {
                put(LensConfConstants.SESSION_CLUSTER_USER, "clusteruser1");
                put(LensConfConstants.MAPRED_JOB_QUEUE_NAME, "queue1");
            }
        });
        Assert.assertEquals(UserConfigLoaderFactory.getUserConfig("user2"), new HashMap<String, String>() {
            {
                put(LensConfConstants.SESSION_CLUSTER_USER, "clusteruser2");
                put(LensConfConstants.MAPRED_JOB_QUEUE_NAME, "queue2");
            }
        });
    }

    /**
     * Setup hsql db.
     *
     * @param dbName        the db name
     * @param path          the path
     * @param changeLogPath the change log path
     * @throws SQLException       the SQL exception
     * @throws LiquibaseException the liquibase exception
     */
    private void setupHsqlDb(String dbName, String path, String changeLogPath)
            throws SQLException, LiquibaseException {
        Server server = new Server();
        server.setLogWriter(new PrintWriter(System.out));
        server.setErrWriter(new PrintWriter(System.out));
        server.setSilent(true);
        server.setDatabaseName(0, dbName);
        server.setDatabasePath(0, "file:" + path);
        server.start();
        BasicDataSource ds = UtilityMethods.getDataSourceFromConf(conf);

        Liquibase liquibase = new Liquibase(UserConfigLoader.class.getResource(changeLogPath).getFile(),
                new FileSystemResourceAccessor(), new HsqlConnection(ds.getConnection()));
        liquibase.update("");
    }

    /**
     * Test database.
     *
     * @throws LensException      the lens exception
     * @throws SQLException       the SQL exception
     * @throws LiquibaseException the liquibase exception
     */
    @Test
    public void testDatabase() throws LensException, SQLException, LiquibaseException {
        String path = "target/queries.db";
        String dbName = "main";
        conf.addResource(TestUserConfigLoader.class.getResourceAsStream("/user/database.xml"));
        UserConfigLoaderFactory.init(conf);
        setupHsqlDb(dbName, path, "/user/db_changelog.xml");
        String[][] valuesToVerify = new String[][] { { "user1", "clusteruser1", "queue12" },
                { "user2", "clusteruser2", "queue12" }, { "user3", "clusteruser3", "queue34" },
                { "user4", "clusteruser4", "queue34" }, };
        for (final String[] sa : valuesToVerify) {
            Assert.assertEquals(UserConfigLoaderFactory.getUserConfig(sa[0]), new HashMap<String, String>() {
                {
                    put(LensConfConstants.SESSION_CLUSTER_USER, sa[1]);
                    put(LensConfConstants.MAPRED_JOB_QUEUE_NAME, sa[2]);
                }
            });
        }
    }

    /**
     * Test custom.
     *
     * @throws LensException the lens exception
     */
    @Test
    public void testCustom() throws LensException {
        conf.addResource(TestUserConfigLoader.class.getResourceAsStream("/user/custom.xml"));
        UserConfigLoaderFactory.init(conf);
        Assert.assertEquals(UserConfigLoaderFactory.getUserConfig("user1"), FooBarConfigLoader.CONST_HASH_MAP);
    }
}