diff tests/Test_PSgeneratedkeys.java @ 0:a5a898f6886c

Copy of MonetDB java directory changeset e6e32756ad31.
author Sjoerd Mullender <sjoerd@acm.org>
date Wed, 21 Sep 2016 09:34:48 +0200 (2016-09-21)
parents
children 04fbf3655452
line wrap: on
line diff
new file mode 100644
--- /dev/null
+++ b/tests/Test_PSgeneratedkeys.java
@@ -0,0 +1,71 @@
+/*
+ * This Source Code Form is subject to the terms of the Mozilla Public
+ * License, v. 2.0.  If a copy of the MPL was not distributed with this
+ * file, You can obtain one at http://mozilla.org/MPL/2.0/.
+ *
+ * Copyright 1997 - July 2008 CWI, August 2008 - 2016 MonetDB B.V.
+ */
+
+import java.sql.*;
+
+public class Test_PSgeneratedkeys {
+	public static void main(String[] args) throws Exception {
+		Class.forName("nl.cwi.monetdb.jdbc.MonetDriver");
+		Connection con = DriverManager.getConnection(args[0]);
+		con.setAutoCommit(false);
+		// >> false: auto commit was just switched off
+		System.out.println("0. false\t" + con.getAutoCommit());
+
+		try {
+			Statement stmt = con.createStatement();
+			stmt.executeUpdate(
+				"CREATE TABLE psgenkey (" +
+				"       id       serial," +
+				"       val      varchar(20)" +
+				")");
+			stmt.close();
+		} catch (SQLException e) {
+			System.out.println(e);
+			System.out.println("Creation of test table failed! :(");
+			System.out.println("ABORTING TEST!!!");
+			System.exit(-1);
+		}
+
+		try {
+			PreparedStatement pstmt = con.prepareStatement(
+				"INSERT INTO psgenkey (val) VALUES ('this is a test')",
+				Statement.RETURN_GENERATED_KEYS);
+
+			System.out.print("1. inserting a record...");
+			pstmt.executeUpdate();
+			System.out.println("success :)");
+
+			// now get the generated keys
+			System.out.print("2. getting generated keys...");
+			ResultSet keys = pstmt.getGeneratedKeys();
+			if (keys == null || !keys.next()) {
+				System.out.println("there are no keys! :(");
+				System.out.println("ABORTING TEST!!!");
+				System.exit(-1);
+			}
+
+			System.out.println("generated key index: " + keys.getInt(1));
+			while (keys.next()) {
+				System.out.println("generated key index: " + keys.getInt(1));
+			}
+
+			if (keys.getStatement() == null) {
+				System.out.println("ResultSet.getStatement() should never return null!");
+			}
+
+			keys.close();
+			pstmt.close();
+		} catch (SQLException e) {
+			System.out.println("FAILED :( "+ e.getMessage());
+			System.out.println("ABORTING TEST!!!");
+		}
+
+		con.rollback();
+		con.close();
+	}
+}