spec/lib/indexing/bundle_spec.rb in picky-1.3.0 vs spec/lib/indexing/bundle_spec.rb in picky-1.3.1

@@ -1,13 +1,13 @@
 require 'spec_helper'
 
 describe Indexing::Bundle do
 
   before(:each) do
-    @index         = stub :index, :name => :some_index
-    @category      = stub :category, :name => :some_category
-    @configuration = Configuration::Index.new @index, @category
+    @internal_index = stub :index, :name => :some_index
+    @category       = stub :category, :name => :some_category
+    @configuration  = Configuration::Index.new @internal_index, @category
     
     @partial     = stub :partial
     @weights     = stub :weights
     @similarity  = stub :similarity
     @index       = Indexing::Bundle.new :some_name, @configuration, @similarity, @partial, @weights
@@ -40,20 +40,50 @@
         @index.index[:some_token].should == :already_assigned
       end
     end
   end
   
-  # TODO
-  #
-  # describe 'retrieve' do
-  #   it 'should call the other methods correctly' do
-  #     results = stub :results
-  #     @index.stub! :execute_query => results
-  #     @index.should_receive(:extract).once.with results
-  #     
-  #     @index.retrieve
-  #   end
-  # end
+  describe 'retrieve' do
+    before(:each) do
+      files = stub :files
+      files.should_receive(:retrieve).once.and_yield '  1234', :some_token
+      @index.stub! :files => files
+      
+      @ary = stub :ary
+      @internal_index.should_receive(:[]).any_number_of_times.and_return @ary
+      @index.stub! :index => @internal_index
+    end
+    context 'id key format' do
+      before(:each) do
+        @index.should_receive(:[]).once.with(:key_format).and_return :to_i
+      end
+      it 'should call the other methods correctly' do
+        @ary.should_receive(:<<).once.with 1234
+        
+        @index.retrieve
+      end
+    end
+    context 'other key format' do
+      before(:each) do
+        @index.should_receive(:[]).once.with(:key_format).and_return :strip
+      end
+      it 'should call the other methods correctly' do
+        @ary.should_receive(:<<).once.with '1234'
+        
+        @index.retrieve
+      end
+    end
+    context 'no key format - default' do
+      before(:each) do
+        @index.should_receive(:[]).once.with(:key_format).and_return nil
+      end
+      it 'should call the other methods correctly' do
+        @ary.should_receive(:<<).once.with 1234
+        
+        @index.retrieve
+      end
+    end
+  end
 
   describe 'load_from_index_file' do
     it 'should call two methods in order' do
       @index.should_receive(:load_from_index_generation_message).once.ordered
       @index.should_receive(:clear).once.ordered
\ No newline at end of file