package gitbucket.core.service

import org.specs2.mutable.Specification


class WebHookServiceSpec extends Specification with ServiceSpecBase {
  lazy val service = new WebHookPullRequestService with AccountService with RepositoryService with PullRequestService with IssuesService

  "WebHookPullRequestService.getPullRequestsByRequestForWebhook" should {
    "find from request branch" in { withTestDB { implicit session =>
      val user1 = generateNewUserWithDBRepository("user1","repo1")
      val user2 = generateNewUserWithDBRepository("user2","repo2")
      val user3 = generateNewUserWithDBRepository("user3","repo3")
      val (issue1, pullreq1) = generateNewPullRequest("user1/repo1/master1", "user2/repo2/master2")
      val (issue3, pullreq3) = generateNewPullRequest("user3/repo3/master3", "user2/repo2/master2")
      val (issue32, pullreq32) = generateNewPullRequest("user3/repo3/master32", "user2/repo2/master2")
      generateNewPullRequest("user2/repo2/master2", "user1/repo1/master2")
      service.addWebHookURL("user1", "repo1", "webhook1-1")
      service.addWebHookURL("user1", "repo1", "webhook1-2")
      service.addWebHookURL("user2", "repo2", "webhook2-1")
      service.addWebHookURL("user2", "repo2", "webhook2-2")
      service.addWebHookURL("user3", "repo3", "webhook3-1")
      service.addWebHookURL("user3", "repo3", "webhook3-2")

      service.getPullRequestsByRequestForWebhook("user1","repo1","master1") must_== Map.empty

      var r = service.getPullRequestsByRequestForWebhook("user2","repo2","master2").mapValues(_.map(_.url).toSet)
      r.size must_== 3
      r((issue1, pullreq1, user1, user2)) must_== Set("webhook1-1","webhook1-2")
      r((issue3, pullreq3, user3, user2)) must_== Set("webhook3-1","webhook3-2")
      r((issue32, pullreq32, user3, user2)) must_== Set("webhook3-1","webhook3-2")

      // when closed, it not founds.
      service.updateClosed("user1","repo1",issue1.issueId, true)

      var r2 = service.getPullRequestsByRequestForWebhook("user2","repo2","master2").mapValues(_.map(_.url).toSet)
      r2.size must_== 2
      r2((issue3, pullreq3, user3, user2)) must_== Set("webhook3-1","webhook3-2")
      r2((issue32, pullreq32, user3, user2)) must_== Set("webhook3-1","webhook3-2")
    } }
  }
}