[FIX] Doh! Forgot to add files to previous commit.

Author: jasonallen

lib/sql_tree/node/begin_statement.rb

@@ -0,0 +1,12 @@
+module SQLTree::Node
+  class BeginStatement < Base
+    def to_sql(options = {})
+      "BEGIN"
+    end
+
+    def self.parse(tokens)
+      tokens.consume(SQLTree::Token::BEGIN)
+      return self.new
+    end
+  end
+end

lib/sql_tree/node/commit_statement.rb

@@ -0,0 +1,12 @@
+module SQLTree::Node
+  class CommitStatement < Base
+    def to_sql(options = {})
+      "COMMIT"
+    end
+
+    def self.parse(tokens)
+      tokens.consume(SQLTree::Token::COMMIT)
+      return self.new
+    end
+  end
+end

lib/sql_tree/node/set_query.rb

@@ -0,0 +1,38 @@
+module SQLTree::Node
+
+  # The <tt>DeleteQuery</tt> node represents an SQL DELETE query.
+  #
+  # This node has two children: <tt>table</tt> and <tt>where</tt>.
+  class SetQuery < Base
+
+    # The variable (<tt>SQLTree::Node::Field</tt>) that is being set.
+    child :variable
+    
+    # The <tt>SQLTree::Node::Expression</tt> value that the variable is being
+    # set to.
+    child :value
+
+    # Initializes a new DeleteQuery instance.
+    def initialize(variable, value)
+      @variable, @value = variable, value
+    end
+
+    # Generates an SQL DELETE query from this node.
+    def to_sql(options = {})
+      sql = "SET #{variable.to_sql(options)}"
+      sql << " TO #{value.to_sql(options)}"
+      sql
+    end
+    
+    # Parses a SET query from a stream of tokens.
+    # <tt>tokens</tt>:: The token stream to parse from, which is an instance
+    #                   of <tt> SQLTree::Parser</tt>.
+    def self.parse(tokens)
+      tokens.consume(SQLTree::Token::SET)
+      variable = SQLTree::Node::Expression::Field.parse(tokens)
+      tokens.consume(SQLTree::Token::TO)
+      value = SQLTree::Node::Expression::Value.parse(tokens)
+      return self.new(variable, value)
+    end
+  end
+end

spec/unit/set_query_spec.rb

@@ -0,0 +1,11 @@
+require "#{File.dirname(__FILE__)}/../spec_helper"
+
+describe SQLTree::Node::SetQuery do
+
+  it "should parse a set query correctly" do
+    set = SQLTree::Node::SetQuery["SET foo TO 'var'"]
+    set.variable.should == SQLTree::Node::Expression::Field.new("foo")
+    set.value.should == SQLTree::Node::Expression::Value.new("var")
+  end
+
+end

sql_tree.gemspec

@@ -22,6 +22,6 @@ Gem::Specification.new do |s|
 
   # Do not modify the files and test_files values by hand, because this will
   # automatically by them gem release script.
-  s.files = %w(spec/unit/select_query_spec.rb spec/unit/insert_query_spec.rb spec/spec_helper.rb lib/sql_tree/tokenizer.rb lib/sql_tree/node/join.rb .gitignore lib/sql_tree/node/ordering.rb LICENSE spec/lib/matchers.rb lib/sql_tree/parser.rb sql_tree.gemspec spec/unit/tokenizer_spec.rb spec/unit/expression_node_spec.rb spec/unit/delete_query_spec.rb spec/unit/leaf_node_spec.rb lib/sql_tree/token.rb lib/sql_tree/node/table_reference.rb lib/sql_tree/node/source.rb lib/sql_tree/node/insert_query.rb Rakefile tasks/github-gem.rake spec/unit/update_query_spec.rb spec/integration/parse_and_generate_spec.rb lib/sql_tree/node/select_query.rb lib/sql_tree/node.rb README.rdoc spec/integration/api_spec.rb lib/sql_tree/node/expression.rb lib/sql_tree/node/delete_query.rb lib/sql_tree/node/select_declaration.rb lib/sql_tree.rb lib/sql_tree/node/update_query.rb)
-  s.test_files = %w(spec/unit/select_query_spec.rb spec/unit/insert_query_spec.rb spec/unit/tokenizer_spec.rb spec/unit/expression_node_spec.rb spec/unit/delete_query_spec.rb spec/unit/leaf_node_spec.rb spec/unit/update_query_spec.rb spec/integration/parse_and_generate_spec.rb spec/integration/api_spec.rb)
+  s.files = %w(spec/spec_helper.rb lib/sql_tree/tokenizer.rb lib/sql_tree/node/join.rb sql_tree.gemspec spec/unit/select_query_spec.rb spec/lib/matchers.rb lib/sql_tree/parser.rb .gitignore lib/sql_tree/node/source.rb LICENSE spec/unit/expression_node_spec.rb spec/unit/delete_query_spec.rb Gemfile spec/unit/leaf_node_spec.rb lib/sql_tree/token.rb Rakefile spec/unit/update_query_spec.rb spec/unit/tokenizer_spec.rb spec/integration/api_spec.rb lib/sql_tree/node/select_query.rb lib/sql_tree/node.rb README.rdoc lib/sql_tree/node/table_reference.rb lib/sql_tree/node/insert_query.rb tasks/github-gem.rake spec/integration/parse_and_generate_spec.rb lib/sql_tree/node/expression.rb spec/unit/insert_query_spec.rb lib/sql_tree/node/select_declaration.rb lib/sql_tree.rb lib/sql_tree/node/update_query.rb lib/sql_tree/node/ordering.rb lib/sql_tree/node/delete_query.rb)
+  s.test_files = %w(spec/unit/select_query_spec.rb spec/unit/expression_node_spec.rb spec/unit/delete_query_spec.rb spec/unit/leaf_node_spec.rb spec/unit/update_query_spec.rb spec/unit/tokenizer_spec.rb spec/integration/api_spec.rb spec/integration/parse_and_generate_spec.rb spec/unit/insert_query_spec.rb)
 end