scope_chain 0.0.7 → 0.0.8

data/lib/scope_chain.rb

@@ -16,32 +16,43 @@ module ScopeChain
   end
 
   class AssociationChain
-    attr_accessor :instance, :association
+    class MissingAssociationError < StandardError; end
+    attr_accessor :instance, :association_name, :association
 
     def initialize(instance)
       self.instance = instance
     end
 
     def as(association_name)
-      self.association = instance.class.reflect_on_association(association_name)
+      self.association_name = association_name
 
       chain
     end
 
     private
-    def association_name(klass)
-      klass.name.underscore.pluralize
-    end
-
     def chain
+      raise MissingAssociationError.new("Can't find association #{association_name} on #{instance.class.name}") if association.nil?
+
       ScopeChain::Chain.new(association.klass).tap do |chain|
         instance.stubs(association.name => association.klass)
+
+        # This is a nasty hack
+        chain.define_singleton_method(:build) do |*arguments|
+          add_link :build, *arguments
+        end
+
+        association.klass.define_singleton_method(:build) {}
       end
     end
+
+    def association
+      # TODO See if we can just mock for the reflection
+      @association ||= instance.class.reflect_on_association(association_name)
+    end
   end
 
   class Chain
-    LINKS = [:select, :where, :includes, :order, :find, :sum, :new, :create, :create!]
+    LINKS = [:select, :where, :includes, :order, :find, :sum, :new, :create, :create!, :limit, :joins]
     ALIASES = {} 
 
     class ConflictedExistenceError < StandardError

data/lib/scope_chain/version.rb

@@ -1,3 +1,3 @@
 module ScopeChain
-  VERSION = "0.0.7"
+  VERSION = "0.0.8"
 end

data/spec/scope_chain_spec.rb

@@ -147,6 +147,10 @@ describe ScopeChain::Chain do
   end
 
   context "with associations" do
+    it "raises an error with a missing association" do
+      expect { ScopeChain.on(Owner.new).as(:missing_association)}.to raise_error(ScopeChain::AssociationChain::MissingAssociationError)
+    end
+
     describe "has_many" do
       it "properly sets stuff up" do
         source = Owner.new

metadata

@@ -2,7 +2,7 @@
 name: scope_chain
 version: !ruby/object:Gem::Version 
   prerelease: 
-  version: 0.0.7
+  version: 0.0.8
 platform: ruby
 authors: 
 - Jon Moses