benhoskings-ambitious-activerecord 0.1.3

data/test/benchmark.rb

@@ -0,0 +1,69 @@
+unless defined? Test
+  $:.unshift File.dirname(__FILE__) + '/../../../lib'
+  %w( rubygems ambition/adapters/active_record benchmark ).each { |f| require f }
+
+  class User < ActiveRecord::Base
+    def self.reflections
+      return @reflections if @reflections
+      @reflections = {}
+      @reflections[:ideas]    = Reflection.new(:has_many,   'user_id',     :ideas,   'ideas')
+      @reflections[:invites]  = Reflection.new(:has_many,   'referrer_id', :invites, 'invites')
+      @reflections[:profile]  = Reflection.new(:has_one,    'user_id',     :profile, 'profiles')
+      @reflections[:account]  = Reflection.new(:belongs_to, 'account_id',  :account, 'accounts')
+      @reflections
+    end
+
+    def self.table_name
+      'users'
+    end
+  end
+
+  class Reflection < Struct.new(:macro, :primary_key_name, :name, :table_name)
+  end
+
+  Times = 10000
+
+  Benchmark.bm(30) do |x|
+    x.report 'simple select' do
+      Times.times do
+        User.select { |u| u.id == 20 }.to_hash
+      end
+    end
+
+    x.report 'simple select w/ eval' do
+      Times.times do
+        User.select { |u| u.created_at == Time.now }.to_hash
+      end
+    end
+
+    x.report 'dual select' do
+      Times.times do
+        User.select { |u| u.id == 20 && u.age > 20 }.to_hash
+      end
+    end
+
+    x.report 'join select' do
+      Times.times do
+        User.select { |u| u.id == 20 && u.ideas.name =~ /stuff/ }.to_hash
+      end
+    end
+
+    x.report 'dual select w/ sort' do
+      Times.times do
+        User.select { |u| u.id == 20 && u.age > 20 }.sort_by { |u| u.id }.to_hash
+      end
+    end
+
+    x.report 'dual select w/ sort & first' do
+      Times.times do
+        User.select { |u| u.id == 20 && u.age > 20 }.sort_by { |u| u.id }.first(20).to_hash
+      end
+    end
+
+    x.report "it's complicated" do
+      Times.times do
+        User.select { |u| (u.id == 20 && u.age > 20) || u.profile.name == 'Jon' }.sort_by { |u| [u.id, -u.name] }.first(20).to_hash
+      end
+    end
+  end
+end

data/test/chaining_test.rb

@@ -0,0 +1,36 @@
+require File.dirname(__FILE__) + '/helper'
+
+context "ActiveRecord Adapter" do
+  context "Chaining" do
+    specify "should join selects with AND" do
+      sql = User.select { |m| m.name == 'jon' }
+      sql = sql.select { |m| m.age == 22 }
+      sql.to_s.should == "SELECT * FROM users WHERE users.name = 'jon' AND users.age = 22"
+    end
+
+    specify "should join sort_bys with a comma" do
+      sql = User.select { |m| m.name == 'jon' }
+      sql = sql.sort_by { |m| m.name }
+      sql = sql.sort_by { |m| m.age }
+      sql.to_s.should == "SELECT * FROM users WHERE users.name = 'jon' ORDER BY users.name, users.age"
+    end
+
+    specify "should join selects and sorts intelligently" do
+      sql = User.select { |m| m.name == 'jon' }
+      sql = sql.select { |m| m.age == 22 }
+      sql = sql.sort_by { |m| -m.name }
+      sql = sql.sort_by { |m| m.age }
+      sql.to_s.should == "SELECT * FROM users WHERE users.name = 'jon' AND users.age = 22 ORDER BY users.name DESC, users.age"
+    end
+
+    specify "should join lots of selects and sorts intelligently" do
+      sql = User.select { |m| m.name == 'jon' }
+      sql = sql.select { |m| m.age == 22 }
+      sql = sql.sort_by { |m| m.name }
+      sql = sql.select { |m| m.power == true }
+      sql = sql.sort_by { |m| m.email }
+      sql = sql.select { |m| m.admin == true && m.email == 'chris@ozmm.org' }
+      sql.to_s.should == "SELECT * FROM users WHERE users.name = 'jon' AND users.age = 22 AND users.power = 1 AND (users.admin = 1 AND users.email = 'chris@ozmm.org') ORDER BY users.name, users.email"
+    end
+  end
+end

data/test/count_test.rb

@@ -0,0 +1,19 @@
+require File.dirname(__FILE__) + '/helper'
+
+context "ActiveRecord Adapter" do
+  context "Count" do
+    setup do
+      hash = { :conditions => "users.name = 'jon'" }
+      User.expects(:count).with(hash)
+      @sql = User.select { |m| m.name == 'jon' }
+    end
+
+    specify "size" do
+      @sql.size
+    end
+
+    specify "length" do
+      @sql.length
+    end
+  end
+end

data/test/enumerable_test.rb

@@ -0,0 +1,88 @@
+require File.dirname(__FILE__) + '/helper'
+
+context "ActiveRecord Adapter" do
+  context "Each" do
+    specify "simple ==" do
+      hash = { :conditions => "users.age = 21" }
+      User.expects(:find).with(:all, hash).returns([])
+      User.select { |m| m.age == 21 }.each do |user|
+        puts user.name
+      end
+    end
+
+    specify "limit and conditions" do
+      hash = { :limit => 5, :conditions => "users.age = 21" }
+      User.expects(:find).with(:all, hash).returns([])
+      User.select { |m| m.age == 21 }.first(5).each do |user|
+        puts user.name
+      end
+    end
+
+    specify "limit and conditions and order" do
+      hash = { :limit => 5, :conditions => "users.age = 21", :order => 'users.name' }
+      User.expects(:find).with(:all, hash).returns([])
+      User.select { |m| m.age == 21 }.sort_by { |m| m.name }.first(5).each do |user|
+        puts user.name
+      end
+    end
+
+    specify "limit and order" do
+      hash = { :limit => 5, :order => 'users.name' }
+      User.expects(:find).with(:all, hash).returns([])
+      User.sort_by { |m| m.name }.first(5).each do |user|
+        puts user.name
+      end
+    end
+  end
+
+  context "Enumerable Methods" do
+    specify "map" do
+      hash = { :conditions => "users.age = 21" }
+      User.expects(:find).with(:all, hash).returns([])
+      User.select { |m| m.age == 21 }.map { |u| u.name }
+    end
+
+    specify "each_with_index" do
+      hash = { :conditions => "users.age = 21" }
+      User.expects(:find).with(:all, hash).returns([])
+      User.select { |m| m.age == 21 }.each_with_index do |user, i|
+        puts "#{i}: #{user.name}"
+      end
+    end
+
+    specify "any?" do
+      User.expects(:count).with(:conditions => "users.age > 21").returns(1)
+      User.any? { |u| u.age > 21 }.should == true
+    end
+
+    specify "all?" do
+      User.expects(:count).at_least_once.returns(10, 20)
+      User.all? { |u| u.age > 21 }.should == false
+
+      User.expects(:count).at_least_once.returns(10, 10)
+      User.all? { |u| u.age > 21 }.should == true
+    end
+
+    specify "empty?" do
+      User.expects(:count).with(:conditions => "users.age > 21").returns(1)
+      User.select { |u| u.age > 21 }.empty?.should.equal false
+
+      User.expects(:count).with(:conditions => "users.age > 21").returns(0)
+      User.select { |u| u.age > 21 }.empty?.should.equal true
+    end
+
+    specify "entries" do
+      User.expects(:find).with(:all, {})
+      User.entries
+
+      hash = { :conditions => "users.age = 21" }
+      User.expects(:find).with(:all, hash).returns([])
+      User.select { |m| m.age == 21 }.entries
+    end
+
+    specify "to_a" do
+      User.expects(:find).with(:all, {})
+      User.to_a
+    end
+  end
+end

data/test/helper.rb

@@ -0,0 +1,47 @@
+%w( rubygems test/spec mocha redgreen English ).each { |f| require f }
+
+$LOAD_PATH.unshift *[ File.dirname(__FILE__) + '/../lib', File.dirname(__FILE__) + '/../../../lib' ]
+require 'ambition/adapters/active_record'
+
+class User < ActiveRecord::Base 
+  def self.reflections
+    return @reflections if @reflections
+    @reflections = {}
+    @reflections[:ideas]    = Reflection.new(:has_many,   'user_id',     :ideas,   'ideas')
+    @reflections[:invites]  = Reflection.new(:has_many,   'referrer_id', :invites, 'invites')
+    @reflections[:profile]  = Reflection.new(:has_one,    'user_id',     :profile, 'profiles')
+    @reflections[:account]  = Reflection.new(:belongs_to, 'account_id',  :account, 'accounts')
+    @reflections
+  end
+
+  def self.table_name
+    'users'
+  end
+end
+
+class Reflection < Struct.new(:macro, :primary_key_name, :name, :table_name)
+end
+
+module ActiveRecord
+  module ConnectionAdapters
+    class MysqlAdapter  
+      def initialize(*args)
+        super
+      end
+
+      def connect(*args)
+        true
+      end
+    end
+
+    class PostgreSQLAdapter  
+      def connect(*args)
+        true
+      end
+      class PGError; end
+    end
+
+    class FakeAdapter < AbstractAdapter
+    end
+  end
+end

data/test/join_test.rb

@@ -0,0 +1,62 @@
+require File.dirname(__FILE__) + '/helper'
+
+context "ActiveRecord Adapter" do
+  context "Joins" do
+    specify "simple == on an association" do
+      sql = User.select { |m| m.account.email == 'chris@ozmm.org' }
+      sql.to_hash.should ==  { 
+        :conditions => "accounts.email = 'chris@ozmm.org'", 
+        :include => [:account] 
+      }
+    end
+
+    specify "simple mixed == on an association" do
+      sql = User.select { |m| m.name == 'chris' && m.account.email == 'chris@ozmm.org' }
+      sql.to_hash.should ==  { 
+        :conditions => "(users.name = 'chris' AND accounts.email = 'chris@ozmm.org')", 
+        :include => [:account] 
+      }
+    end
+
+    specify "multiple associations" do
+      sql = User.select { |m| m.ideas.title == 'New Freezer' || m.invites.email == 'pj@hyett.com' }
+      sql.to_hash.should ==  { 
+        :conditions => "(ideas.title = 'New Freezer' OR invites.email = 'pj@hyett.com')",
+        :include => [:ideas, :invites]
+      }
+    end
+
+    specify "belongs_to" do
+      sql = User.select { |m| m.account.id > 20 }
+      sql.to_hash.should ==  { 
+        :conditions => "accounts.id > 20",
+        :include    => [:account]
+      }
+    end
+
+    specify "complex joins have no to_s" do
+      sql = User.select { |m| m.account.id > 20 }
+      should.raise { sql.to_s }
+    end
+
+    specify "non-existant associations" do
+      should.raise { User.select { |m| m.liquor.brand == 'Jack' } }
+    end
+
+    specify "in order" do
+      sql = User.sort_by { |m| m.ideas.title }
+      sql.to_hash.should ==  { 
+        :order   => "ideas.title",
+        :include => [:ideas]
+      }
+    end
+
+    specify "in a more complex order" do
+      sql = User.sort_by { |m| [ m.ideas.title, -m.invites.email ] }
+      sql.to_hash.should ==  { 
+        :order   => "ideas.title, invites.email DESC",
+        :include => [:ideas, :invites]
+      }
+    end
+  end
+end

data/test/profiler.rb

@@ -0,0 +1,36 @@
+$:.unshift File.dirname(__FILE__) + '/../../../lib'
+require 'rubygems'
+require 'ambition'
+require 'ambition/adapters/active_record'
+require 'ruby-prof'
+
+class User < ActiveRecord::Base
+  def self.reflections
+    return @reflections if @reflections
+    @reflections = {}
+    @reflections[:ideas]    = Reflection.new(:has_many,   'user_id',     :ideas,   'ideas')
+    @reflections[:invites]  = Reflection.new(:has_many,   'referrer_id', :invites, 'invites')
+    @reflections[:profile]  = Reflection.new(:has_one,    'user_id',     :profile, 'profiles')
+    @reflections[:account]  = Reflection.new(:belongs_to, 'account_id',  :account, 'accounts')
+    @reflections
+  end
+
+  def self.table_name
+    'users'
+  end
+end
+
+class Reflection < Struct.new(:macro, :primary_key_name, :name, :table_name)
+end
+
+result = RubyProf.profile do
+  1000.times do
+    User.select { |u| (u.id == 20 && u.age > 20) || u.profile.name == 'Jon' }.sort_by { |u| [u.id, -u.name] }.first(20).to_hash
+  end
+end
+
+printer = RubyProf::FlatPrinter.new(result)
+#printer = RubyProf::GraphPrinter.new(result)
+printer.print(STDOUT, 0) 
+
+puts User.select { |u| (u.id == 20 && u.age > 20) || u.profile.name == 'Jon' }.sort_by { |u| [u.id, -u.name] }.first(20).to_hash.inspect

data/test/ruby_test.rb

@@ -0,0 +1,11 @@
+require File.dirname(__FILE__) + '/helper'
+
+context "ActiveRecord Adapter" do
+  context "Inline Ruby" do
+    xspecify "should know what to return" do
+      name = 'David'
+      sql = User.select { |u| name.nil? || u.name == name }.to_s
+      sql.should == "SELECT * FROM users WHERE (users.name = 'David')"
+    end
+  end
+end

data/test/select_test.rb

@@ -0,0 +1,259 @@
+require File.dirname(__FILE__) + '/helper'
+
+context "ActiveRecord Adapter" do
+  context "Select" do
+    specify "simple ==" do
+      sql = User.select { |m| m.name == 'jon' }.to_s
+      sql.should == "SELECT * FROM users WHERE users.name = 'jon'"
+    end
+
+    specify "simple !=" do
+      sql = User.select { |m| m.name != 'jon' }.to_s
+      sql.should == "SELECT * FROM users WHERE users.name <> 'jon'"
+    end
+
+    specify "simple == && ==" do
+      sql = User.select { |m| m.name == 'jon' && m.age == 21 }.to_s
+      sql.should == "SELECT * FROM users WHERE (users.name = 'jon' AND users.age = 21)"
+    end
+
+    specify "simple == || ==" do
+      sql = User.select { |m| m.name == 'jon' || m.age == 21 }.to_s
+      sql.should == "SELECT * FROM users WHERE (users.name = 'jon' OR users.age = 21)"
+    end
+
+    specify "mixed && and ||" do
+      sql = User.select { |m| m.name == 'jon' || m.age == 21 && m.password == 'pass' }.to_s
+      sql.should == "SELECT * FROM users WHERE (users.name = 'jon' OR (users.age = 21 AND users.password = 'pass'))"
+    end
+
+    specify "grouped && and ||" do
+      sql = User.select { |m| (m.name == 'jon' || m.name == 'rick') && m.age == 21 }.to_s
+      sql.should == "SELECT * FROM users WHERE ((users.name = 'jon' OR users.name = 'rick') AND users.age = 21)"
+    end
+
+    specify "simple >/<" do
+      sql = User.select { |m| m.age > 21 }.to_s
+      sql.should == "SELECT * FROM users WHERE users.age > 21"
+
+      sql = User.select { |m| m.age >= 21 }.to_s
+      sql.should == "SELECT * FROM users WHERE users.age >= 21"
+
+      sql = User.select { |m| m.age < 21 }.to_s
+      sql.should == "SELECT * FROM users WHERE users.age < 21"
+    end
+
+    specify "array.include? item" do
+      sql = User.select { |m| [1, 2, 3, 4].include? m.id }.to_s
+      sql.should == "SELECT * FROM users WHERE users.id IN (1, 2, 3, 4)"
+    end
+
+    specify "variable'd array.include? item" do
+      array = [1, 2, 3, 4]
+      sql = User.select { |m| array.include? m.id }.to_s
+      sql.should == "SELECT * FROM users WHERE users.id IN (1, 2, 3, 4)"
+    end
+
+    specify "simple == with variables" do
+      me = 'chris'
+      sql = User.select { |m| m.name == me }.to_s
+      sql.should == "SELECT * FROM users WHERE users.name = '#{me}'"
+    end
+
+    specify "simple == with method arguments" do
+      def test_it(name)
+        sql = User.select { |m| m.name == name }.to_s
+        sql.should == "SELECT * FROM users WHERE users.name = '#{name}'"
+      end
+
+      test_it('chris')
+    end
+
+    specify "simple == with instance variables" do
+      @me = 'chris'
+      sql = User.select { |m| m.name == @me }.to_s
+      sql.should == "SELECT * FROM users WHERE users.name = '#{@me}'"
+    end
+
+    specify "simple == with instance variable method call" do
+      require 'ostruct'
+      @person = OpenStruct.new(:name => 'chris')
+
+      sql = User.select { |m| m.name == @person.name }.to_s
+      sql.should == "SELECT * FROM users WHERE users.name = '#{@person.name}'"
+    end
+
+    specify "simple == with global variables" do
+      $my_name = 'boston'
+      sql = User.select { |m| m.name == $my_name }.to_s
+      sql.should == "SELECT * FROM users WHERE users.name = '#{$my_name}'"
+    end
+
+    specify "simple == with method call" do
+      def band
+        'megadeth'
+      end
+
+      sql = User.select { |m| m.name == band }.to_s
+      sql.should == "SELECT * FROM users WHERE users.name = '#{band}'"
+    end
+
+    specify "simple =~ with string" do
+      sql = User.select { |m| m.name =~ 'chris' }.to_s
+      sql.should == "SELECT * FROM users WHERE users.name LIKE 'chris'"
+
+      sql = User.select { |m| m.name =~ 'chri%' }.to_s
+      sql.should == "SELECT * FROM users WHERE users.name LIKE 'chri%'"
+    end
+
+    specify "simple !~ with string" do
+      sql = User.select { |m| m.name !~ 'chris' }.to_s
+      sql.should == "SELECT * FROM users WHERE users.name NOT LIKE 'chris'"
+
+      sql = User.select { |m| !(m.name =~ 'chris') }.to_s
+      sql.should == "SELECT * FROM users WHERE users.name NOT LIKE 'chris'"
+    end
+
+    specify "simple =~ with regexp" do
+      sql = User.select { |m| m.name =~ /chris/ }.to_s
+      sql.should == "SELECT * FROM users WHERE users.name REGEXP 'chris'"
+    end
+
+    specify "simple =~ with regexp flags" do
+      sql = User.select { |m| m.name =~ /chris/i }.to_s
+      sql.should == "SELECT * FROM users WHERE users.name REGEXP 'chris'"
+    end
+
+    specify "simple LOWER()" do
+      sql = User.select { |m| m.name.downcase =~ 'chris%' }.to_s
+      sql.should == "SELECT * FROM users WHERE LOWER(users.name) LIKE 'chris%'"
+    end
+
+    specify "simple UPPER()" do
+      sql = User.select { |m| m.name.upcase =~ 'chris%' }.to_s
+      sql.should == "SELECT * FROM users WHERE UPPER(users.name) LIKE 'chris%'"
+    end
+
+    specify "undefined equality symbol" do
+      should.raise { User.select { |m| m.name =* /chris/ }.to_s }
+    end
+
+    specify "block variable / assigning variable conflict" do
+      m = User.select { |m| m.name == 'chris' }.to_s
+      m.should == "SELECT * FROM users WHERE users.name = 'chris'"
+    end
+    
+    specify "simple == with inline ruby" do
+      sql = User.select { |m| m.created_at == 2.days.ago.to_s(:db) }.to_s
+      sql.should == "SELECT * FROM users WHERE users.created_at = '#{2.days.ago.to_s(:db)}'"
+    end
+
+    specify "deep chains" do
+      should.raise do
+        User.select { |m| m.created_at.something.else.perhaps }.to_s
+      end
+    end
+
+    specify "inspect" do
+      User.select { |u| u.name }.inspect.should.match %r(call #to_s or #to_hash)
+    end
+  end
+
+  context "Detect" do
+    specify "simple ==" do
+      User.expects(:select).returns(mock(:first => true))
+      User.detect { |m| m.name == 'chris' }
+    end
+
+    specify "nothing found" do
+      User.expects(:select).returns(mock(:first => nil))
+      User.detect { |m| m.name == 'chris' }.should.be.nil
+    end
+  end
+
+  xcontext "[]" do
+    specify "finds a single row" do
+      User.expects(:find).with(1)
+      User[1]
+    end
+  end
+
+  context "PostgreSQL specific"  do
+    setup do
+      ActiveRecord::Base.connection = ActiveRecord::ConnectionAdapters::PostgreSQLAdapter.new 'fake_connection', 'fake_logger'
+    end
+
+    teardown do
+      ActiveRecord::Base.remove_connection
+    end
+
+    specify "quoting of column name" do
+      me = 'chris'
+      sql = User.select { |m| m.name == me }.to_s
+      sql.should == %(SELECT * FROM users WHERE users."name" = '#{me}')
+    end
+
+    specify "simple =~ with regexp" do
+      sql = User.select { |m| m.name =~ /chris/ }.to_s
+      sql.should == %(SELECT * FROM users WHERE users."name" ~ 'chris')
+    end
+
+    specify "insensitive =~" do
+      sql = User.select { |m| m.name =~ /chris/i }.to_s
+      sql.should == %(SELECT * FROM users WHERE users."name" ~* 'chris')
+    end
+
+    specify "negated =~" do
+      sql = User.select { |m| m.name !~ /chris/ }.to_s
+      sql.should == %(SELECT * FROM users WHERE users."name" !~ 'chris')
+    end
+
+    specify "negated insensitive =~" do
+      sql = User.select { |m| m.name !~ /chris/i }.to_s
+      sql.should == %(SELECT * FROM users WHERE users."name" !~* 'chris')
+    end
+  end
+
+  context "MySQL specific" do
+    setup do
+      ActiveRecord::Base.connection = ActiveRecord::ConnectionAdapters::MysqlAdapter.new('connection', 'logger')
+    end
+
+    teardown do
+      ActiveRecord::Base.connection = ActiveRecord::ConnectionAdapters::FakeAdapter.new('connection', 'logger')
+    end
+
+    specify "quoting of column name" do
+      me = 'chris'
+      sql = User.select { |m| m.name == me }.to_s
+      sql.should == "SELECT * FROM users WHERE users.`name` = '#{me}'"
+    end
+
+    specify "simple =~ with regexp" do
+      sql = User.select { |m| m.name =~ /chris/ }.to_s
+      sql.should == "SELECT * FROM users WHERE users.`name` REGEXP 'chris'"
+    end
+
+    specify "negated =~" do
+      sql = User.select { |m| m.name !~ 'chris' }.to_s
+      sql.should == "SELECT * FROM users WHERE users.`name` NOT LIKE 'chris'"
+    end
+
+    specify "negated =~ with regexp" do
+      sql = User.select { |m| m.name !~ /chris/ }.to_s
+      sql.should == "SELECT * FROM users WHERE users.`name` NOT REGEXP 'chris'"
+    end
+  end
+
+  context "Adapter without overrides" do
+    setup do
+      ActiveRecord::Base.connection = ActiveRecord::ConnectionAdapters::FakeAdapter.new('connection', 'logger')
+    end
+
+    specify "quoting of column name" do
+      me = 'chris'
+      sql = User.select { |m| m.name == me }.to_s
+      sql.should == "SELECT * FROM users WHERE users.name = '#{me}'"
+    end
+  end
+end