From 1e370263d2f59da04fd1f8fe55bb83afdc0a51dc Mon Sep 17 00:00:00 2001 From: Dave Cramer Date: Fri, 15 Nov 2019 07:42:29 -0500 Subject: [PATCH] fix: PgSQLXML setCharacterStream() results in null value (#1608) * fix: PgSQLXML setCharacterStream() results in null value fixes Issue #731 --- .../java/org/postgresql/jdbc/PgSQLXML.java | 1 + .../org/postgresql/jdbc/PgSQLXMLTest.java | 50 +++++++++++++++++++ .../postgresql/test/jdbc2/Jdbc2TestSuite.java | 2 + 3 files changed, 53 insertions(+) create mode 100644 pgjdbc/src/test/java/org/postgresql/jdbc/PgSQLXMLTest.java diff --git a/pgjdbc/src/main/java/org/postgresql/jdbc/PgSQLXML.java b/pgjdbc/src/main/java/org/postgresql/jdbc/PgSQLXML.java index 3cf1277f8e..919df065b9 100644 --- a/pgjdbc/src/main/java/org/postgresql/jdbc/PgSQLXML.java +++ b/pgjdbc/src/main/java/org/postgresql/jdbc/PgSQLXML.java @@ -175,6 +175,7 @@ public synchronized OutputStream setBinaryStream() throws SQLException { public synchronized Writer setCharacterStream() throws SQLException { checkFreed(); initialize(); + active = true; stringWriter = new StringWriter(); return stringWriter; } diff --git a/pgjdbc/src/test/java/org/postgresql/jdbc/PgSQLXMLTest.java b/pgjdbc/src/test/java/org/postgresql/jdbc/PgSQLXMLTest.java new file mode 100644 index 0000000000..53fd56dc13 --- /dev/null +++ b/pgjdbc/src/test/java/org/postgresql/jdbc/PgSQLXMLTest.java @@ -0,0 +1,50 @@ +/* + * Copyright (c) 2019, PostgreSQL Global Development Group + * See the LICENSE file in the project root for more information. + */ + +package org.postgresql.jdbc; + +import static org.junit.Assert.assertEquals; +import static org.junit.Assert.assertNotNull; +import static org.junit.Assert.assertTrue; + +import org.postgresql.test.TestUtil; +import org.postgresql.test.jdbc2.BaseTest4; + +import org.junit.Before; +import org.junit.Test; + +import java.io.Writer; +import java.sql.PreparedStatement; +import java.sql.ResultSet; +import java.sql.SQLXML; +import java.sql.Statement; + +public class PgSQLXMLTest extends BaseTest4 { + + @Override + @Before + public void setUp() throws Exception { + super.setUp(); + TestUtil.createTempTable(con, "xmltab","x xml"); + } + + @Test + public void setCharacterStream() throws Exception { + String exmplar = "value"; + SQLXML pgSQLXML = con.createSQLXML(); + Writer writer = pgSQLXML.setCharacterStream(); + writer.write(exmplar); + PreparedStatement preparedStatement = con.prepareStatement("insert into xmltab values (?)"); + preparedStatement.setSQLXML(1,pgSQLXML); + preparedStatement.execute(); + + Statement statement = con.createStatement(); + ResultSet rs = statement.executeQuery("select * from xmltab"); + assertTrue(rs.next()); + SQLXML result = rs.getSQLXML(1); + assertNotNull(result); + assertEquals(exmplar, result.getString()); + } +} diff --git a/pgjdbc/src/test/java/org/postgresql/test/jdbc2/Jdbc2TestSuite.java b/pgjdbc/src/test/java/org/postgresql/test/jdbc2/Jdbc2TestSuite.java index 7dc50232d0..004f1b614a 100644 --- a/pgjdbc/src/test/java/org/postgresql/test/jdbc2/Jdbc2TestSuite.java +++ b/pgjdbc/src/test/java/org/postgresql/test/jdbc2/Jdbc2TestSuite.java @@ -13,6 +13,7 @@ import org.postgresql.core.ReturningParserTest; import org.postgresql.core.v3.V3ParameterListTests; import org.postgresql.jdbc.DeepBatchedInsertStatementTest; +import org.postgresql.jdbc.PgSQLXMLTest; import org.postgresql.jdbc.PrimitiveArraySupportTest; import org.postgresql.test.core.JavaVersionTest; import org.postgresql.test.core.LogServerMessagePropertyTest; @@ -82,6 +83,7 @@ PGPropertyTest.class, PGTimestampTest.class, PGTimeTest.class, + PgSQLXMLTest.class, PreparedStatementTest.class, PrimitiveArraySupportTest.class, QuotationTest.class,