Handle updates via git webhooks

test/api_git_test.go

@@ -0,0 +1,104 @@
+package test
+
+import (
+	"bytes"
+	"crypto"
+	"crypto/hmac"
+	"encoding/hex"
+	"fmt"
+	"net/http"
+	"src/task_tracker/api"
+	"src/task_tracker/config"
+	"testing"
+)
+
+func TestWebHookNoSignature(t *testing.T) {
+
+	r := Post("/git/receivehook", api.GitPayload{})
+
+	fmt.Println(r.StatusCode)
+
+	if r.StatusCode != 403 {
+		t.Error()
+	}
+}
+
+func TestWebHookInvalidSignature(t *testing.T) {
+
+	req, _ := http.NewRequest("POST", "http://"+config.Cfg.ServerAddr+"/git/receivehook", nil)
+	req.Header.Add("X-Hub-Signature", "invalid")
+
+	client := http.Client{}
+	r, _ := client.Do(req)
+
+	fmt.Println(r.StatusCode)
+
+	if r.StatusCode != 403 {
+		t.Error()
+	}
+}
+
+func TestWebHookDontUpdateVersion(t *testing.T) {
+
+	resp := createProject(api.CreateProjectRequest{
+		Name:    "My version should not be updated",
+		Version: "old",
+		GitRepo: "username/not_this_one",
+	})
+
+	body := []byte(`{"ref": "refs/heads/master", "after": "new", "repository": {"full_name": "username/repo_name"}}`)
+	bodyReader := bytes.NewReader(body)
+
+	mac := hmac.New(crypto.SHA1.New, config.Cfg.WebHookSecret)
+	mac.Write(body)
+	signature := hex.EncodeToString(mac.Sum(nil))
+	signature = "sha1=" + signature
+
+	req, _ := http.NewRequest("POST", "http://"+config.Cfg.ServerAddr+"/git/receivehook", bodyReader)
+	req.Header.Add("X-Hub-Signature", signature)
+
+	client := http.Client{}
+	r, _ := client.Do(req)
+
+	if r.StatusCode != 200 {
+		t.Error()
+	}
+
+	getResp, _ := getProject(resp.Id)
+
+	if getResp.Project.Version != "old" {
+		t.Error()
+	}
+}
+func TestWebHookUpdateVersion(t *testing.T) {
+
+	resp := createProject(api.CreateProjectRequest{
+		Name:    "My version should be updated",
+		Version: "old",
+		GitRepo: "username/repo_name",
+	})
+
+	body := []byte(`{"ref": "refs/heads/master", "after": "new", "repository": {"full_name": "username/repo_name"}}`)
+	bodyReader := bytes.NewReader(body)
+
+	mac := hmac.New(crypto.SHA1.New, config.Cfg.WebHookSecret)
+	mac.Write(body)
+	signature := hex.EncodeToString(mac.Sum(nil))
+	signature = "sha1=" + signature
+
+	req, _ := http.NewRequest("POST", "http://"+config.Cfg.ServerAddr+"/git/receivehook", bodyReader)
+	req.Header.Add("X-Hub-Signature", signature)
+
+	client := http.Client{}
+	r, _ := client.Do(req)
+
+	if r.StatusCode != 200 {
+		t.Error()
+	}
+
+	getResp, _ := getProject(resp.Id)
+
+	if getResp.Project.Version != "new" {
+		t.Error()
+	}
+}