postgres/src/interfaces/jdbc/org/postgresql/test/jdbc2/UpdateableResultTest.java

package org.postgresql.test.jdbc2;

import java.sql.*;
import junit.framework.TestCase;

import org.postgresql.test.TestUtil;
/**
 * <p>Title: </p>
 * <p>Description: </p>
 * <p>Copyright: Copyright (c) 2001</p>
 * <p>Company: </p>
 * @author unascribed
 * @version 1.0
 */

public class UpdateableResultTest extends TestCase
{

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

	public void testUpdateable()
	{
		try
		{
			Connection con = TestUtil.openDB();
			TestUtil.createTable(con, "updateable", "id int primary key, name text, notselected text");
			TestUtil.createTable(con, "second", "id1 int primary key, name1 text");

			// put some dummy data into second
			Statement st2 = con.createStatement();
			st2.execute( "insert into second values (1,'anyvalue' )");
			st2.close();

			Statement st = con.createStatement( ResultSet.TYPE_SCROLL_INSENSITIVE, ResultSet.CONCUR_UPDATABLE );
			ResultSet rs = st.executeQuery( "select * from updateable");
			assertNotNull( rs );
			rs.moveToInsertRow();
			rs.updateInt( 1, 1 );
			rs.updateString( 2, "jake" );
			rs.updateString( 3, "avalue" );
			rs.insertRow();
			rs.first();

			rs.updateInt( "id", 2 );
			rs.updateString( "name", "dave" );
			rs.updateRow();

			assertTrue( rs.getInt("id") == 2 );
			assertTrue( rs.getString("name").equals("dave"));
			assertTrue( rs.getString("notselected").equals("avalue") );

			rs.deleteRow();
			rs.moveToInsertRow();
			rs.updateInt("id", 3);
			rs.updateString("name", "paul");

			rs.insertRow();
			rs.refreshRow();
			assertTrue( rs.getInt("id") == 3 );
			assertTrue( rs.getString("name").equals("paul"));
			assertTrue( rs.getString("notselected") == null );


			rs.close();

			rs = st.executeQuery("select id1, id, name, name1 from updateable, second" );
			try
			{
				while ( rs.next() )
				{
					rs.updateInt( "id", 2 );
					rs.updateString( "name", "dave" );
					rs.updateRow();
				}


				assertTrue( "should not get here, update should fail", false );
			}
			catch (SQLException ex)
			{}

			try
			{
				rs = st.executeQuery("select oid,* from updateable");
				if ( rs.first() )
				{
					rs.updateInt( "id", 3 );
					rs.updateString( "name", "dave3");
					rs.updateRow();
					assertTrue(rs.getInt("id") == 3 );
					assertTrue(rs.getString("name").equals("dave3"));

					rs.moveToInsertRow();
					rs.updateInt( "id", 4 );
					rs.updateString( "name", "dave4" );

					rs.insertRow();
					rs.updateInt("id", 5 );
					rs.updateString( "name", "dave5" );
					rs.insertRow();

					rs.moveToCurrentRow();
					assertTrue(rs.getInt("id") == 3 );
					assertTrue(rs.getString("name").equals("dave3"));

					assertTrue( rs.next() );
					assertTrue(rs.getInt("id") == 4 );
					assertTrue(rs.getString("name").equals("dave4"));

					assertTrue( rs.next() );
					assertTrue(rs.getInt("id") == 5 );
					assertTrue(rs.getString("name").equals("dave5"));

				}
			}
			catch (SQLException ex)
			{
				fail(ex.getMessage());
			}

			st.close();

			TestUtil.dropTable( con, "updateable" );
			TestUtil.dropTable( con, "second" );
			TestUtil.closeDB( con );
		}
		catch (Exception ex)
		{
			fail(ex.getMessage());
		}
	}


}