Make sure ROLLBACK statements are actually parsed and add some specs for control statements.

Author: wvanbergen

lib/sql_tree/parser.rb

@@ -112,13 +112,14 @@ def debug
   #
   def parse!
     case self.peek
-    when SQLTree::Token::SELECT then SQLTree::Node::SelectQuery.parse(self)
-    when SQLTree::Token::INSERT then SQLTree::Node::InsertQuery.parse(self)
-    when SQLTree::Token::DELETE then SQLTree::Node::DeleteQuery.parse(self)
-    when SQLTree::Token::UPDATE then SQLTree::Node::UpdateQuery.parse(self)
-    when SQLTree::Token::BEGIN  then SQLTree::Node::BeginStatement.parse(self)
-    when SQLTree::Token::COMMIT then SQLTree::Node::CommitStatement.parse(self)
-    when SQLTree::Token::SET    then SQLTree::Node::SetQuery.parse(self)
+    when SQLTree::Token::SELECT;   SQLTree::Node::SelectQuery.parse(self)
+    when SQLTree::Token::INSERT;   SQLTree::Node::InsertQuery.parse(self)
+    when SQLTree::Token::DELETE;   SQLTree::Node::DeleteQuery.parse(self)
+    when SQLTree::Token::UPDATE;   SQLTree::Node::UpdateQuery.parse(self)
+    when SQLTree::Token::BEGIN;    SQLTree::Node::BeginStatement.parse(self)
+    when SQLTree::Token::COMMIT;   SQLTree::Node::CommitStatement.parse(self)
+    when SQLTree::Token::ROLLBACK; SQLTree::Node::RollbackStatement.parse(self)
+    when SQLTree::Token::SET;      SQLTree::Node::SetQuery.parse(self)
     else raise UnexpectedToken.new(self.peek)
     end
   end

spec/unit/control_statements_spec.rb

@@ -0,0 +1,17 @@
+require 'spec_helper'
+
+describe SQLTree, 'for transaction statements' do
+  
+  it "should parse a BEGIN statement correctly" do
+    SQLTree['BEGIN'].should be_kind_of(SQLTree::Node::BeginStatement)
+  end
+
+  it "should parse a COMMIT statement correctly" do
+    SQLTree['COMMIT'].should be_kind_of(SQLTree::Node::CommitStatement)
+  end
+
+  it "should parse a ROLLBACK statement correctly" do
+    SQLTree['ROLLBACK'].should be_kind_of(SQLTree::Node::RollbackStatement)
+  end
+  
+end