Add tests for following new threads on a post

Author: trichoplax

test/comments_test_helpers.rb

@@ -71,6 +71,18 @@ def try_unfollow_thread(thread)
     post :thread_unrestrict, params: { id: thread.id, type: 'follow' }
   end
 
+  # Attempts to follow new threads on a given post
+  # @param post [Post] post to follow
+  def try_post_follow(test_post)
+    post :post_follow, params: { post_id: test_post.id }
+  end
+
+  # Attempts to unfollow new threads on a given post
+  # @param post [Post] post to unfollow
+  def try_post_unfollow(test_post)
+    post :post_follow, params: { post_id: test_post.id }
+  end
+
   # Attempts to lock a given comment thread
   # @param thread [CommentThread] thread to lock
   # @param duration [Integer] lock duration, in days

test/controllers/comments/post_follow_test.rb

@@ -0,0 +1,34 @@
+require 'test_helper'
+require 'comments_test_helpers'
+
+class CommentsControllerTest < ActionController::TestCase
+  test 'post follower can unfollow post' do
+    user = users(:standard_user)
+    sign_in user
+    question = posts(:question_one)
+
+    # Assert user follows post
+    assert_equal 1, ThreadFollower.where(['post_id = ? AND user_id = ?', question, user]).count
+
+    try_post_unfollow(question)
+    assert_response(:found)
+
+    # Assert user does not follow post
+    assert_equal 0, ThreadFollower.where(['post_id = ? AND user_id = ?', question, user]).count
+  end
+
+  test 'non-follower can follow post' do
+    user = users(:basic_user)
+    sign_in user
+    question = posts(:question_one)
+
+    # Assert user does not follow post
+    assert_equal 0, ThreadFollower.where(['post_id = ? AND user_id = ?', question, user]).count
+
+    try_post_follow(question)
+    assert_response(:found)
+
+    # Assert user follows post
+    assert_equal 1, ThreadFollower.where(['post_id = ? AND user_id = ?', question, user]).count
+  end
+end

test/fixtures/thread_followers.yml

@@ -5,3 +5,7 @@ deleter_normal:
 standard_author_normal:
   user: standard_user
   comment_thread: normal
+
+standard_author_question_one:
+  user: standard_user
+  post: question_one