added post flag test

test/unit/post_flag_test.rb

@@ -1,8 +1,57 @@
 require 'test_helper'
 
 class PostFlagTest < ActiveSupport::TestCase
-  # Replace this with your real tests.
-  test "the truth" do
-    assert true
+  context "In all cases" do
+    setup do
+      @alice = Factory.create(:user)
+      CurrentUser.user = @alice
+      CurrentUser.ip_addr = "127.0.0.1"
+      MEMCACHE.flush_all
+    end
+
+    teardown do
+      CurrentUser.user = nil
+      CurrentUser.ip_addr = nil
+    end
+    
+    context "a user" do
+      setup do
+        @post = Factory.create(:post, :tag_string => "aaa")
+      end
+      
+      should "not be able to flag a post more than twice" do
+        assert_difference("PostFlag.count", 1) do
+          @post_flag = PostFlag.create(:post => @post, :reason => "aaa")
+        end
+        
+        assert_difference("PostFlag.count", 0) do
+          @post_flag = PostFlag.create(:post => @post, :reason => "aaa")
+        end
+        
+        assert_equal(["Creator has already flagged this post"], @post_flag.errors.full_messages)
+      end
+      
+      should "not be able to flag more than 10 posts in 24 hours" do
+        @post_flag = PostFlag.new(:post => @post, :reason => "aaa")
+        @post_flag.expects(:flag_count_for_creator).returns(10)
+        assert_difference("PostFlag.count", 0) do
+          @post_flag.save
+        end
+        assert_equal(["Creator can flag 10 posts a day"], @post_flag.errors.full_messages)
+      end
+      
+      should "not be able to flag a deleted post" do
+        @post.update_attribute(:is_deleted, true)
+        assert_difference("PostFlag.count", 0) do
+          @post_flag = PostFlag.create(:post => @post, :reason => "aaa")
+        end
+        assert_equal(["Post is deleted"], @post_flag.errors.full_messages)
+      end
+      
+      should "initialize its creator" do
+        @post_flag = PostFlag.create(:post => @post, :reason => "aaa")
+        assert_equal(@alice.id, @post_flag.creator_id)
+      end
+    end
   end
 end