spec/prepare.rb @@ -1,3 +1,4 @@ .DS_Store pkg doc +*.sqlite3 .gitignore @@ -1,4 +1,17 @@ module ScopeBuilder - class Builder < ActiveRecord::NamedScope::Scope + class Builder + def initialize(proxy_scope) + @proxy_scope = proxy_scope + end + + def method_missing(method, *args, &block) + result = @proxy_scope.send(method, *args, &block) + if result.class == ActiveRecord::NamedScope::Scope + @proxy_scope = result + self + else + result + end + end end end lib/scope_builder/builder.rb @@ -1,7 +1,7 @@ module ScopeBuilder module ModelAdditions def scope_builder - Builder.new(self, {}) + Builder.new(scoped({})) end end end lib/scope_builder/model_additions.rb @@ -1,11 +1,16 @@ require File.dirname(__FILE__) + '/../spec_helper' class Product < ActiveRecord::Base - named_scope :released + named_scope :released, :conditions => ['released=?', true] end describe ScopeBuilder::Builder do before(:each) do + Product.delete_all + Product.create!(:name => 'a', :released => true) + Product.create!(:name => 'b', :released => false) + Product.create!(:name => 'c', :released => true) + @builder = Product.scope_builder end @@ -16,4 +21,22 @@ describe ScopeBuilder::Builder do it "should allow named scopes to be called through it" do @builder.released.proxy_options.should == Product.released.proxy_options end + + it "should remember scope calls" do + @builder.released + @builder.proxy_options.should == Product.released.proxy_options + end + + it "should build up scopes" do + @builder.released.scoped(:limit => 1) + @builder.scoped(:offset => 1) + @builder.all.should == Product.released.scoped(:limit => 1, :offset => 1).all + end + + it "should enumerate like an array" do + products = Product.find(:all) + @builder.each_with_index do |product, index| + product.should == products[index] + end + end end spec/scope_builder/builder_spec.rb @@ -6,6 +6,12 @@ require 'action_controller' require 'action_view' require File.dirname(__FILE__) + '/../lib/scope_builder.rb' +# setup database adapter +ActiveRecord::Base.establish_connection({ + :adapter => "sqlite3", + :dbfile => File.dirname(__FILE__) + "/test.sqlite3" +}) + Spec::Runner.configure do |config| config.mock_with :mocha end spec/spec_helper.rb spec/scope_builder/model_additions_spec.rb a0c33fce5a5276e8ab10a4164069a688ac00eaea Ryan Bates ryan@railscasts.com http://github.com/ryanb/scope-builder/commit/dbfa699ca9c22a7467e6f8c10fa3b9e51591d27e dbfa699ca9c22a7467e6f8c10fa3b9e51591d27e 2008-06-26T09:27:18-07:00 2008-06-26T09:27:18-07:00 rewriting builder to act as a simple delegator b3b6b08ec8a34cf7003fee3f73513d634591b13e Ryan Bates ryan@railscasts.com