project works with strings

1 files changed, 20 insertions, 0 deletions

diff --git a/spec/arel/tree_manager.rb b/spec/arel/tree_manager.rb
index 44dbb9f727..92af2f3dcd 100644
--- a/spec/arel/tree_manager.rb
+++ b/spec/arel/tree_manager.rb
@@ -2,6 +2,26 @@ require 'spec_helper'
 
 module Arel
   describe 'tree manager' do
+    describe 'project' do
+      it 'takes strings' do
+        table   = Table.new :users
+        manager = Arel::TreeManager.new Table.engine
+        manager.project '*'
+        manager.to_sql.should == %{
+          SELECT *
+        }.gsub("\n", '').gsub(/(^\s*|\s*$)/, '').squeeze(' ')
+      end
+
+      it "takes sql literals" do
+        table   = Table.new :users
+        manager = Arel::TreeManager.new Table.engine
+        manager.project Nodes::SqlLiteral.new '*'
+        manager.to_sql.should == %{
+          SELECT *
+        }.gsub("\n", '').gsub(/(^\s*|\s*$)/, '').squeeze(' ')
+      end
+    end
+
     describe 'take' do
       it "knows take" do
         table   = Table.new :users