1 files changed, 95 insertions, 0 deletions
diff --git a/libdbpp/unittests/testParse.cpp b/libdbpp/unittests/testParse.cpp
new file mode 100644
index 0000000..1272f9b
--- /dev/null
+++ b/libdbpp/unittests/testParse.cpp
@@ -0,0 +1,95 @@
+#define BOOST_TEST_MODULE DbConnection
+#include <boost/test/unit_test.hpp>
+
+#include <factory.h>
+#include <connection.h>
+#include <definedDirs.h>
+#include <fstream>
+#include <vector>
+#include <error.h>
+#include <sqlParse.h>
+
+typedef std::vector<std::string> SQLs;
+BOOST_TEST_SPECIALIZED_COLLECTION_COMPARE(SQLs);
+
+class RecordingParser : std::fstream, public DB::SqlParse {
+	public:
+		RecordingParser(const boost::filesystem::path & p) :
+			std::fstream(p.string()),
+			DB::SqlParse(*this, p.parent_path())
+		{
+		}
+
+    void Comment(const std::string & c) const
+		{
+			comments.push_back(c);
+		}
+
+    void Statement(const std::string & s) const
+		{
+			executed.push_back(s);
+		}
+
+		mutable SQLs comments;
+		mutable SQLs executed;
+};
+
+template<typename E = DB::SqlParseException>
+void assertFail(const boost::filesystem::path & p)
+{
+	BOOST_TEST_CONTEXT(p) {
+		BOOST_REQUIRE_THROW({
+			RecordingParser s(p);
+			s.Execute();
+		}, E);
+	}
+}
+
+BOOST_AUTO_TEST_CASE( parseBad )
+{
+	assertFail("/bad");
+}
+
+BOOST_AUTO_TEST_CASE( parse )
+{
+	RecordingParser p(rootDir / "parseTest.sql");
+	p.Execute();
+	BOOST_REQUIRE_EQUAL(p.executed.size(), 3);
+	BOOST_REQUIRE_EQUAL(p.executed[1], "INSERT INTO name(t, i) VALUES('string', 3)");
+}
+
+BOOST_AUTO_TEST_CASE( parseDollarQuote )
+{
+	RecordingParser p(rootDir / "dollarQuote.sql");
+	p.Execute();
+}
+
+BOOST_AUTO_TEST_CASE( parseScriptDir )
+{
+	RecordingParser p(rootDir / "scriptDir.sql");
+	p.Execute();
+}
+
+BOOST_AUTO_TEST_CASE( parseStringParse )
+{
+	RecordingParser p(rootDir / "stringParse.sql");
+	p.Execute();
+	BOOST_REQUIRE_EQUAL(2, p.executed.size());
+	BOOST_REQUIRE_EQUAL("INSERT INTO name(t, i) VALUES('fancy string '' \\' \\r \\n', 7)", p.executed[1]);
+}
+
+BOOST_AUTO_TEST_CASE( parseUnterminateComment )
+{
+	assertFail(rootDir / "unterminatedComment.sql");
+}
+
+BOOST_AUTO_TEST_CASE( parseUnterminateDollarQuote )
+{
+	assertFail(rootDir / "unterminatedDollarQuote.sql");
+}
+
+BOOST_AUTO_TEST_CASE( parseUnterminateString )
+{
+	assertFail(rootDir / "unterminatedString.sql");
+}
+