6 files changed, 65 insertions, 0 deletions
diff --git a/libdbpp/unittests/Jamfile.jam b/libdbpp/unittests/Jamfile.jam
index d0500f0..83e1ad4 100644
--- a/libdbpp/unittests/Jamfile.jam
+++ b/libdbpp/unittests/Jamfile.jam
@@ -29,6 +29,10 @@ run
 	<library>..//dbppcore
 	<library>..//adhocutil
 	<library>boost_utf
+	<dependency>commentsMixedIn.sql
+	<dependency>indentedStatement.sql
+	<dependency>indentedOneLineComment.sql
+	<dependency>indentedBlockComment.sql
 	<dependency>parseTest.sql
 	<dependency>unterminatedComment.sql
 	<dependency>unterminatedString.sql
diff --git a/libdbpp/unittests/commentsMixedIn.sql b/libdbpp/unittests/commentsMixedIn.sql
new file mode 100644
index 0000000..1766309
--- /dev/null
+++ b/libdbpp/unittests/commentsMixedIn.sql
@@ -0,0 +1,7 @@
+CREATE TABLE foo(/*
+										Foo contains test things
+										*/
+		id int, -- Every table deserves an Id, right?
+		timestamp time stamp -- And a timestamp
+		);
+
diff --git a/libdbpp/unittests/indentedBlockComment.sql b/libdbpp/unittests/indentedBlockComment.sql
new file mode 100644
index 0000000..45294a0
--- /dev/null
+++ b/libdbpp/unittests/indentedBlockComment.sql
@@ -0,0 +1,5 @@
+
+
+     /* Some comment text
+			*/
+
diff --git a/libdbpp/unittests/indentedOneLineComment.sql b/libdbpp/unittests/indentedOneLineComment.sql
new file mode 100644
index 0000000..d72bb31
--- /dev/null
+++ b/libdbpp/unittests/indentedOneLineComment.sql
@@ -0,0 +1,5 @@
+
+
+     -- Some comment text
+
+
diff --git a/libdbpp/unittests/indentedStatement.sql b/libdbpp/unittests/indentedStatement.sql
new file mode 100644
index 0000000..4273f35
--- /dev/null
+++ b/libdbpp/unittests/indentedStatement.sql
@@ -0,0 +1,5 @@
+
+
+    SELECT 1;
+
+
diff --git a/libdbpp/unittests/testParse.cpp b/libdbpp/unittests/testParse.cpp
index 1272f9b..093cc7d 100644
--- a/libdbpp/unittests/testParse.cpp
+++ b/libdbpp/unittests/testParse.cpp
@@ -78,6 +78,45 @@ BOOST_AUTO_TEST_CASE( parseStringParse )
 	BOOST_REQUIRE_EQUAL("INSERT INTO name(t, i) VALUES('fancy string '' \\' \\r \\n', 7)", p.executed[1]);
 }
 
+BOOST_AUTO_TEST_CASE( indentedStatement )
+{
+	RecordingParser p(rootDir / "indentedStatement.sql");
+	p.Execute();
+	BOOST_REQUIRE_EQUAL(1, p.executed.size());
+	BOOST_REQUIRE_EQUAL("SELECT 1", p.executed[0]);
+	BOOST_REQUIRE(p.comments.empty());
+}
+
+BOOST_AUTO_TEST_CASE( indentedOneLineComment )
+{
+	RecordingParser p(rootDir / "indentedOneLineComment.sql");
+	p.Execute();
+	BOOST_REQUIRE_EQUAL(1, p.comments.size());
+	BOOST_REQUIRE_EQUAL("Some comment text", p.comments[0]);
+	BOOST_REQUIRE(p.executed.empty());
+}
+
+BOOST_AUTO_TEST_CASE( indentedBlockComment )
+{
+	RecordingParser p(rootDir / "indentedBlockComment.sql");
+	p.Execute();
+	BOOST_REQUIRE_EQUAL(1, p.comments.size());
+	BOOST_REQUIRE_EQUAL("Some comment text", p.comments[0]);
+	BOOST_REQUIRE(p.executed.empty());
+}
+
+BOOST_AUTO_TEST_CASE( commentsMixedIn )
+{
+	RecordingParser p(rootDir / "commentsMixedIn.sql");
+	p.Execute();
+	BOOST_REQUIRE_EQUAL(1, p.executed.size());
+	BOOST_REQUIRE_EQUAL("CREATE TABLE foo(\n\t\tid int,\n\t\ttimestamp time stamp\n\t\t)", p.executed[0]);
+	BOOST_REQUIRE_EQUAL(3, p.comments.size());
+	BOOST_REQUIRE_EQUAL("Foo contains test things", p.comments[0]);
+	BOOST_REQUIRE_EQUAL("Every table deserves an Id, right?", p.comments[1]);
+	BOOST_REQUIRE_EQUAL("And a timestamp", p.comments[2]);
+}
+
 BOOST_AUTO_TEST_CASE( parseUnterminateComment )
 {
 	assertFail(rootDir / "unterminatedComment.sql");