Function TestRewriteSchemes

pkg/webui/csp_test.go:59–93

(t *testing.T)

}

func TestRewriteSchemes(t *testing.T) {
	t.Parallel()
	for _, tc := range []struct {
		name     string
		baseURLs []string
		rewrites map[string]string
		expected []string
	}{
		{
			name:     "no match",
			baseURLs: []string{"https://foo.example.com", "https://bar.example.com"},
			rewrites: map[string]string{
				"http": "ws",
			},
			expected: []string{"https://foo.example.com", "https://bar.example.com"},
		},
		{
			name:     "match",
			baseURLs: []string{"https://foo.example.com", "https://bar.example.com"},
			rewrites: map[string]string{
				"https": "wss",
			},
			expected: []string{
				"https://foo.example.com", "wss://foo.example.com", "https://bar.example.com", "wss://bar.example.com",
			},
		},
	} {
		t.Run(tc.name, func(t *testing.T) {
			t.Parallel()
			a := assertions.New(t)
			actual := webui.RewriteSchemes(tc.rewrites, tc.baseURLs...)
			a.So(actual, should.Resemble, tc.expected)
		})
	}
}