Merge pull request #49 from code-check/branchOp

Add BranchOp

Author: shunjikonishi

src/main/scala/codecheck/github/api/GitHubAPI.scala

@@ -27,6 +27,7 @@ class GitHubAPI(token: String, client: AsyncHttpClient, tokenType: String = "tok
   with MilestoneOp
   with WebhookOp
   with CollaboratorOp
+  with BranchOp
 {
 
   private val endpoint = "https://api.github.com"

src/main/scala/codecheck/github/models/Branch.scala

@@ -0,0 +1,16 @@
+package codecheck.github.models
+import org.json4s.JValue
+
+case class BranchListItem(value: JValue) extends AbstractJson(value) {
+  def name = get("name")
+  lazy val commit = CommitInfo(value \ "commit")
+}
+
+case class CommitInfo(value: JValue) extends AbstractJson(value) {
+  def sha = get("sha")
+  def url = get("url")
+}
+
+case class Branch(value: JValue) extends AbstractJson(value) {
+  def name = get("name")
+}

src/main/scala/codecheck/github/operations/BranchOp.scala

@@ -0,0 +1,36 @@
+package codecheck.github.operations
+
+import java.net.URLEncoder
+import scala.concurrent.Future
+import scala.concurrent.ExecutionContext.Implicits.global
+import org.json4s.JArray
+import org.json4s.JString
+import org.json4s.JNothing
+
+import codecheck.github.api.GitHubAPI
+import codecheck.github.exceptions.NotFoundException
+import codecheck.github.models.BranchListItem
+import codecheck.github.models.Branch
+
+trait BranchOp {
+  self: GitHubAPI =>
+
+  def listBranches(owner: String, repo: String): Future[List[BranchListItem]] = {
+    exec("GET", s"/repos/$owner/$repo/branches").map(
+      _.body match {
+        case JArray(arr) => arr.map(v => BranchListItem(v))
+        case _ => throw new IllegalStateException()
+      }
+    )
+  }
+
+  def getBranch(owner: String, repo: String, branch: String): Future[Option[Branch]] = {
+    exec("GET", s"/repos/$owner/$repo/branches/$branch", fail404=false).map{ res =>
+      res.statusCode match {
+        case 404 => None
+        case 200 => Some(Branch(res.body))
+      }
+    }
+  }
+
+}

src/test/scala/BranchOpSpec.scala

@@ -0,0 +1,34 @@
+import org.scalatest.FunSpec
+import org.scalatest.BeforeAndAfterAll
+import codecheck.github.exceptions.NotFoundException
+import codecheck.github.models._
+import codecheck.github.exceptions.GitHubAPIException
+import codecheck.github.exceptions.NotFoundException
+import scala.concurrent.Await
+import scala.concurrent.ExecutionContext.Implicits.global
+import codecheck.github.models.UserInput
+
+class BranchOpSpec extends FunSpec
+  with Constants
+  with BeforeAndAfterAll
+{
+  describe("getBranch") {
+    it("with valid repo and branch should succeed") {
+      val branchOp = Await.result(api.getBranch(user, userRepo, "master"), TIMEOUT)
+      assert(branchOp.isDefined)
+      assert(branchOp.get.name == "master")
+    }
+    it("with invalid branch should be None") {
+      val branchOp = Await.result(api.getBranch(user, userRepo, "unknown"), TIMEOUT)
+      assert(branchOp.isEmpty)
+    }
+  }
+
+  describe("listBranches") {
+    it("with valid repo should succeed") {
+      val list = Await.result(api.listBranches(user, userRepo), TIMEOUT)
+      assert(list.length > 0)
+      assert(list.exists(_.name == "master"))
+    }
+  }
+}