/*
* Copyright 2015 Red Hat, Inc. and/or its affiliates.
*
* 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 org.dashbuilder.dataprovider.sql;
import java.sql.ResultSet;
import java.sql.ResultSetMetaData;
import java.sql.Types;
import java.util.List;
import org.dashbuilder.dataprovider.sql.JDBCUtils;
import org.dashbuilder.dataprovider.sql.model.Column;
import org.dashbuilder.dataset.ColumnType;
import org.dashbuilder.dataset.def.SQLDataSourceDef;
import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.Mock;
import org.mockito.runners.MockitoJUnitRunner;
import static org.junit.Assert.*;
import static org.mockito.Mockito.*;
@RunWith(MockitoJUnitRunner.class)
public class JDBCUtilsTest {
@Mock
ResultSet resultSet;
@Mock
ResultSetMetaData metaData;
@Before
public void setUp() throws Exception {
when(resultSet.getMetaData()).thenReturn(metaData);
}
@Test
public void testListDataSourceDefs() throws Exception {
List<SQLDataSourceDef> defList = JDBCUtils.listDatasourceDefs();
assertEquals(defList.size(), 0);
}
@Test
public void testFixSQLCase() throws Exception {
String sql = "SELECT \"ID\" FROM TABLE";
String fix = JDBCUtils.changeCaseExcludeQuotes(sql, false);
assertEquals(fix, "select \"ID\" from table");
}
@Test
public void testSupportedTypes() throws Exception {
when(metaData.getColumnCount()).thenReturn(35);
when(metaData.getColumnType(1)).thenReturn(Types.CHAR);
when(metaData.getColumnType(2)).thenReturn(Types.VARCHAR);
when(metaData.getColumnType(3)).thenReturn(Types.NCHAR);
when(metaData.getColumnType(4)).thenReturn(Types.NVARCHAR);
when(metaData.getColumnType(5)).thenReturn(Types.BIT);
when(metaData.getColumnType(6)).thenReturn(Types.BOOLEAN);
when(metaData.getColumnType(7)).thenReturn(Types.LONGVARCHAR);
when(metaData.getColumnType(8)).thenReturn(Types.LONGNVARCHAR);
when(metaData.getColumnType(9)).thenReturn(Types.TINYINT);
when(metaData.getColumnType(10)).thenReturn(Types.BIGINT);
when(metaData.getColumnType(11)).thenReturn(Types.INTEGER);
when(metaData.getColumnType(12)).thenReturn(Types.DECIMAL);
when(metaData.getColumnType(13)).thenReturn(Types.DOUBLE);
when(metaData.getColumnType(14)).thenReturn(Types.FLOAT);
when(metaData.getColumnType(15)).thenReturn(Types.NUMERIC);
when(metaData.getColumnType(16)).thenReturn(Types.REAL);
when(metaData.getColumnType(17)).thenReturn(Types.SMALLINT);
when(metaData.getColumnType(18)).thenReturn(Types.DATE);
when(metaData.getColumnType(19)).thenReturn(Types.TIME);
when(metaData.getColumnType(20)).thenReturn(Types.TIMESTAMP);
when(metaData.getColumnType(21)).thenReturn(Types.VARBINARY);
when(metaData.getColumnType(22)).thenReturn(Types.LONGVARBINARY);
when(metaData.getColumnType(23)).thenReturn(Types.NULL);
when(metaData.getColumnType(24)).thenReturn(Types.OTHER);
when(metaData.getColumnType(25)).thenReturn(Types.JAVA_OBJECT);
when(metaData.getColumnType(26)).thenReturn(Types.DISTINCT);
when(metaData.getColumnType(27)).thenReturn(Types.STRUCT);
when(metaData.getColumnType(28)).thenReturn(Types.ARRAY);
when(metaData.getColumnType(29)).thenReturn(Types.BLOB);
when(metaData.getColumnType(30)).thenReturn(Types.CLOB);
when(metaData.getColumnType(31)).thenReturn(Types.REF);
when(metaData.getColumnType(32)).thenReturn(Types.ROWID);
when(metaData.getColumnType(33)).thenReturn(Types.SQLXML);
when(metaData.getColumnType(34)).thenReturn(Types.DATALINK);
List<Column> columns = JDBCUtils.getColumns(resultSet, null);
assertEquals(columns.size(), 20);
assertEquals(columns.get(0).getType(), ColumnType.LABEL);
assertEquals(columns.get(1).getType(), ColumnType.LABEL);
assertEquals(columns.get(2).getType(), ColumnType.LABEL);
assertEquals(columns.get(4).getType(), ColumnType.LABEL);
assertEquals(columns.get(5).getType(), ColumnType.LABEL);
assertEquals(columns.get(6).getType(), ColumnType.TEXT);
assertEquals(columns.get(7).getType(), ColumnType.TEXT);
assertEquals(columns.get(8).getType(), ColumnType.NUMBER);
assertEquals(columns.get(9).getType(), ColumnType.NUMBER);
assertEquals(columns.get(10).getType(), ColumnType.NUMBER);
assertEquals(columns.get(11).getType(), ColumnType.NUMBER);
assertEquals(columns.get(12).getType(), ColumnType.NUMBER);
assertEquals(columns.get(13).getType(), ColumnType.NUMBER);
assertEquals(columns.get(14).getType(), ColumnType.NUMBER);
assertEquals(columns.get(15).getType(), ColumnType.NUMBER);
assertEquals(columns.get(16).getType(), ColumnType.NUMBER);
assertEquals(columns.get(17).getType(), ColumnType.DATE);
assertEquals(columns.get(18).getType(), ColumnType.DATE);
assertEquals(columns.get(19).getType(), ColumnType.DATE);
}
}