Merge pull request #303 from projecthydra/relation_select

Added #select to Relation

lib/active_fedora/associations/collection_association.rb

@@ -54,6 +54,11 @@ def ids_writer(ids)
         #send("#{reflection.name}=", ids.collect { |id| reflection.klass.find(id)})
       end
 
+      def reset
+        reset_target!
+        @loaded = false
+      end
+
       def first(*args)
         first_or_last(:first, *args)
       end
@@ -111,11 +116,6 @@ def to_ary
       end
       alias_method :to_a, :to_ary
 
-      def reset
-        reset_target!
-        @loaded = false
-      end
-
       def build(attributes = {}, &block)
         if attributes.is_a?(Array)
           attributes.collect { |attr| build(attr, &block) }

lib/active_fedora/relation/query_methods.rb

@@ -50,5 +50,9 @@ def extending!(*modules, &block) # :nodoc:
 
       self
     end
+
+    def select
+      to_a.select { |*block_args| yield(*block_args) }
+    end
   end
 end

spec/integration/relation_spec.rb

@@ -0,0 +1,42 @@
+require 'spec_helper'
+
+describe ActiveFedora::Base do
+  before :all do
+    class Library < ActiveFedora::Base 
+      has_many :books, property: :has_member
+    end
+    class Book < ActiveFedora::Base; end
+  end
+  after :all do
+    Library.delete_all
+    Object.send(:remove_const, :Library)
+    Object.send(:remove_const, :Book)
+  end
+
+  subject {Library.all} 
+  its(:class) {should eq ActiveFedora::Relation }
+
+  describe "#find" do
+    before do
+      Library.create
+      @library = Library.create
+    end
+    it "should find one of them" do
+      expect(subject.find(@library.id)).to eq @library
+    end
+    it "should find with a block" do
+      expect(subject.find { |l| l.id == @library.id}).to eq @library
+    end
+  end
+
+  describe "#select" do
+    before do
+      Library.create
+      @library = Library.create
+    end
+    it "should find with a block" do
+      expect(subject.select { |l| l.id == @library.id}).to eq [@library]
+    end
+  end
+end
+