2016-12-20 17:45:22 +00:00
|
|
|
package lfsapi
|
|
|
|
|
|
|
|
import (
|
2016-12-20 22:37:11 +00:00
|
|
|
"encoding/json"
|
2016-12-22 21:35:57 +00:00
|
|
|
"fmt"
|
2016-12-20 22:37:11 +00:00
|
|
|
"net/http"
|
|
|
|
"net/http/httptest"
|
|
|
|
"sync/atomic"
|
2016-12-20 17:45:22 +00:00
|
|
|
"testing"
|
|
|
|
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
)
|
|
|
|
|
2016-12-20 22:37:11 +00:00
|
|
|
type redirectTest struct {
|
|
|
|
Test string
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestClientRedirect(t *testing.T) {
|
|
|
|
var called1 uint32
|
|
|
|
var called2 uint32
|
|
|
|
srv2 := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
|
|
atomic.AddUint32(&called2, 1)
|
|
|
|
t.Logf("srv2 req %s %s", r.Method, r.URL.Path)
|
|
|
|
assert.Equal(t, "POST", r.Method)
|
|
|
|
|
|
|
|
switch r.URL.Path {
|
|
|
|
case "/ok":
|
2016-12-21 17:06:55 +00:00
|
|
|
assert.Equal(t, "", r.Header.Get("Authorization"))
|
|
|
|
assert.Equal(t, "1", r.Header.Get("A"))
|
2016-12-20 22:37:11 +00:00
|
|
|
body := &redirectTest{}
|
|
|
|
err := json.NewDecoder(r.Body).Decode(body)
|
|
|
|
assert.Nil(t, err)
|
|
|
|
assert.Equal(t, "External", body.Test)
|
|
|
|
|
|
|
|
w.WriteHeader(200)
|
|
|
|
default:
|
|
|
|
w.WriteHeader(404)
|
|
|
|
}
|
|
|
|
}))
|
|
|
|
|
|
|
|
srv1 := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
|
|
atomic.AddUint32(&called1, 1)
|
|
|
|
t.Logf("srv1 req %s %s", r.Method, r.URL.Path)
|
|
|
|
assert.Equal(t, "POST", r.Method)
|
|
|
|
|
|
|
|
switch r.URL.Path {
|
|
|
|
case "/local":
|
|
|
|
w.Header().Set("Location", "/ok")
|
|
|
|
w.WriteHeader(307)
|
|
|
|
case "/external":
|
|
|
|
w.Header().Set("Location", srv2.URL+"/ok")
|
|
|
|
w.WriteHeader(307)
|
|
|
|
case "/ok":
|
2016-12-21 17:06:55 +00:00
|
|
|
assert.Equal(t, "auth", r.Header.Get("Authorization"))
|
|
|
|
assert.Equal(t, "1", r.Header.Get("A"))
|
2016-12-20 22:37:11 +00:00
|
|
|
body := &redirectTest{}
|
|
|
|
err := json.NewDecoder(r.Body).Decode(body)
|
|
|
|
assert.Nil(t, err)
|
|
|
|
assert.Equal(t, "Local", body.Test)
|
|
|
|
|
|
|
|
w.WriteHeader(200)
|
|
|
|
default:
|
|
|
|
w.WriteHeader(404)
|
|
|
|
}
|
|
|
|
}))
|
|
|
|
defer srv1.Close()
|
|
|
|
defer srv2.Close()
|
|
|
|
|
|
|
|
c := &Client{}
|
|
|
|
|
|
|
|
// local redirect
|
|
|
|
req, err := http.NewRequest("POST", srv1.URL+"/local", nil)
|
|
|
|
require.Nil(t, err)
|
2016-12-21 17:06:55 +00:00
|
|
|
req.Header.Set("Authorization", "auth")
|
|
|
|
req.Header.Set("A", "1")
|
|
|
|
|
2016-12-20 22:37:11 +00:00
|
|
|
require.Nil(t, MarshalToRequest(req, &redirectTest{Test: "Local"}))
|
|
|
|
|
|
|
|
res, err := c.Do(req)
|
|
|
|
require.Nil(t, err)
|
|
|
|
assert.Equal(t, 200, res.StatusCode)
|
|
|
|
assert.EqualValues(t, 2, called1)
|
|
|
|
assert.EqualValues(t, 0, called2)
|
|
|
|
|
|
|
|
// external redirect
|
|
|
|
req, err = http.NewRequest("POST", srv1.URL+"/external", nil)
|
|
|
|
require.Nil(t, err)
|
2016-12-21 17:06:55 +00:00
|
|
|
req.Header.Set("Authorization", "auth")
|
|
|
|
req.Header.Set("A", "1")
|
|
|
|
|
2016-12-20 22:37:11 +00:00
|
|
|
require.Nil(t, MarshalToRequest(req, &redirectTest{Test: "External"}))
|
|
|
|
|
|
|
|
res, err = c.Do(req)
|
|
|
|
require.Nil(t, err)
|
|
|
|
assert.Equal(t, 200, res.StatusCode)
|
|
|
|
assert.EqualValues(t, 3, called1)
|
|
|
|
assert.EqualValues(t, 1, called2)
|
|
|
|
}
|
|
|
|
|
2016-12-20 17:45:22 +00:00
|
|
|
func TestNewClient(t *testing.T) {
|
2017-01-06 18:34:43 +00:00
|
|
|
c, err := NewClient(TestEnv(map[string]string{}), TestEnv(map[string]string{
|
2016-12-20 17:45:22 +00:00
|
|
|
"lfs.dialtimeout": "151",
|
|
|
|
"lfs.keepalive": "152",
|
|
|
|
"lfs.tlstimeout": "153",
|
|
|
|
"lfs.concurrenttransfers": "154",
|
|
|
|
}))
|
|
|
|
|
|
|
|
require.Nil(t, err)
|
|
|
|
assert.Equal(t, 151, c.DialTimeout)
|
|
|
|
assert.Equal(t, 152, c.KeepaliveTimeout)
|
|
|
|
assert.Equal(t, 153, c.TLSTimeout)
|
|
|
|
assert.Equal(t, 154, c.ConcurrentTransfers)
|
|
|
|
}
|
2016-12-20 18:22:20 +00:00
|
|
|
|
|
|
|
func TestNewClientWithGitSSLVerify(t *testing.T) {
|
|
|
|
c, err := NewClient(nil, nil)
|
|
|
|
assert.Nil(t, err)
|
|
|
|
assert.False(t, c.SkipSSLVerify)
|
|
|
|
|
|
|
|
for _, value := range []string{"true", "1", "t"} {
|
2017-01-06 18:34:43 +00:00
|
|
|
c, err = NewClient(TestEnv(map[string]string{}), TestEnv(map[string]string{
|
2016-12-20 18:22:20 +00:00
|
|
|
"http.sslverify": value,
|
|
|
|
}))
|
|
|
|
t.Logf("http.sslverify: %q", value)
|
|
|
|
assert.Nil(t, err)
|
|
|
|
assert.False(t, c.SkipSSLVerify)
|
|
|
|
}
|
|
|
|
|
|
|
|
for _, value := range []string{"false", "0", "f"} {
|
2017-01-06 18:34:43 +00:00
|
|
|
c, err = NewClient(TestEnv(map[string]string{}), TestEnv(map[string]string{
|
2016-12-20 18:22:20 +00:00
|
|
|
"http.sslverify": value,
|
|
|
|
}))
|
|
|
|
t.Logf("http.sslverify: %q", value)
|
|
|
|
assert.Nil(t, err)
|
|
|
|
assert.True(t, c.SkipSSLVerify)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestNewClientWithOSSSLVerify(t *testing.T) {
|
|
|
|
c, err := NewClient(nil, nil)
|
|
|
|
assert.Nil(t, err)
|
|
|
|
assert.False(t, c.SkipSSLVerify)
|
|
|
|
|
|
|
|
for _, value := range []string{"false", "0", "f"} {
|
2017-01-06 18:34:43 +00:00
|
|
|
c, err = NewClient(TestEnv(map[string]string{
|
2016-12-20 18:22:20 +00:00
|
|
|
"GIT_SSL_NO_VERIFY": value,
|
2017-01-06 18:34:43 +00:00
|
|
|
}), TestEnv(map[string]string{}))
|
2016-12-20 18:22:20 +00:00
|
|
|
t.Logf("GIT_SSL_NO_VERIFY: %q", value)
|
|
|
|
assert.Nil(t, err)
|
|
|
|
assert.False(t, c.SkipSSLVerify)
|
|
|
|
}
|
|
|
|
|
|
|
|
for _, value := range []string{"true", "1", "t"} {
|
2017-01-06 18:34:43 +00:00
|
|
|
c, err = NewClient(TestEnv(map[string]string{
|
2016-12-20 18:22:20 +00:00
|
|
|
"GIT_SSL_NO_VERIFY": value,
|
2017-01-06 18:34:43 +00:00
|
|
|
}), TestEnv(map[string]string{}))
|
2016-12-20 18:22:20 +00:00
|
|
|
t.Logf("GIT_SSL_NO_VERIFY: %q", value)
|
|
|
|
assert.Nil(t, err)
|
|
|
|
assert.True(t, c.SkipSSLVerify)
|
|
|
|
}
|
|
|
|
}
|
2016-12-22 21:35:57 +00:00
|
|
|
|
|
|
|
func TestNewRequest(t *testing.T) {
|
|
|
|
tests := [][]string{
|
|
|
|
{"https://example.com", "a", "https://example.com/a"},
|
|
|
|
{"https://example.com/", "a", "https://example.com/a"},
|
|
|
|
{"https://example.com/a", "b", "https://example.com/a/b"},
|
|
|
|
{"https://example.com/a/", "b", "https://example.com/a/b"},
|
|
|
|
}
|
|
|
|
|
|
|
|
for _, test := range tests {
|
2017-01-06 18:34:43 +00:00
|
|
|
c, err := NewClient(nil, TestEnv(map[string]string{
|
2016-12-22 21:35:57 +00:00
|
|
|
"lfs.url": test[0],
|
|
|
|
}))
|
|
|
|
require.Nil(t, err)
|
|
|
|
|
|
|
|
req, err := c.NewRequest("POST", c.Endpoints.Endpoint("", ""), test[1], nil)
|
|
|
|
require.Nil(t, err)
|
|
|
|
assert.Equal(t, "POST", req.Method)
|
|
|
|
assert.Equal(t, test[2], req.URL.String(), fmt.Sprintf("endpoint: %s, suffix: %s, expected: %s", test[0], test[1], test[2]))
|
|
|
|
}
|
|
|
|
}
|
2017-01-25 15:59:11 +00:00
|
|
|
|
|
|
|
func TestNewRequestWithBody(t *testing.T) {
|
|
|
|
c, err := NewClient(nil, TestEnv(map[string]string{
|
|
|
|
"lfs.url": "https://example.com",
|
|
|
|
}))
|
|
|
|
require.Nil(t, err)
|
|
|
|
|
|
|
|
body := struct {
|
|
|
|
Test string
|
|
|
|
}{Test: "test"}
|
|
|
|
req, err := c.NewRequest("POST", c.Endpoints.Endpoint("", ""), "body", body)
|
|
|
|
require.Nil(t, err)
|
|
|
|
|
|
|
|
assert.NotNil(t, req.Body)
|
|
|
|
assert.Equal(t, "15", req.Header.Get("Content-Length"))
|
|
|
|
assert.EqualValues(t, 15, req.ContentLength)
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestMarshalToRequest(t *testing.T) {
|
|
|
|
req, err := http.NewRequest("POST", "https://foo/bar", nil)
|
|
|
|
require.Nil(t, err)
|
|
|
|
|
|
|
|
assert.Nil(t, req.Body)
|
|
|
|
assert.Equal(t, "", req.Header.Get("Content-Length"))
|
|
|
|
assert.EqualValues(t, 0, req.ContentLength)
|
|
|
|
|
|
|
|
body := struct {
|
|
|
|
Test string
|
|
|
|
}{Test: "test"}
|
|
|
|
require.Nil(t, MarshalToRequest(req, body))
|
|
|
|
|
|
|
|
assert.NotNil(t, req.Body)
|
|
|
|
assert.Equal(t, "15", req.Header.Get("Content-Length"))
|
|
|
|
assert.EqualValues(t, 15, req.ContentLength)
|
|
|
|
}
|