90 lines
2.1 KiB
Ruby
90 lines
2.1 KiB
Ruby
require 'test_helper'
|
|
|
|
class PostViewCountServiceTest < ActiveSupport::TestCase
|
|
def setup
|
|
super
|
|
|
|
CurrentUser.user = FactoryBot.create(:user)
|
|
CurrentUser.ip_addr = "127.0.0.1"
|
|
|
|
PostViewCountService.stubs(:enabled?).returns(true)
|
|
Danbooru.config.stubs(:reportbooru_server).returns("http://localhost:1234")
|
|
@post = FactoryBot.create(:post)
|
|
end
|
|
|
|
def teardown
|
|
super
|
|
CurrentUser.user = nil
|
|
CurrentUser.ip_addr = nil
|
|
end
|
|
|
|
subject { PostViewCountService.new }
|
|
|
|
context "#popular_posts" do
|
|
setup do
|
|
subject.stubs(:fetch_rank).returns([[@post.id, 1]])
|
|
end
|
|
|
|
should "return the posts" do
|
|
posts = subject.popular_posts
|
|
assert_equal(@post.id, posts[0].id)
|
|
end
|
|
end
|
|
|
|
context "#fetch_rank" do
|
|
context "success" do
|
|
setup do
|
|
@date = "2000-01-01"
|
|
@body = "[[1,1.0],[2,2.0]]"
|
|
stub_request(:get, "localhost:1234/post_views/rank").with(query: {"date" => @date}).to_return(body: @body)
|
|
end
|
|
|
|
should "return a list" do
|
|
json = subject.fetch_rank(@date)
|
|
assert(json.is_a?(Array))
|
|
assert_equal(1, json[0][0])
|
|
assert_equal(2, json[1][0])
|
|
end
|
|
end
|
|
|
|
context "failure" do
|
|
setup do
|
|
@date = "2000-01-01"
|
|
stub_request(:get, "localhost:1234/post_views/rank").with(query: {"date" => @date}).to_return(body: "", status: 400)
|
|
end
|
|
|
|
should "return nil" do
|
|
json = subject.fetch_rank(@date)
|
|
assert_nil(json)
|
|
end
|
|
end
|
|
end
|
|
|
|
context "#fetch_count" do
|
|
context "success" do
|
|
setup do
|
|
@body = "[[1,5],[2,20]]"
|
|
stub_request(:get, "localhost:1234/post_views/#{@post.id}").to_return(body: @body)
|
|
end
|
|
|
|
should "return a list" do
|
|
json = subject.fetch_count(@post.id)
|
|
assert(json.is_a?(Array))
|
|
assert_equal(1, json[0][0])
|
|
assert_equal(2, json[1][0])
|
|
end
|
|
end
|
|
|
|
context "failure" do
|
|
setup do
|
|
stub_request(:get, "localhost:1234/post_views/#{@post.id}").to_return(body: "", status: 400)
|
|
end
|
|
|
|
should "return nil" do
|
|
json = subject.fetch_count(@post.id)
|
|
assert_nil(json)
|
|
end
|
|
end
|
|
end
|
|
end
|