This commit is contained in:
r888888888
2013-05-24 14:21:01 -07:00
parent 560aad8a2d
commit 677d6c11a5
2 changed files with 6 additions and 9 deletions

View File

@@ -34,17 +34,17 @@ module Danbooru
c = limit(records_per_page) c = limit(records_per_page)
if before_id.to_i > 0 if before_id.to_i > 0
c = c.where("id < ?", before_id.to_i) c = c.where("#{table_name}.id < ?", before_id.to_i)
end end
c.reorder("id desc").tap do |obj| c.reorder("#{table_name}.id desc").tap do |obj|
obj.extend(SequentialCollectionExtension) obj.extend(SequentialCollectionExtension)
obj.sequential_paginator_mode = :before obj.sequential_paginator_mode = :before
end end
end end
def paginate_sequential_after(after_id) def paginate_sequential_after(after_id)
limit(records_per_page).where("id > ?", after_id.to_i).reorder("id asc").tap do |obj| limit(records_per_page).where("#{table_name}.id > ?", after_id.to_i).reorder("#{table_name}.id asc").tap do |obj|
obj.extend(SequentialCollectionExtension) obj.extend(SequentialCollectionExtension)
obj.sequential_paginator_mode = :after obj.sequential_paginator_mode = :after
end end

View File

@@ -65,9 +65,8 @@ class DmailsControllerTest < ActionController::TestCase
end end
should "not show dmails not owned by the current user" do should "not show dmails not owned by the current user" do
assert_raise(User::PrivilegeError) do get :show, {:id => @dmail.id}, {:user_id => @unrelated_user.id}
get :show, {:id => @dmail.id}, {:user_id => @unrelated_user.id} assert_redirected_to(new_session_path(:url => "/dmails/#{@dmail.id}"))
end
end end
end end
@@ -95,9 +94,7 @@ class DmailsControllerTest < ActionController::TestCase
should "not allow deletion if the dmail is not owned by the current user" do should "not allow deletion if the dmail is not owned by the current user" do
assert_difference("Dmail.count", 0) do assert_difference("Dmail.count", 0) do
assert_raises(User::PrivilegeError) do post :destroy, {:id => @dmail.id}, {:user_id => @unrelated_user.id}
post :destroy, {:id => @dmail.id}, {:user_id => @unrelated_user.id}
end
end end
end end
end end