From 5726bc43ddde9ca69fb6f861b5f2d7efefbe5fdb Mon Sep 17 00:00:00 2001
From: r888888888 <r888888888@gmail.com>
Date: Thu, 16 Nov 2017 16:27:52 -0800
Subject: [PATCH] add test for PostViewCountService

---
 test/unit/post_view_count_service_test.rb | 86 +++++++++++++++++++++++
 1 file changed, 86 insertions(+)

diff --git a/test/unit/post_view_count_service_test.rb b/test/unit/post_view_count_service_test.rb
index c33d6ebce..3099bf005 100644
--- a/test/unit/post_view_count_service_test.rb
+++ b/test/unit/post_view_count_service_test.rb
@@ -1,2 +1,88 @@
+require 'test_helper'
+
 class PostViewCountServiceTest < ActiveSupport::TestCase
+  def setup
+    super
+
+    CurrentUser.user = FactoryGirl.create(:user)
+    CurrentUser.ip_addr = "127.0.0.1"
+
+    PostViewCountService.stubs(:enabled?).returns(true)
+    Danbooru.config.stubs(:reportbooru_server).returns("http://localhost:1234")
+    @post = FactoryGirl.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
+        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