f67374da83
* Refactored the favorites code a bit. Adding a favorite from either an user or a post now works and will update all necessary records.
49 lines
1.3 KiB
Ruby
49 lines
1.3 KiB
Ruby
require_relative '../test_helper'
|
|
|
|
class FavoriteTest < ActiveSupport::TestCase
|
|
setup do
|
|
user = Factory.create(:user)
|
|
CurrentUser.user = user
|
|
CurrentUser.ip_addr = "127.0.0.1"
|
|
MEMCACHE.flush_all
|
|
Favorite # need to force loading the favorite model
|
|
end
|
|
|
|
teardown do
|
|
CurrentUser.user = nil
|
|
CurrentUser.ip_addr = nil
|
|
end
|
|
|
|
context "A favorite" do
|
|
should "know which table it belongs to" do
|
|
user1 = Factory.create(:user)
|
|
user2 = Factory.create(:user)
|
|
p1 = Factory.create(:post)
|
|
p2 = Factory.create(:post)
|
|
|
|
user1.add_favorite!(p1)
|
|
user1.add_favorite!(p2)
|
|
user2.add_favorite!(p1)
|
|
|
|
favorites = user1.favorites.order("id desc")
|
|
assert_equal(2, favorites.count)
|
|
assert_equal(p2.id, favorites[0].post_id)
|
|
assert_equal(p1.id, favorites[1].post_id)
|
|
|
|
favorites = user2.favorites.order("id desc")
|
|
assert_equal(1, favorites.count)
|
|
assert_equal(p1.id, favorites[0].post_id)
|
|
end
|
|
|
|
should "not allow duplicates" do
|
|
user1 = Factory.create(:user)
|
|
p1 = Factory.create(:post)
|
|
p2 = Factory.create(:post)
|
|
user1.add_favorite!(p1)
|
|
user1.add_favorite!(p1)
|
|
|
|
assert_equal(1, user1.favorites.count)
|
|
end
|
|
end
|
|
end
|