require 'test_helper'

module Shipit
  class MergePullRequestsJobTest < ActiveSupport::TestCase
    setup do
      @stack = shipit_stacks(:shipit)
      @job = MergePullRequestsJob.new

      @pending_pr = shipit_pull_requests(:shipit_pending)
      @unmergeable_pr = shipit_pull_requests(:shipit_pending_unmergeable)
      @not_ready_pr = shipit_pull_requests(:shipit_pending_not_mergeable_yet)
      @closed_pr = shipit_pull_requests(:shipit_pending_closed)
      @merged_pr = shipit_pull_requests(:shipit_pending_merged)
    end

    test "#perform rejects unmergeable PRs and merge the others" do
      PullRequest.any_instance.stubs(:refresh!)
      FakeWeb.register_uri(:put, "#{@pending_pr.api_url}/merge", status: %w(200 OK), body: {
        sha: "6dcb09b5b57875f334f61aebed695e2e4193db5e",
        merged: true,
        message: "Pull Request successfully merged",
      }.to_json)
      branch_url = "https://api.github.com/repos/shopify/shipit-engine/git/refs/heads/feature-62"
      FakeWeb.register_uri(:delete, branch_url, status: %w(204 No content))
      pulls_url = "https://api.github.com/repos/shopify/shipit-engine/pulls?base=feature-62"
      FakeWeb.register_uri(:get, pulls_url, status: %w(200 OK), body: '[]')

      @job.perform(@stack)

      assert_predicate @pending_pr.reload, :merged?
      assert_predicate @unmergeable_pr.reload, :rejected?
    end

    test "#perform rejects PRs if the merge attempt fails" do
      PullRequest.any_instance.stubs(:refresh!)
      FakeWeb.register_uri(:put, "#{@pending_pr.api_url}/merge", status: %w(405 Method not allowed), body: {
        message: "Pull Request is not mergeable",
        documentation_url: "https://developer.github.com/v3/pulls/#merge-a-pull-request-merge-button",
      }.to_json)

      @job.perform(@stack)

      assert_predicate @pending_pr.reload, :rejected?
    end

    test "#perform rejects PRs but do not attempt to merge any if the stack doesn't allow merges" do
      PullRequest.any_instance.stubs(:refresh!)
      @stack.update!(lock_reason: 'Maintenance')
      @job.perform(@stack)
      assert_predicate @pending_pr.reload, :pending?
    end

    test "#perform schedules a new job if the first PR in the queue is not mergeable yet" do
      PullRequest.any_instance.stubs(:refresh!)

      @pending_pr.update!(mergeable: nil)
      assert_enqueued_with(job: MergePullRequestsJob) do
        @job.perform(@stack)
      end
      assert_predicate @pending_pr.reload, :pending?
    end

    test "#perform cancels merge requests for closed PRs" do
      @pending_pr.cancel!
      PullRequest.any_instance.stubs(:refresh!)
      @job.perform(@stack)
      assert_predicate @closed_pr.reload, :canceled?
    end

    test "#perform completes merge requests for already merged PRs" do
      @pending_pr.cancel!
      PullRequest.any_instance.stubs(:refresh!)
      @job.perform(@stack)
      assert_predicate @merged_pr.reload, :merged?
    end
  end
end