From ef1cdf683d3389c2802ae5e089da8fcf5c5cda01 Mon Sep 17 00:00:00 2001 From: hantmac Date: Fri, 26 Apr 2024 14:28:00 +0800 Subject: [PATCH 1/4] feat: add single insert into test --- .../com/databend/jdbc/TestBasicDriver.java | 37 +++++++++++++++++++ 1 file changed, 37 insertions(+) diff --git a/databend-jdbc/src/test/java/com/databend/jdbc/TestBasicDriver.java b/databend-jdbc/src/test/java/com/databend/jdbc/TestBasicDriver.java index 9cf737d2..711a39af 100644 --- a/databend-jdbc/src/test/java/com/databend/jdbc/TestBasicDriver.java +++ b/databend-jdbc/src/test/java/com/databend/jdbc/TestBasicDriver.java @@ -67,6 +67,43 @@ public void testBasic() } } + @Test + public void TestInsertInto() throws SQLException { + try (Connection connection = createConnection()) { + DatabendStatement statement = (DatabendStatement) connection.createStatement(); + statement.execute("CREATE TABLE IF NOT EXISTS test_basic_driver.target_table (\n" + + " ID INT,\n" + + " Name VARCHAR(50),\n" + + " Age INT,\n" + + " City VARCHAR(50)\n" + + ");"); + statement.execute("INSERT INTO test_basic_driver.target_table (ID, Name, Age, City)\n" + + "VALUES\n" + + " (1, 'Alice', 25, 'Toronto'),\n" + + " (2, 'Bob', 30, 'Vancouver'),\n" + + " (3, 'Carol', 28, 'Montreal');"); + statement.execute("CREATE TABLE IF NOT EXISTS test_basic_driver.source_table (\n" + + " ID INT,\n" + + " Name VARCHAR(50),\n" + + " Age INT,\n" + + " City VARCHAR(50)\n" + + ");"); + statement.execute("INSERT INTO test_basic_driver.source_table (ID, Name, Age, City)\n" + + "VALUES\n" + + " (1, 'David', 27, 'Calgary'),\n" + + " (2, 'Emma', 29, 'Ottawa'),\n" + + " (4, 'Frank', 32, 'Edmonton');"); + statement.execute("INSERT INTO test_basic_driver.target_table (ID, Name, Age, City)\n" + + "SELECT * FROM test_basic_driver.source_table;"); + ResultSet r = statement.getResultSet(); + r.next(); + Assert.assertEquals(3, statement.getUpdateCount()); + System.out.println(statement.getUpdateCount()); + } catch (SQLException throwables) { + throwables.printStackTrace(); + } + } + @Test public void TestMergeinto() throws SQLException { try (Connection connection = createConnection()) { From 146dac30e3ccc0ca8010b2c8b0f1119e3dcc46a9 Mon Sep 17 00:00:00 2001 From: hantmac Date: Fri, 26 Apr 2024 14:32:13 +0800 Subject: [PATCH 2/4] fix --- .../src/test/java/com/databend/jdbc/TestBasicDriver.java | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/databend-jdbc/src/test/java/com/databend/jdbc/TestBasicDriver.java b/databend-jdbc/src/test/java/com/databend/jdbc/TestBasicDriver.java index 31d3f3b8..41a17668 100644 --- a/databend-jdbc/src/test/java/com/databend/jdbc/TestBasicDriver.java +++ b/databend-jdbc/src/test/java/com/databend/jdbc/TestBasicDriver.java @@ -69,12 +69,10 @@ public void testBasic() } @Test -<<<<<< Date: Fri, 26 Apr 2024 14:40:14 +0800 Subject: [PATCH 3/4] fix test --- .../src/test/java/com/databend/jdbc/TestBasicDriver.java | 1 + 1 file changed, 1 insertion(+) diff --git a/databend-jdbc/src/test/java/com/databend/jdbc/TestBasicDriver.java b/databend-jdbc/src/test/java/com/databend/jdbc/TestBasicDriver.java index 41a17668..96668590 100644 --- a/databend-jdbc/src/test/java/com/databend/jdbc/TestBasicDriver.java +++ b/databend-jdbc/src/test/java/com/databend/jdbc/TestBasicDriver.java @@ -105,6 +105,7 @@ public void TestInsertInto() throws SQLException { } } + @Test public void testCreateUserFunction() throws SQLException { String s = "create or replace function add_plus(int,int)\n" + "returns int\n" + From 85469e7568a8da6566399f9a7d6c2f81840205e9 Mon Sep 17 00:00:00 2001 From: hantmac Date: Fri, 26 Apr 2024 14:49:07 +0800 Subject: [PATCH 4/4] fix tests --- .../src/test/java/com/databend/jdbc/TestBasicDriver.java | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/databend-jdbc/src/test/java/com/databend/jdbc/TestBasicDriver.java b/databend-jdbc/src/test/java/com/databend/jdbc/TestBasicDriver.java index 96668590..8845d6ab 100644 --- a/databend-jdbc/src/test/java/com/databend/jdbc/TestBasicDriver.java +++ b/databend-jdbc/src/test/java/com/databend/jdbc/TestBasicDriver.java @@ -102,6 +102,7 @@ public void TestInsertInto() throws SQLException { System.out.println(statement.getUpdateCount()); } catch (SQLException throwables) { throwables.printStackTrace(); + throw throwables; } } @@ -131,6 +132,7 @@ public void testCreateUserFunction() throws SQLException { Assert.assertEquals(r.getInt(1), 3); } catch (SQLException throwables) { throwables.printStackTrace(); + throw throwables; } } @@ -138,7 +140,7 @@ public void testCreateUserFunction() throws SQLException { public void TestMergeinto() throws SQLException { try (Connection connection = createConnection()) { DatabendStatement statement = (DatabendStatement) connection.createStatement(); - statement.execute("CREATE TABLE IF NOT EXISTS test_basic_driver.target_table (\n" + + statement.execute("CREATE OR REPLACE TABLE test_basic_driver.target_table (\n" + " ID INT,\n" + " Name VARCHAR(50),\n" + " Age INT,\n" + @@ -149,7 +151,7 @@ public void TestMergeinto() throws SQLException { " (1, 'Alice', 25, 'Toronto'),\n" + " (2, 'Bob', 30, 'Vancouver'),\n" + " (3, 'Carol', 28, 'Montreal');"); - statement.execute("CREATE TABLE IF NOT EXISTS test_basic_driver.source_table (\n" + + statement.execute("CREATE OR REPLACE TABLE test_basic_driver.source_table (\n" + " ID INT,\n" + " Name VARCHAR(50),\n" + " Age INT,\n" + @@ -174,6 +176,7 @@ public void TestMergeinto() throws SQLException { System.out.println(statement.getUpdateCount()); } catch (SQLException throwables) { throwables.printStackTrace(); + throw throwables; } }