actions-runner-controller/apis/actions.github.com/v1alpha1/proxy_config_test.go at 53ef94453a954f7d06a5f76347218b060f5232ed · actions/actions-runner-controller · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
package v1alpha1_test

import (
	"net/http"
	"testing"

	corev1 "k8s.io/api/core/v1"

	"github.com/actions/actions-runner-controller/apis/actions.github.com/v1alpha1"
	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
)

func TestProxyConfig_ToSecret(t *testing.T) {
	config := &v1alpha1.ProxyConfig{
		HTTP: &v1alpha1.ProxyServerConfig{
			URL:                 "http://proxy.example.com:8080",
			CredentialSecretRef: "my-secret",
		},
		HTTPS: &v1alpha1.ProxyServerConfig{
			URL:                 "https://proxy.example.com:8080",
			CredentialSecretRef: "my-secret",
		},
		NoProxy: []string{
			"noproxy.example.com",
			"noproxy2.example.com",
		},
	}

	secretFetcher := func(string) (*corev1.Secret, error) {
		return &corev1.Secret{
			Data: map[string][]byte{
				"username": []byte("username"),
				"password": []byte("password"),
			},
		}, nil
	}

	result, err := config.ToSecretData(secretFetcher)
	require.NoError(t, err)
	require.NotNil(t, result)

	assert.Equal(t, "http://username:password@proxy.example.com:8080", string(result["http_proxy"]))
	assert.Equal(t, "https://username:password@proxy.example.com:8080", string(result["https_proxy"]))
	assert.Equal(t, "noproxy.example.com,noproxy2.example.com", string(result["no_proxy"]))
}

func TestProxyConfig_ProxyFunc(t *testing.T) {
	config := &v1alpha1.ProxyConfig{
		HTTP: &v1alpha1.ProxyServerConfig{
			URL:                 "http://proxy.example.com:8080",
			CredentialSecretRef: "my-secret",
		},
		HTTPS: &v1alpha1.ProxyServerConfig{
			URL:                 "https://proxy.example.com:8080",
			CredentialSecretRef: "my-secret",
		},
		NoProxy: []string{
			"noproxy.example.com",
			"noproxy2.example.com",
		},
	}

	secretFetcher := func(string) (*corev1.Secret, error) {
		return &corev1.Secret{
			Data: map[string][]byte{
				"username": []byte("username"),
				"password": []byte("password"),
			},
		}, nil
	}

	result, err := config.ProxyFunc(secretFetcher)
	require.NoError(t, err)

	tests := []struct {
		name string
		in   string
		out  string
	}{
		{
			name: "http target",
			in:   "http://target.com",
			out:  "http://username:password@proxy.example.com:8080",
		},
		{
			name: "https target",
			in:   "https://target.com",
			out:  "https://username:password@proxy.example.com:8080",
		},
		{
			name: "no proxy",
			in:   "https://noproxy.example.com",
			out:  "",
		},
		{
			name: "no proxy 2",
			in:   "https://noproxy2.example.com",
			out:  "",
		},
	}

	for _, test := range tests {
		t.Run(test.name, func(t *testing.T) {
			req, err := http.NewRequest("GET", test.in, nil)
			require.NoError(t, err)
			u, err := result(req)
			require.NoError(t, err)

			if test.out == "" {
				assert.Nil(t, u)
				return
			}

			assert.Equal(t, test.out, u.String())
		})
	}
}