/*-------------------------------------------------------------------------
*
* Copyright (c) 2004-2008, PostgreSQL Global Development Group
*
* IDENTIFICATION
*   $PostgreSQL: pgjdbc/org/postgresql/test/jdbc2/DateTest.java,v 1.14 2008/01/08 06:56:31 jurka Exp $
*
*-------------------------------------------------------------------------
*/
package org.postgresql.test.jdbc2;

import org.postgresql.test.TestUtil;
import junit.framework.TestCase;
import java.sql.*;

/*
 * Some simple tests based on problems reported by users. Hopefully these will
 * help prevent previous problems from re-occuring ;-)
 *
 */
public class DateTest extends TestCase
{

    private Connection con;

    public DateTest(String name)
    {
        super(name);
    }

    protected void setUp() throws Exception
    {
        con = TestUtil.openDB();
        TestUtil.createTable(con, "testdate", "dt date");
    }

    protected void tearDown() throws Exception
    {
        TestUtil.dropTable(con, "testdate");
        TestUtil.closeDB(con);
    }

    /*
     * Tests the time methods in ResultSet
     */
    public void testGetDate() throws SQLException
    {
        Statement stmt = con.createStatement();

        assertEquals(1, stmt.executeUpdate(TestUtil.insertSQL("testdate", "'1950-02-07'")));
        assertEquals(1, stmt.executeUpdate(TestUtil.insertSQL("testdate", "'1970-06-02'")));
        assertEquals(1, stmt.executeUpdate(TestUtil.insertSQL("testdate", "'1999-08-11'")));
        assertEquals(1, stmt.executeUpdate(TestUtil.insertSQL("testdate", "'2001-02-13'")));
        assertEquals(1, stmt.executeUpdate(TestUtil.insertSQL("testdate", "'1950-04-02'")));
        assertEquals(1, stmt.executeUpdate(TestUtil.insertSQL("testdate", "'1970-11-30'")));
        assertEquals(1, stmt.executeUpdate(TestUtil.insertSQL("testdate", "'1988-01-01'")));
        assertEquals(1, stmt.executeUpdate(TestUtil.insertSQL("testdate", "'2003-07-09'")));
        assertEquals(1, stmt.executeUpdate(TestUtil.insertSQL("testdate", "'1934-02-28'")));
        assertEquals(1, stmt.executeUpdate(TestUtil.insertSQL("testdate", "'1969-04-03'")));
        assertEquals(1, stmt.executeUpdate(TestUtil.insertSQL("testdate", "'1982-08-03'")));
        assertEquals(1, stmt.executeUpdate(TestUtil.insertSQL("testdate", "'2012-03-15'")));
        assertEquals(1, stmt.executeUpdate(TestUtil.insertSQL("testdate", "'1912-05-01'")));
        assertEquals(1, stmt.executeUpdate(TestUtil.insertSQL("testdate", "'1971-12-15'")));
        assertEquals(1, stmt.executeUpdate(TestUtil.insertSQL("testdate", "'1984-12-03'")));
        assertEquals(1, stmt.executeUpdate(TestUtil.insertSQL("testdate", "'2000-01-01'")));
        assertEquals(1, stmt.executeUpdate(TestUtil.insertSQL("testdate", "'23456-01-01'")));
        assertEquals(1, stmt.executeUpdate(TestUtil.insertSQL("testdate", "'0101-01-01 BC'")));

        /* dateTest() contains all of the tests */
        dateTest();

        assertEquals(18, stmt.executeUpdate("DELETE FROM " + "testdate"));
        stmt.close();
    }

    /*
     * Tests the time methods in PreparedStatement
     */
    public void testSetDate() throws SQLException
    {
        Statement stmt = con.createStatement();
        PreparedStatement ps = con.prepareStatement(TestUtil.insertSQL("testdate", "?"));

        ps.setDate(1, makeDate(1950, 2, 7));
        assertEquals(1, ps.executeUpdate());

        ps.setDate(1, makeDate(1970, 6, 2));
        assertEquals(1, ps.executeUpdate());

        ps.setDate(1, makeDate(1999, 8, 11));
        assertEquals(1, ps.executeUpdate());

        ps.setDate(1, makeDate(2001, 2, 13));
        assertEquals(1, ps.executeUpdate());

        ps.setObject(1, java.sql.Timestamp.valueOf("1950-04-02 12:00:00"), java.sql.Types.DATE);
        assertEquals(1, ps.executeUpdate());

        ps.setObject(1, java.sql.Timestamp.valueOf("1970-11-30 3:00:00"), java.sql.Types.DATE);
        assertEquals(1, ps.executeUpdate());

        ps.setObject(1, java.sql.Timestamp.valueOf("1988-01-01 13:00:00"), java.sql.Types.DATE);
        assertEquals(1, ps.executeUpdate());

        ps.setObject(1, java.sql.Timestamp.valueOf("2003-07-09 12:00:00"), java.sql.Types.DATE);
        assertEquals(1, ps.executeUpdate());

        ps.setObject(1, "1934-02-28", java.sql.Types.DATE);
        assertEquals(1, ps.executeUpdate());

        ps.setObject(1, "1969-04-3", java.sql.Types.DATE);
        assertEquals(1, ps.executeUpdate());

        ps.setObject(1, "1982-08-03", java.sql.Types.DATE);
        assertEquals(1, ps.executeUpdate());

        ps.setObject(1, "2012-3-15", java.sql.Types.DATE);
        assertEquals(1, ps.executeUpdate());

        ps.setObject(1, java.sql.Date.valueOf("1912-5-1"), java.sql.Types.DATE);
        assertEquals(1, ps.executeUpdate());

        ps.setObject(1, java.sql.Date.valueOf("1971-12-15"), java.sql.Types.DATE);
        assertEquals(1, ps.executeUpdate());

        ps.setObject(1, java.sql.Date.valueOf("1984-12-03"), java.sql.Types.DATE);
        assertEquals(1, ps.executeUpdate());

        ps.setObject(1, java.sql.Date.valueOf("2000-1-1"), java.sql.Types.DATE);
        assertEquals(1, ps.executeUpdate());

        ps.setObject(1, java.sql.Date.valueOf("23456-1-1"), java.sql.Types.DATE);
        assertEquals(1, ps.executeUpdate());

        // We can't use valueOf on BC dates.
        ps.setObject(1, makeDate(-100,1,1));
        assertEquals(1, ps.executeUpdate());

        ps.close();

        dateTest();

        assertEquals(18, stmt.executeUpdate("DELETE FROM testdate"));
        stmt.close();
    }

    /*
     * Helper for the date tests. It tests what should be in the db
     */
    private void dateTest() throws SQLException
    {
        Statement st = con.createStatement();
        ResultSet rs;
        java.sql.Date d;

        rs = st.executeQuery(TestUtil.selectSQL("testdate", "dt"));
        assertNotNull(rs);

        assertTrue(rs.next());
        d = rs.getDate(1);
        assertNotNull(d);
        assertEquals(makeDate(1950, 2, 7), d);


        assertTrue(rs.next());
        d = rs.getDate(1);
        assertNotNull(d);
        assertEquals(makeDate(1970, 6, 2), d);

        assertTrue(rs.next());
        d = rs.getDate(1);
        assertNotNull(d);
        assertEquals(makeDate(1999, 8, 11), d);

        assertTrue(rs.next());
        d = rs.getDate(1);
        assertNotNull(d);
        assertEquals(makeDate(2001, 2, 13), d);

        assertTrue(rs.next());
        d = rs.getDate(1);
        assertNotNull(d);
        assertEquals(makeDate(1950, 4, 2), d);

        assertTrue(rs.next());
        d = rs.getDate(1);
        assertNotNull(d);
        assertEquals(makeDate(1970, 11, 30), d);

        assertTrue(rs.next());
        d = rs.getDate(1);
        assertNotNull(d);
        assertEquals(makeDate(1988, 1, 1), d);

        assertTrue(rs.next());
        d = rs.getDate(1);
        assertNotNull(d);
        assertEquals(makeDate(2003, 7, 9), d);

        assertTrue(rs.next());
        d = rs.getDate(1);
        assertNotNull(d);
        assertEquals(makeDate(1934, 2, 28), d);

        assertTrue(rs.next());
        d = rs.getDate(1);
        assertNotNull(d);
        assertEquals(makeDate(1969, 4, 3), d);

        assertTrue(rs.next());
        d = rs.getDate(1);
        assertNotNull(d);
        assertEquals(makeDate(1982, 8, 3), d);

        assertTrue(rs.next());
        d = rs.getDate(1);
        assertNotNull(d);
        assertEquals(makeDate(2012, 3, 15), d);

        assertTrue(rs.next());
        d = rs.getDate(1);
        assertNotNull(d);
        assertEquals(makeDate(1912, 5, 1), d);

        assertTrue(rs.next());
        d = rs.getDate(1);
        assertNotNull(d);
        assertEquals(makeDate(1971, 12, 15), d);

        assertTrue(rs.next());
        d = rs.getDate(1);
        assertNotNull(d);
        assertEquals(makeDate(1984, 12, 3), d);

        assertTrue(rs.next());
        d = rs.getDate(1);
        assertNotNull(d);
        assertEquals(makeDate(2000, 1, 1), d);

        assertTrue(rs.next());
        d = rs.getDate(1);
        assertNotNull(d);
        assertEquals(makeDate(23456, 1, 1), d);

        assertTrue(rs.next());
        d = rs.getDate(1);
        assertNotNull(d);
        assertEquals(makeDate(-100, 1, 1), d);

        assertTrue(!rs.next());

        rs.close();
        st.close();
    }

    private java.sql.Date makeDate(int y, int m, int d)
    {
        return new java.sql.Date(y - 1900, m - 1, d);
    }
}
