summaryrefslogtreecommitdiff
path: root/client/httpclient_test.go
diff options
context:
space:
mode:
Diffstat (limited to 'client/httpclient_test.go')
-rw-r--r--client/httpclient_test.go133
1 files changed, 0 insertions, 133 deletions
diff --git a/client/httpclient_test.go b/client/httpclient_test.go
deleted file mode 100644
index f7a6340..0000000
--- a/client/httpclient_test.go
+++ /dev/null
@@ -1,133 +0,0 @@
-package client
-
-import (
- "context"
- "encoding/base64"
- "fmt"
- "io"
- "net/http"
- "strings"
- "testing"
- "time"
-
- "github.com/stretchr/testify/require"
- "gitlab.com/gitlab-org/gitlab-shell/client/testserver"
-)
-
-func TestReadTimeout(t *testing.T) {
- expectedSeconds := uint64(300)
-
- client, err := NewHTTPClientWithOpts("http://localhost:3000", "", "", "", false, expectedSeconds, nil)
- require.NoError(t, err)
-
- require.NotNil(t, client)
- require.Equal(t, time.Duration(expectedSeconds)*time.Second, client.Client.Timeout)
-}
-
-const (
- username = "basic_auth_user"
- password = "basic_auth_password"
-)
-
-func TestBasicAuthSettings(t *testing.T) {
- requests := []testserver.TestRequestHandler{
- {
- Path: "/api/v4/internal/get_endpoint",
- Handler: func(w http.ResponseWriter, r *http.Request) {
- require.Equal(t, http.MethodGet, r.Method)
-
- fmt.Fprint(w, r.Header.Get("Authorization"))
- },
- },
- {
- Path: "/api/v4/internal/post_endpoint",
- Handler: func(w http.ResponseWriter, r *http.Request) {
- require.Equal(t, http.MethodPost, r.Method)
-
- fmt.Fprint(w, r.Header.Get("Authorization"))
- },
- },
- }
-
- client := setup(t, username, password, requests)
-
- response, err := client.Get(context.Background(), "/get_endpoint")
- require.NoError(t, err)
- testBasicAuthHeaders(t, response)
-
- response, err = client.Post(context.Background(), "/post_endpoint", nil)
- require.NoError(t, err)
- testBasicAuthHeaders(t, response)
-}
-
-func testBasicAuthHeaders(t *testing.T, response *http.Response) {
- defer response.Body.Close()
-
- require.NotNil(t, response)
- responseBody, err := io.ReadAll(response.Body)
- require.NoError(t, err)
-
- headerParts := strings.Split(string(responseBody), " ")
- require.Equal(t, "Basic", headerParts[0])
-
- credentials, err := base64.StdEncoding.DecodeString(headerParts[1])
- require.NoError(t, err)
-
- require.Equal(t, username+":"+password, string(credentials))
-}
-
-func TestEmptyBasicAuthSettings(t *testing.T) {
- requests := []testserver.TestRequestHandler{
- {
- Path: "/api/v4/internal/empty_basic_auth",
- Handler: func(w http.ResponseWriter, r *http.Request) {
- require.Equal(t, "", r.Header.Get("Authorization"))
- },
- },
- }
-
- client := setup(t, "", "", requests)
-
- _, err := client.Get(context.Background(), "/empty_basic_auth")
- require.NoError(t, err)
-}
-
-func TestRequestWithUserAgent(t *testing.T) {
- const gitalyUserAgent = "gitaly/13.5.0"
- requests := []testserver.TestRequestHandler{
- {
- Path: "/api/v4/internal/default_user_agent",
- Handler: func(w http.ResponseWriter, r *http.Request) {
- require.Equal(t, defaultUserAgent, r.UserAgent())
- },
- },
- {
- Path: "/api/v4/internal/override_user_agent",
- Handler: func(w http.ResponseWriter, r *http.Request) {
- require.Equal(t, gitalyUserAgent, r.UserAgent())
- },
- },
- }
-
- client := setup(t, "", "", requests)
-
- _, err := client.Get(context.Background(), "/default_user_agent")
- require.NoError(t, err)
-
- client.SetUserAgent(gitalyUserAgent)
- _, err = client.Get(context.Background(), "/override_user_agent")
- require.NoError(t, err)
-
-}
-
-func setup(t *testing.T, username, password string, requests []testserver.TestRequestHandler) *GitlabNetClient {
- url := testserver.StartHttpServer(t, requests)
-
- httpClient, err := NewHTTPClientWithOpts(url, "", "", "", false, 1, nil)
- require.NoError(t, err)
-
- client, err := NewGitlabNetClient(username, password, "", httpClient)
- require.NoError(t, err)
-
- return client
-}