test/unit/helpers/pagination_helper_test.rb in paginary-0.0.1.pre1 vs test/unit/helpers/pagination_helper_test.rb in paginary-0.0.1.pre2

@@ -19,9 +19,16 @@
     builder = nil
     paginate(@relation) { |b| builder = b }
     assert_kind_of Paginary::Helpers::PageBuilder, builder
   end
   
+  test "paginate should accept builder class" do
+    custom_builder = Class.new(Paginary::Helpers::PageBuilder)
+    builder = nil
+    paginate(@relation, :builder => custom_builder) { |b| builder = b }
+    assert_kind_of custom_builder, builder
+  end
+  
   test "paginate should query database for total number of items only once" do
     num = count_queries do
       paginate(@relation) { |page| page.links; page.items.to_a }
     end
     assert_equal 2, num