bitbucket_hook_test.go

package git

import (
	"bytes"
	"net/http"
	"net/http/httptest"
	"testing"
	"time"
)

func TestBitbucketDeployPush(t *testing.T) {
	repo := &Repo{Branch: "master", Hook: HookConfig{URL: "/bitbucket_deploy"}}
	bbHook := BitbucketHook{}

	remoteIP := "18.246.31.128"
	atlassianIPsMu.Lock()
	atlassianIPs = atlassianIPResponse{
		Items: []atlassianIPRange{
			{
				Network: remoteIP,
				MaskLen: 25,
				CIDR:    remoteIP + "/25",
				Mask:    "255.255.255.128",
			},
			{
				Network: "2600:1f18:2146:e306:939f:d1b3:aa36:ac42",
				MaskLen: 56,
				CIDR:    "2600:1f18:2146:e300::/56",
				Mask:    "ffff:ffff:ffff:ff00::",
			},
		},
		lastUpdated: time.Now(),
	}
	atlassianIPsMu.Unlock()

	for i, test := range []struct {
		ip           string
		body         string
		event        string
		responseBody string
		code         int
	}{
		{remoteIP, "", "", "", 400},
		{"131.103.20.160", "", "", "", 403},
		{remoteIP, "", "repo:push", "", 400},
		{remoteIP, pushBBBodyValid, "repo:push", "", 200},
		{"2600:1f18:2146:e306:939f:d1b3:aa36:ac42", pushBBBodyValid, "repo:push", "", 200}, // test IPv6
		{remoteIP, pushBBBodyValid, "repo:push", "", 200},
		{remoteIP, pushBBBodyEmptyBranch, "repo:push", "", 400},
		{remoteIP, pushBBBodyDeleteBranch, "repo:push", "", 400},
	} {

		req, err := http.NewRequest("POST", "/bitbucket_deploy", bytes.NewBuffer([]byte(test.body)))
		if err != nil {
			t.Fatalf("Test %v: Could not create HTTP request: %v", i, err)
		}
		req.RemoteAddr = test.ip

		if test.event != "" {
			req.Header.Add("X-Event-Key", test.event)
		}

		rec := httptest.NewRecorder()

		code, err := bbHook.Handle(rec, req, repo)

		if code != test.code {
			t.Errorf("Test %d: Expected response code to be %d but was %d", i, test.code, code)
		}

		if rec.Body.String() != test.responseBody {
			t.Errorf("Test %d: Expected response body to be '%v' but was '%v'", i, test.responseBody, rec.Body.String())
		}
	}

}

var pushBBBodyEmptyBranch = `
{
	"push": {
		"changes": [
			{
				"new": {
					"type": "branch",
					"name": "",
					"target": {
						"hash": "709d658dc5b6d6afcd46049c2f332ee3f515a67d"
					}
				}
			}
		]
	}
}
`

var pushBBBodyValid = `
{
	"push": {
		"changes": [
			{
				"new": {
					"type": "branch",
					"name": "master",
					"target": {
						"hash": "709d658dc5b6d6afcd46049c2f332ee3f515a67d"
					}
				}
			}
		]
	}
}
`

var pushBBBodyDeleteBranch = `
{
	"push": {
		"changes": [
		]
	}
}
`