test resumption tokens to exercise partial lists

Author: barmintor

lib/oai/provider/model.rb

@@ -29,7 +29,7 @@ module OAI::Provider
   # see the ResumptionToken class for more details.
   #
   class Model
-    attr_reader :timestamp_field, :identifier_field
+    attr_reader :timestamp_field, :identifier_field, :limit
 
     def initialize(limit = nil, timestamp_field = 'updated_at', identifier_field = 'id')
       @limit = limit

test/provider/tc_functional_tokens.rb

@@ -5,6 +5,11 @@ class ResumptionTokenFunctionalTest < Test::Unit::TestCase
 
   def setup
     @provider = ComplexProvider.new
+    @provider.model.instance_variable_set(:@limit, 120)
+  end
+
+  def teardown
+    @provider.model.instance_variable_set(:@limit, 100)
   end
 
   def test_resumption_tokens
@@ -13,15 +18,15 @@ def test_resumption_tokens
     end
     doc = Document.new(@provider.list_records(:metadata_prefix => 'oai_dc'))
     assert_not_nil doc.elements["/OAI-PMH/ListRecords/resumptionToken"]
-    assert_equal 101, doc.elements["/OAI-PMH/ListRecords"].to_a.size
+    assert_equal (@provider.model.limit + 1), doc.elements["/OAI-PMH/ListRecords"].to_a.size
     token = doc.elements["/OAI-PMH/ListRecords/resumptionToken"].text
     doc = Document.new(@provider.list_records(:resumption_token => token))
     assert_not_nil doc.elements["/OAI-PMH/ListRecords/resumptionToken"]
-    assert_equal 101, doc.elements["/OAI-PMH/ListRecords"].to_a.size
+    assert_equal (@provider.model.limit + 1), doc.elements["/OAI-PMH/ListRecords"].to_a.size
   end
 
   def test_from_and_until_with_resumption_tokens
-    # Should return 300 records broken into 3 groups of 100.
+    # Should return 300 records broken into 3 groups of 120, 120, and 60.
     assert_nothing_raised do
       Document.new(@provider.list_records(:metadata_prefix => 'oai_dc'))
     end
@@ -31,17 +36,19 @@ def test_from_and_until_with_resumption_tokens
         :from => Time.parse("September 1 2004"),
         :until => Time.parse("November 30 2004"))
       )
-    assert_equal 101, doc.elements["/OAI-PMH/ListRecords"].to_a.size
+    assert_equal (@provider.model.limit + 1), doc.elements["/OAI-PMH/ListRecords"].to_a.size
+    assert_not_nil doc.elements["/OAI-PMH/ListRecords/resumptionToken"]
     token = doc.elements["/OAI-PMH/ListRecords/resumptionToken"].text
 
     doc = Document.new(@provider.list_records(:resumption_token => token))
     assert_not_nil doc.elements["/OAI-PMH/ListRecords/resumptionToken"]
-    assert_equal 101, doc.elements["/OAI-PMH/ListRecords"].to_a.size
+    assert_equal (@provider.model.limit + 1), doc.elements["/OAI-PMH/ListRecords"].to_a.size
     token = doc.elements["/OAI-PMH/ListRecords/resumptionToken"].text
 
     doc = Document.new(@provider.list_records(:resumption_token => token))
     assert_nil doc.elements["/OAI-PMH/ListRecords/resumptionToken"]
-    assert_equal 100, doc.elements["/OAI-PMH/ListRecords"].to_a.size
+    assert_equal (300 % @provider.model.limit), doc.elements["/OAI-PMH/ListRecords"].to_a.size
+    token = doc.elements["/OAI-PMH/ListRecords/resumptionToken"].text
   end
 
 end