view blocks flow

spec/couchrest/helpers/pager_spec.rb

@@ -0,0 +1,122 @@
+require File.dirname(__FILE__) + '/../../spec_helper'
+
+describe CouchRest::Pager do
+  before(:all) do
+    @cr = CouchRest.new(COUCHHOST)
+    @db = @cr.database(TESTDB)
+    @db.delete! rescue nil
+    @db = @cr.create_db(TESTDB) rescue nil
+    @pager = CouchRest::Pager.new(@db)
+  end
+  
+  after(:all) do
+    begin
+      @db.delete!
+    rescue RestClient::Request::RequestFailed
+    end
+  end
+  
+  it "should store the db" do
+    @pager.db.should == @db
+  end
+  
+  describe "paging all docs" do
+    before(:all) do
+      @docs = []
+      100.times do |i|
+        @docs << ({:number => (i % 10)})
+      end
+      @db.bulk_save(@docs)
+    end
+    it "should yield total_docs / count times" do
+      n = 0
+      @pager.all_docs(10) do |doc|
+        n += 1
+      end
+      n.should == 10
+    end
+    it "should yield each docrow group without duplicate docs" do
+      docids = {}
+      @pager.all_docs(10) do |docrows|
+        docrows.each do |row|
+          docids[row['id']].should be_nil
+          docids[row['id']] = true
+        end
+      end      
+      docids.keys.length.should == 100
+    end
+    it "should yield each docrow group" do
+      @pager.all_docs(10) do |docrows|
+        doc = @db.get(docrows[0]['id'])
+        doc['number'].class.should == Fixnum
+      end      
+    end
+  end
+  
+  describe "Pager with a view and docs" do
+    before(:all) do
+      @docs = []
+      100.times do |i|
+        @docs << ({:number => (i % 10)})
+      end
+      @db.bulk_save(@docs)
+      @db.save({
+        '_id' => '_design/magic',
+        'views' => {
+          'number' => {
+            'map' => 'function(doc){emit(doc.number,null)}'
+          }
+        }
+      })
+    end
+    
+    it "should have docs" do
+      @docs.length.should == 100
+      @db.documents['rows'].length.should == 101
+    end
+    
+    it "should have a view" do
+      @db.view('magic/number', :count => 10)['rows'][0]['key'].should == 0
+    end
+    
+    it "should yield once per key" do
+      results = {}
+      @pager.key_reduce('magic/number', 20) do |k,vs|
+        results[k] = vs.length
+      end
+      results[0].should == 10
+      results[3].should == 10
+    end
+    
+    it "with a small step size should yield once per key" do
+      results = {}
+      @pager.key_reduce('magic/number', 7) do |k,vs|
+        results[k] = vs.length
+      end
+      results[0].should == 10
+      results[3].should == 10
+      results[9].should == 10
+    end
+    it "with a large step size should yield once per key" do
+      results = {}
+      @pager.key_reduce('magic/number', 1000) do |k,vs|
+        results[k] = vs.length
+      end
+      results[0].should == 10
+      results[3].should == 10
+      results[9].should == 10
+    end
+    it "with a begin and end should only yield in the range (and leave out the lastkey)" do
+      results = {}
+      @pager.key_reduce('magic/number', 1000, 4, 7) do |k,vs|
+        results[k] = vs.length
+      end
+      results[0].should be_nil
+      results[4].should == 10
+      results[6].should == 10
+      results[7].should be_nil
+      results[8].should be_nil
+      results[9].should be_nil
+    end
+  end
+end