#-- encoding: UTF-8
#-- copyright
# ChiliProject is a project management system.
#
# Copyright (C) 2010-2013 the ChiliProject Team
#
# This program is free software; you can redistribute it and/or
# modify it under the terms of the GNU General Public License
# as published by the Free Software Foundation; either version 2
# of the License, or (at your option) any later version.
#
# See doc/COPYRIGHT.rdoc for more details.
#++
require File.expand_path('../../test_helper', __FILE__)
require 'watchers_controller'

# Re-raise errors caught by the controller.
class WatchersController; def rescue_action(e) raise e end; end

class WatchersControllerTest < ActionController::TestCase
  fixtures :projects, :users, :roles, :members, :member_roles, :enabled_modules,
           :issues, :trackers, :projects_trackers, :issue_statuses, :enumerations, :watchers,
           :wikis, :wiki_pages

  def setup
    @controller = WatchersController.new
    @request    = ActionController::TestRequest.new
    @response   = ActionController::TestResponse.new
    User.current = nil
  end

  def test_get_watch_should_be_invalid
    @request.session[:user_id] = 3
    get :watch, :object_type => 'issue', :object_id => '1'
    assert_response 405
  end

  def test_watch
    @request.session[:user_id] = 3
    assert_difference('Watcher.count') do
      xhr :post, :watch, :object_type => 'issue', :object_id => '1'
      assert_response :success
      assert @response.body.include? "$$(\"#watcher\").each"
      assert @response.body.include? "value.replace"
    end
    assert Issue.find(1).watched_by?(User.find(3))
  end

  def test_watch_should_be_denied_without_permission
    Role.find(2).remove_permission! :view_issues
    @request.session[:user_id] = 3
    assert_no_difference('Watcher.count') do
      xhr :post, :watch, :object_type => 'issue', :object_id => '1'
      assert_response 403
    end
  end

  def test_watch_with_multiple_replacements
    @request.session[:user_id] = 3
    assert_difference('Watcher.count') do
      xhr :post, :watch, :object_type => 'issue', :object_id => '1', :replace => ['#watch_item_1','.watch_item_2']
      assert_response :success
      assert @response.body.include? "$$(\"#watch_item_1\").each"
      assert @response.body.include? "$$(\".watch_item_2\").each"
      assert @response.body.include? "value.replace"
    end
  end

  def test_watch_with_watchers_special_logic
    @request.session[:user_id] = 3
    assert_difference('Watcher.count') do
      xhr :post, :watch, :object_type => 'issue', :object_id => '1', :replace => ['#watchers', '.watcher']
      assert_response :success
      assert_select_rjs :replace_html, 'watchers'
      assert @response.body.include? "$$(\".watcher\").each"
      assert @response.body.include? "value.replace"
    end
  end

  def test_unwatch
    @request.session[:user_id] = 3
    assert_difference('Watcher.count', -1) do
      xhr :post, :unwatch, :object_type => 'issue', :object_id => '2'
      assert_response :success
      assert @response.body.include? "$$(\"#watcher\").each"
      assert @response.body.include? "value.replace"
    end
    assert !Issue.find(1).watched_by?(User.find(3))
  end

  def test_unwatch_with_multiple_replacements
    @request.session[:user_id] = 3
    assert_difference('Watcher.count', -1) do
      xhr :post, :unwatch, :object_type => 'issue', :object_id => '2', :replace => ['#watch_item_1', '.watch_item_2']
      assert_response :success
      assert @response.body.include? "$$(\"#watch_item_1\").each"
      assert @response.body.include? "$$(\".watch_item_2\").each"
      assert @response.body.include? "value.replace"
    end
    assert !Issue.find(1).watched_by?(User.find(3))
  end

  def test_unwatch_with_watchers_special_logic
    @request.session[:user_id] = 3
    assert_difference('Watcher.count', -1) do
      xhr :post, :unwatch, :object_type => 'issue', :object_id => '2', :replace => ['#watchers', '.watcher']
      assert_response :success
      assert_select_rjs :replace_html, 'watchers'
      assert @response.body.include? "$$(\".watcher\").each"
      assert @response.body.include? "value.replace"
    end
    assert !Issue.find(1).watched_by?(User.find(3))
  end

  def test_new_watcher
    @request.session[:user_id] = 2
    assert_difference('Watcher.count') do
      xhr :post, :new, :object_type => 'issue', :object_id => '2', :user_ids => ['4']
      assert_response :success
      assert_select_rjs :replace_html, 'watchers'
    end
    assert Issue.find(2).watched_by?(User.find(4))
  end

  def test_new_multiple_users
    @request.session[:user_id] = 2
    assert_difference('Watcher.count', 2) do
      xhr :post, :new, :object_type => 'issue', :object_id => '2', :user_ids => ['4','7']
      assert_response :success
      assert_select_rjs :replace_html, 'watchers'
    end
    assert Issue.find(2).watched_by?(User.find(4))
    assert Issue.find(2).watched_by?(User.find(7))
  end

  context "POST :new" do
    should "add groups" do
      @group = Group.generate!.reload
      Member.generate!(:project => Project.find(1), :roles => [Role.find(1)], :principal => @group)

      @request.session[:user_id] = 2
      assert_difference('Watcher.count') do
        xhr :post, :new, :object_type => 'issue', :object_id => '2', :user_ids => [@group.id.to_s]
        assert_response :success
        assert_select_rjs :replace_html, 'watchers'
      end
      assert Issue.find(2).watched_by?(@group)
    end

  end

  def test_new_multiple_users_watching_wiki_page
    Role.find(1).add_permission! :add_wiki_page_watchers

    @request.session[:user_id] = 2
    @page = WikiPage.find(1)
    assert !@page.watched_by?(User.find(2))
    assert !@page.watched_by?(User.find(4))
    assert !@page.watched_by?(User.find(7))

    assert_difference('Watcher.count', 3) do
      xhr :post, :new, :object_type => 'wiki_page', :object_id => '1', :user_ids => ['2','4','7']
      assert_response :success
      assert_select_rjs :replace_html, 'watchers'
    end
    @page.reload
    assert @page.watched_by?(User.find(2))
    assert @page.watched_by?(User.find(4))
    assert @page.watched_by?(User.find(7))
  end

  def test_new_multiple_users_watching_board
    Role.find(1).add_permission! :add_board_watchers

    @request.session[:user_id] = 2
    @project = Project.find(1)
    @board = Board.generate!(:project => @project)
    assert !@board.watched_by?(User.find(2))
    assert !@board.watched_by?(User.find(4))

    assert_difference('Watcher.count', 2) do
      xhr :post, :new, :object_type => 'board', :object_id => @board.id, :user_ids => ['2','4']
      assert_response :success
      assert_select_rjs :replace_html, 'watchers'
    end
    @board.reload
    assert @board.watched_by?(User.find(2))
    assert @board.watched_by?(User.find(4))
  end

  def test_new_multiple_users_watching_message
    Role.find(1).add_permission! :add_message_watchers

    @request.session[:user_id] = 2
    @project = Project.find(1)
    @board = Board.generate!(:project => @project)
    @message = Message.generate!(:board => @board)
    assert !@message.watched_by?(User.find(2))
    assert !@message.watched_by?(User.find(4))

    assert_difference('Watcher.count', 2) do
      xhr :post, :new, :object_type => 'message', :object_id => @message.id, :user_ids => ['2','4']
      assert_response :success
      assert_select_rjs :replace_html, 'watchers'
    end
    @message.reload
    assert @message.watched_by?(User.find(2))
    assert @message.watched_by?(User.find(4))
  end

  def test_new_issue_watcher_without_permission
    Role.find(1).remove_permission! :add_issue_watchers

    @request.session[:user_id] = 2
    assert_difference('Watcher.count',0) do
      xhr :post, :new, :object_type => 'issue', :object_id => '2', :user_ids => ['4']
      assert_response :forbidden
    end
    assert !Issue.find(2).watched_by?(User.find(4))

  end

  def test_new_wiki_page_watcher_without_permission
    Role.find(1).remove_permission! :add_wiki_page_watchers

    @request.session[:user_id] = 2
    @page = WikiPage.find(1)

    assert_difference('Watcher.count',0) do
      xhr :post, :new, :object_type => 'wiki_page', :object_id => '1', :user_ids => ['2']
      assert_response :forbidden
    end
    assert !WikiPage.find(1).watched_by?(User.find(2))

  end

  def test_new_board_watcher_without_permission
    Role.find(1).remove_permission! :add_board_watchers

    @request.session[:user_id] = 2
    @project = Project.find(1)
    @board = Board.generate!(:project => @project)

    assert_difference('Watcher.count',0) do
      xhr :post, :new, :object_type => 'board', :object_id => @board.id, :user_ids => ['2']
      assert_response :forbidden
    end
    assert !Board.find(@board.id).watched_by?(User.find(2))

  end

  def test_new_message_watcher_without_permission
    Role.find(1).remove_permission! :add_message_watchers

    @request.session[:user_id] = 2
    @project = Project.find(1)
    @board = Board.generate!(:project => @project)
    @message = Message.generate!(:board => @board)

    assert_difference('Watcher.count',0) do
      xhr :post, :new, :object_type => 'message', :object_id => @message.id, :user_ids => ['2']
      assert_response :forbidden
    end
    assert !Message.find(@message.id).watched_by?(User.find(2))

  end

  def test_remove_watcher
    Role.find(1).add_permission! :delete_issue_watchers

    @request.session[:user_id] = 2
    assert_difference('Watcher.count', -1) do
      xhr :post, :destroy, :object_type => 'issue', :object_id => '2', :user_id => '3'
      assert_response :success
      assert_select_rjs :replace_html, 'watchers'
    end
    assert !Issue.find(2).watched_by?(User.find(3))
  end

  context "POST :destroy" do
    should "remove a group" do
      @group = Group.generate!.reload
      Member.generate!(:project => Project.find(1), :roles => [Role.find(1)], :principal => @group)
      assert Issue.find(2).add_watcher(@group)
      assert Issue.find(2).watched_by?(@group)

      @request.session[:user_id] = 2
      assert_difference('Watcher.count', -1) do
        xhr :post, :destroy, :object_type => 'issue', :object_id => '2', :user_id => @group.id.to_s
        assert_response :success
        assert_select_rjs :replace_html, 'watchers'
      end
      assert !Issue.find(2).watched_by?(@group)
    end

  end

  def test_remove_wiki_page_watcher
    Role.find(1).add_permission! :delete_wiki_page_watchers

    @request.session[:user_id] = 2
    @page = WikiPage.find(1)
    Watcher.create!(:user_id => 2, :watchable => @page)
    assert @page.watched_by?(User.find(2))

    assert_difference('Watcher.count', -1) do
      xhr :post, :destroy, :object_type => 'wiki_page', :object_id => '1', :user_id => '2'
      assert_response :success
      assert_select_rjs :replace_html, 'watchers'
    end
    assert !WikiPage.find(1).watched_by?(User.find(2))
  end

  def test_remove_board_watcher
    Role.find(1).add_permission! :delete_board_watchers
    @project = Project.find(1)
    @board = Board.generate!(:project => @project)
    Watcher.create!(:user_id => 2, :watchable => @board)
    assert @board.watched_by?(User.find(2))

    @request.session[:user_id] = 2
    assert_difference('Watcher.count', -1) do
      xhr :post, :destroy, :object_type => 'board', :object_id => @board.id, :user_id => '2'
      assert_response :success
      assert_select_rjs :replace_html, 'watchers'
    end
    assert !Board.find(@board.id).watched_by?(User.find(2))
  end

  def test_remove_message_watcher
    Role.find(1).add_permission! :delete_message_watchers
    @project = Project.find(1)
    @board = Board.generate!(:project => @project)
    @message = Message.generate!(:board => @board)
    Watcher.create!(:user_id => 2, :watchable => @message)
    assert @message.watched_by?(User.find(2))

    @request.session[:user_id] = 2
    assert_difference('Watcher.count', -1) do
      xhr :post, :destroy, :object_type => 'message', :object_id => @message.id, :user_id => '2'
      assert_response :success
      assert_select_rjs :replace_html, 'watchers'
    end
    assert !Message.find(@message.id).watched_by?(User.find(2))
  end

  def test_remove_issue_watcher_without_permission
    Role.find(1).remove_permission! :delete_issue_watchers

    @request.session[:user_id] = 2
    assert_difference('Watcher.count',0) do
      xhr :post, :destroy, :object_type => 'issue', :object_id => '2', :user_id => '3'
      assert_response :forbidden
    end
    assert Issue.find(2).watched_by?(User.find(3))

  end

  def test_remove_wiki_page_watcher_without_permission
    Role.find(1).remove_permission! :delete_wiki_page_watchers

    @request.session[:user_id] = 2
    @page = WikiPage.find(1)
    Watcher.create!(:user_id => 2, :watchable => @page)
    assert @page.watched_by?(User.find(2))

    assert_difference('Watcher.count',0) do
      xhr :post, :destroy, :object_type => 'wiki_page', :object_id => '1', :user_id => '2'
      assert_response :forbidden
    end
    assert WikiPage.find(1).watched_by?(User.find(2))

  end

  def test_remove_board_watcher_without_permission
    Role.find(1).remove_permission! :delete_board_watchers
    @project = Project.find(1)
    @board = Board.generate!(:project => @project)
    Watcher.create!(:user_id => 2, :watchable => @board)
    assert @board.watched_by?(User.find(2))

    @request.session[:user_id] = 2
    assert_difference('Watcher.count', 0) do
      xhr :post, :destroy, :object_type => 'board', :object_id => @board.id, :user_id => '2'
      assert_response :forbidden
    end
    assert Board.find(@board.id).watched_by?(User.find(2))
  end

  def test_remove_message_watcher_without_permission
    Role.find(1).remove_permission! :delete_message_watchers
    @project = Project.find(1)
    @board = Board.generate!(:project => @project)
    @message = Message.generate!(:board => @board)
    Watcher.create!(:user_id => 2, :watchable => @message)
    assert @message.watched_by?(User.find(2))

    @request.session[:user_id] = 2
    assert_difference('Watcher.count', 0) do
      xhr :post, :destroy, :object_type => 'message', :object_id => @message.id, :user_id => '2'
      assert_response :forbidden
    end
    assert Message.find(@message.id).watched_by?(User.find(2))
  end
end
