danbooru/test/unit/note_test.rb

require 'test_helper'

class NoteTest < ActiveSupport::TestCase
  context "In all cases" do
    setup do
      @user = FactoryGirl.create(:user)
      CurrentUser.user = @user
      CurrentUser.ip_addr = "127.0.0.1"
      MEMCACHE.flush_all
    end

    teardown do
      CurrentUser.user = nil
      CurrentUser.ip_addr = nil
    end

    context "for a post that already has a note" do
      setup do
        @post = FactoryGirl.create(:post)
        @note = FactoryGirl.create(:note, :post => @post)
      end

      context "when the note is deleted the post" do
        setup do
          @note.toggle!(:is_active)
        end

        should "null out its last_noted_at_field" do
          @post.reload
          assert_nil(@post.last_noted_at)
        end
      end
    end

    context "creating a note" do
      setup do
        @post = FactoryGirl.create(:post, :image_width => 1000, :image_height => 1000)
      end

      should "not validate if the note is outside the image" do
        @note = FactoryGirl.build(:note, :x => 1001, :y => 500, :post => @post)
        @note.save
        assert_equal(["Note must be inside the image"], @note.errors.full_messages)
      end

      should "not validate if the note is larger than the image" do
        @note = FactoryGirl.build(:note, :x => 500, :y => 500, :height => 501, :width => 500, :post => @post)
        @note.save
        assert_equal(["Note must be inside the image"], @note.errors.full_messages)
      end

      should "not validate if the post does not exist" do
        @note = FactoryGirl.build(:note, :x => 500, :y => 500, :post_id => -1)
        @note.save
        assert_equal(["Post must exist"], @note.errors.full_messages)
      end

      should "create a version" do
        assert_difference("NoteVersion.count", 1) do
          Timecop.travel(1.day.from_now) do
            @note = FactoryGirl.create(:note, :post => @post)
          end
        end

        assert_equal(1, @note.versions.count)
        assert_equal(@note.body, @note.versions.first.body)
        assert_equal(1, @note.version)
        assert_equal(1, @note.versions.first.version)
      end

      should "update the post's last_noted_at field" do
        assert_nil(@post.last_noted_at)
        @note = FactoryGirl.create(:note, :post => @post)
        @post.reload
        assert_not_nil(@post.last_noted_at)
      end

      context "for a note-locked post" do
        setup do
          @post.update_attribute(:is_note_locked, true)
        end

        should "fail" do
          assert_difference("Note.count", 0) do
            @note = FactoryGirl.build(:note, :post => @post)
            @note.save
          end
          assert_equal(["Post is note locked"], @note.errors.full_messages)
        end
      end
    end

    context "updating a note" do
      setup do
        @post = FactoryGirl.create(:post, :image_width => 1000, :image_height => 1000)
        @note = FactoryGirl.create(:note, :post => @post)
      end

      should "increment the updater's note_update_count" do
        assert_difference("CurrentUser.note_update_count", 1) do
          @note.update_attributes(:body => "zzz")
        end
      end

      should "update the post's last_noted_at field" do
        assert_nil(@post.last_noted_at)
        @note.update_attributes(:x => 500)
        @post.reload
        assert_equal(@post.last_noted_at.to_i, @note.updated_at.to_i)
      end

      should "create a version" do
        assert_difference("NoteVersion.count", 1) do
          Timecop.travel(1.day.from_now) do
            @note.update_attributes(:body => "fafafa")
          end
        end
        assert_equal(2, @note.versions.count)
        assert_equal(2, @note.versions.last.version)
        assert_equal("fafafa", @note.versions.last.body)
        assert_equal(2, @note.version)
      end

      context "for a note-locked post" do
        setup do
          @post.update_attribute(:is_note_locked, true)
        end

        should "fail" do
          @note.update_attributes(:x => 500)
          assert_equal(["Post is note locked"], @note.errors.full_messages)
        end
      end
    end

    context "when notes have been vandalized by one user" do
      setup do
        @vandal = FactoryGirl.create(:user)
        @note = FactoryGirl.create(:note, :x => 5, :y => 5)
        CurrentUser.scoped(@vandal, "127.0.0.1") do
          @note.update_attributes(:x => 10, :y => 10)
        end
      end

      context "the act of undoing all changes by that user" do
        should "revert any affected notes" do
          Note.undo_changes_by_user(@vandal.id)
          @note.reload
          assert_equal(5, @note.x)
          assert_equal(5, @note.y)
        end
      end
    end

    context "searching for a note" do
      setup do
        @note = FactoryGirl.create(:note, :body => "aaa")
      end

      context "where the body contains the string 'aaa'" do
        should "return a hit" do
          assert_equal(1, Note.body_matches("aaa").count)
        end
      end

      context "where the body contains the string 'bbb'" do
        should "return no hits" do
          assert_equal(0, Note.body_matches("bbb").count)
        end
      end
    end
  end
end