Don't attempt to close cursors when in failed transaction (it's failed and result in sigabrt)libdbpp-postgresql-1.0.4

1 files changed, 32 insertions, 0 deletions

diff --git a/libpqpp/unittests/testpq.cpp b/libpqpp/unittests/testpq.cpp
index 1905192..64d60d9 100644
--- a/libpqpp/unittests/testpq.cpp
+++ b/libpqpp/unittests/testpq.cpp
@@ -11,6 +11,7 @@
 #include <boost/date_time/posix_time/posix_time.hpp>
 #include <pq-error.h>
 #include <pq-connection.h>
+#include <selectcommandUtil.impl.h>
 
 class StandardMockDatabase : public PQ::Mock {
 	public:
@@ -286,6 +287,37 @@ BOOST_AUTO_TEST_CASE( statementReuse )
 	delete ro;
 }
 
+BOOST_AUTO_TEST_CASE( closeOnError )
+{
+	auto ro = DB::ConnectionPtr(DB::MockDatabase::openConnectionTo("pqmock"));
+	BOOST_REQUIRE_THROW({
+			ro->select("SELECT * FROM test")->forEachRow<>([&ro](){
+					ro->execute("nonsense");
+				});
+			}, DB::Error);
+	BOOST_REQUIRE_THROW({
+			ro->select("SELECT * FROM test")->forEachRow<>([&ro](){
+				ro->select("SELECT * FROM test")->forEachRow<>([&ro](){
+						ro->execute("nonsense");
+					});
+				});
+			}, DB::Error);
+	ro->beginTx();
+	BOOST_REQUIRE_THROW({
+			ro->select("SELECT * FROM test")->forEachRow<>([&ro](){
+					ro->execute("nonsense");
+				});
+			}, DB::Error);
+	BOOST_REQUIRE_THROW({
+			ro->select("SELECT * FROM test")->forEachRow<>([&ro](){
+				ro->select("SELECT * FROM test")->forEachRow<>([&ro](){
+						ro->execute("nonsense");
+					});
+				});
+			}, DB::Error);
+	ro->commitTx();
+}
+
 BOOST_AUTO_TEST_SUITE_END();
 
 BOOST_AUTO_TEST_CASE( connfail )