package gitbucket.core.service

import gitbucket.core.model._
import org.scalatest.FunSpec

class PullRequestServiceSpec
    extends FunSpec
    with ServiceSpecBase
    with PullRequestService
    with IssuesService
    with AccountService
    with RepositoryService
    with CommitsService
    with LabelsService
    with MilestonesService
    with PrioritiesService {

  def swap(r: (Issue, PullRequest)) = (r._2 -> r._1)

  describe("PullRequestService.getPullRequestFromBranch") {
    it("""should
    |return pull request if exists pull request from `branch` to `defaultBranch` and not closed.
    |return pull request if exists pull request from `branch` to other branch and not closed.
    |return None if all pull request is closed""".stripMargin.trim) {
      withTestDB { implicit se =>
        generateNewUserWithDBRepository("user1", "repo1")
        generateNewUserWithDBRepository("user1", "repo2")
        generateNewUserWithDBRepository("user2", "repo1")
        generateNewPullRequest("user1/repo1/master", "user1/repo1/head2") // not target branch
        generateNewPullRequest("user1/repo1/head1", "user1/repo1/master") // not target branch ( swap from, to )
        generateNewPullRequest("user1/repo1/master", "user2/repo1/head1") // other user
        generateNewPullRequest("user1/repo1/master", "user1/repo2/head1") // other repository
        val r1 = swap(generateNewPullRequest("user1/repo1/master2", "user1/repo1/head1"))
        val r2 = swap(generateNewPullRequest("user1/repo1/master", "user1/repo1/head1"))
        val r3 = swap(generateNewPullRequest("user1/repo1/master4", "user1/repo1/head1"))
        assert(getPullRequestFromBranch("user1", "repo1", "head1", "master") == Some(r2))
        updateClosed("user1", "repo1", r2._1.issueId, true)
        assert(Seq(r1, r2).contains(getPullRequestFromBranch("user1", "repo1", "head1", "master").get))
        updateClosed("user1", "repo1", r1._1.issueId, true)
        updateClosed("user1", "repo1", r3._1.issueId, true)
        assert(getPullRequestFromBranch("user1", "repo1", "head1", "master") == None)
      }
    }
  }
}