com.netflix.config.sources.JDBCConfigurationSourceTest.java Source code

Introduction

Here is the source code for com.netflix.config.sources.JDBCConfigurationSourceTest.java

Source

/**
 * Copyright 2014 Netflix, Inc.
 *
 * Licensed 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 com.netflix.config.sources;

import static org.junit.Assert.assertEquals;

import java.io.File;
import java.sql.Connection;
import java.sql.Statement;

import javax.sql.DataSource;

import org.apache.commons.io.FileUtils;
import org.apache.derby.jdbc.EmbeddedDataSource40;
import org.junit.Test;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.netflix.config.DynamicConfiguration;
import com.netflix.config.DynamicPropertyFactory;
import com.netflix.config.DynamicStringProperty;
import com.netflix.config.FixedDelayPollingScheduler;
import com.netflix.config.sources.JDBCConfigurationSource;

/**
 * Unit tests for the JDBCConfigurationSource class
 * This class uses the Derby embedded database.
 * In real life one can replace this with RDBMS products like Oracle, MySQL etc.
 * 
 * @author stonse
 *
 */
public class JDBCConfigurationSourceTest {

    private static Logger log = LoggerFactory.getLogger(JDBCConfigurationSourceTest.class);

    /**
     * Set up code for the test
     * @param dbname
     * @return
     * @throws Throwable
     */
    javax.sql.DataSource createDataConfigSource(String dbname) throws Throwable {
        EmbeddedDataSource40 ds = new EmbeddedDataSource40();
        ds.setDatabaseName(dbname);
        ds.setCreateDatabase(dbname);
        ds.setCreateDatabase("create");
        Connection con = null;
        try {
            con = ds.getConnection();

            // Creating a database table

            Statement sta = null;
            try {
                sta = con.createStatement();
                sta.executeUpdate("DROP TABLE MySiteProperties");
                log.info("Table Dropped");
            } catch (Exception e) {
                log.info("Table did not exist.");
            } finally {
                if (sta != null) {
                    sta.close();
                }
            }
            try {
                sta = con.createStatement();
                sta.executeUpdate("CREATE TABLE MySiteProperties (property_key VARCHAR(20),"
                        + " property_value VARCHAR(100))");
                log.info("Table created.");
                sta.close();
            } finally {
                if (sta != null) {
                    sta.close();
                }
            }
            try {
                sta = con.createStatement();

                sta.executeUpdate("insert into MySiteProperties values ('prop1','value1')");
                log.info("Properties for testing inserted");
                sta.close();
            } finally {
                if (sta != null) {
                    sta.close();
                }
            }

        } catch (Exception e) {
            System.err.println("Exception: " + e.getMessage());
        } finally {
            if (con != null) {
                con.close();
            }
        }

        return ds;
    }

    @Test
    public void testSimpleInMemoryJDBCDynamicPropertySource() throws Throwable {
        final String dbName = "MySiteConfiguration";
        DataSource ds = createDataConfigSource(dbName);

        try {
            JDBCConfigurationSource source = new JDBCConfigurationSource(ds,
                    "select distinct property_key, property_value from MySiteProperties", "property_key",
                    "property_value");
            FixedDelayPollingScheduler scheduler = new FixedDelayPollingScheduler(0, 10, false);
            DynamicConfiguration configuration = new DynamicConfiguration(source, scheduler);
            DynamicPropertyFactory.initWithConfigurationSource(configuration);

            DynamicStringProperty defaultProp = DynamicPropertyFactory.getInstance()
                    .getStringProperty("this.prop.does.not.exist.use.default", "default");
            assertEquals("default", defaultProp.get());

            DynamicStringProperty prop1 = DynamicPropertyFactory.getInstance().getStringProperty("prop1",
                    "default");
            assertEquals("value1", prop1.get());
        } finally {
            //Clean up the data files generated by the embedded DB.
            FileUtils.deleteDirectory(new File(".", dbName));
        }

    }

}