HCQLDataSourceTest.java

/**
 * 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 com.datastax.drivers.jdbc.pool.cassandra.jdbc;

import static org.junit.Assert.*;
import static org.mockito.Mockito.mock;

import java.io.IOException;
import java.util.Hashtable;

import javax.naming.Context;
import javax.naming.InitialContext;
import javax.naming.Name;
import javax.naming.Reference;
import javax.naming.StringRefAddr;
import javax.sql.DataSource;

import org.junit.After;
import org.junit.Before;
import org.junit.Test;

import com.datastax.drivers.jdbc.pool.cassandra.BaseEmbededServerSetupTest;
import com.datastax.drivers.jdbc.pool.cassandra.connection.CassandraHostConfigurator;
import com.datastax.drivers.jdbc.pool.cassandra.service.FailoverPolicy;


public class HCQLDataSourceTest extends BaseEmbededServerSetupTest {

  private final static String cassandraUrl = "localhost:9170";


  private HCQLDataSource factory;

  @Before
  public void setupCase() throws Exception {
    factory = new HCQLDataSource();
  }

  @After
  public void teardownCase() throws IOException {
    factory = null;
  }
  
  @Test
  public void testSetFailoverPolicy() throws Exception {
    Reference resource = new Reference("HCQLDataSource");
    resource.add(new StringRefAddr("hosts", cassandraUrl));
    resource.add(new StringRefAddr("clusterName", clusterName));
    resource.add(new StringRefAddr("keyspace", "Keyspace1"));
    resource.add(new StringRefAddr("user", ""));
    resource.add(new StringRefAddr("password", ""));
    resource.add(new StringRefAddr("failoverPolicy", FailoverPolicy.FAIL_FAST.toString()));

    Name jndiName = mock(Name.class);
    Context context = new InitialContext();
    Hashtable<String, String> environment = new Hashtable<String, String>();

    CassandraHostConfigurator dataSource = (CassandraHostConfigurator) factory.getObjectInstance(resource, jndiName, context, environment);
    assertEquals(FailoverPolicy.FAIL_FAST, dataSource.getFailoverPolicy());
  }

  @Test
  public void getObjectInstance() throws Exception {
    Reference resource = new Reference("HCQLDataSource");
    resource.add(new StringRefAddr("hosts", cassandraUrl));
    resource.add(new StringRefAddr("clusterName", clusterName));
    resource.add(new StringRefAddr("keyspace", "Keyspace1"));
    resource.add(new StringRefAddr("user", ""));
    resource.add(new StringRefAddr("password", ""));

    Name jndiName = mock(Name.class);
    Context context = new InitialContext();
    Hashtable<String, String> environment = new Hashtable<String, String>();

    DataSource dataSource = (DataSource) factory.getObjectInstance(resource, jndiName, context, environment);

    assertNotNull(dataSource);
    assertNotNull(dataSource.getConnection());
  }
}