package test;

import java.sql.*;

import org.junit.*;

import static org.junit.Assert.*;

public class RSMetaDataTest

{

private Connection conn;

private Statement stat;

private ResultSetMetaData meta;

@BeforeClass public static void forName() throws Exception {

Class.forName("org.sqlite.JDBC");

}

@Before public void connect() throws Exception {

conn = DriverManager.getConnection("jdbc:sqlite:");

stat = conn.createStatement();

stat.executeUpdate(

"create table People (pid integer primary key autoincrement, "

+ " firstname string, surname string, dob date);");

stat.executeUpdate(

"insert into people values (null, 'Mohandas', 'Gandhi', "

+ " '1869-10-02');");

meta = stat.executeQuery(

"select pid, firstname, surname from people;").getMetaData();

}

@After public void close() throws SQLException {

stat.executeUpdate("drop table people;");

stat.close();

conn.close();

}

@Test public void catalogName() throws SQLException {

assertEquals(meta.getCatalogName(1), "People");

}

@Test public void columns() throws SQLException {

assertEquals(meta.getColumnCount(), 3);

assertEquals(meta.getColumnName(1), "pid");

assertEquals(meta.getColumnName(2), "firstname");

assertEquals(meta.getColumnName(3), "surname");

assertEquals(meta.getColumnType(1), Types.INTEGER);

assertEquals(meta.getColumnType(2), Types.VARCHAR);

assertEquals(meta.getColumnType(3), Types.VARCHAR);

assertEquals(meta.getColumnTypeName(1), "integer");

assertEquals(meta.getColumnTypeName(2), "text");

assertEquals(meta.getColumnTypeName(3), "text");

assertTrue(meta.isAutoIncrement(1));

assertFalse(meta.isAutoIncrement(2));

assertFalse(meta.isAutoIncrement(3));

assertEquals(meta.isNullable(1), meta.columnNoNulls);

assertEquals(meta.isNullable(2), meta.columnNullable);

assertEquals(meta.isNullable(3), meta.columnNullable);

}

@Test public void differentRS() throws SQLException {

meta = stat.executeQuery("select * from people;").getMetaData();

assertEquals(meta.getColumnCount(), 4);

assertEquals(meta.getColumnName(1), "pid");

assertEquals(meta.getColumnName(2), "firstname");

assertEquals(meta.getColumnName(3), "surname");

assertEquals(meta.getColumnName(4), "dob");

}

@Test public void nullable() throws SQLException {

meta = stat.executeQuery("select null;").getMetaData();

assertEquals(meta.isNullable(1), ResultSetMetaData.columnNullable);

}

@Test(expected= SQLException.class)

public void badCatalogIndex() throws SQLException { meta.getCatalogName(4);}

@Test(expected= SQLException.class)

public void badColumnIndex() throws SQLException { meta.getColumnName(4); }

}