tlsutil: convert tests for OutgoingTLSConfigForCheck to a table

In preparation for adding more test cases.
This commit is contained in:
Daniel Nephin 2021-06-24 12:51:40 -04:00
parent e7557ab4a1
commit a920936c86

View File

@ -11,6 +11,8 @@ import (
"strings" "strings"
"testing" "testing"
"github.com/google/go-cmp/cmp"
"github.com/google/go-cmp/cmp/cmpopts"
"github.com/hashicorp/go-hclog" "github.com/hashicorp/go-hclog"
"github.com/hashicorp/yamux" "github.com/hashicorp/yamux"
"github.com/stretchr/testify/require" "github.com/stretchr/testify/require"
@ -913,26 +915,86 @@ func TestConfigurator_IncomingHTTPSConfig(t *testing.T) {
require.Equal(t, []string{"h2", "http/1.1"}, c.IncomingHTTPSConfig().NextProtos) require.Equal(t, []string{"h2", "http/1.1"}, c.IncomingHTTPSConfig().NextProtos)
} }
func TestConfigurator_OutgoingTLSConfigForChecks(t *testing.T) { func TestConfigurator_OutgoingTLSConfigForCheck(t *testing.T) {
c := Configurator{base: &Config{ type testCase struct {
name string
conf func() (*Configurator, error)
skipVerify bool
serverName string
expected *tls.Config
}
cmpTLSConfig := cmp.Options{
cmpopts.IgnoreFields(tls.Config{}, "GetCertificate", "GetClientCertificate"),
cmpopts.IgnoreUnexported(tls.Config{}),
}
run := func(t *testing.T, tc testCase) {
configurator, err := tc.conf()
require.NoError(t, err)
c := configurator.OutgoingTLSConfigForCheck(tc.skipVerify, tc.serverName)
assertDeepEqual(t, tc.expected, c, cmpTLSConfig)
}
testCases := []testCase{
{
name: "default tls, skip verify, no server name",
conf: func() (*Configurator, error) {
return NewConfigurator(Config{
TLSMinVersion: "tls12", TLSMinVersion: "tls12",
EnableAgentTLSForChecks: false, EnableAgentTLSForChecks: false,
}} }, nil)
tlsConf := c.OutgoingTLSConfigForCheck(true, "") },
require.Equal(t, true, tlsConf.InsecureSkipVerify) skipVerify: true,
require.Equal(t, uint16(0), tlsConf.MinVersion) expected: &tls.Config{InsecureSkipVerify: true},
},
{
name: "agent tls, skip verify, default server name",
conf: func() (*Configurator, error) {
return NewConfigurator(Config{
TLSMinVersion: "tls12",
EnableAgentTLSForChecks: true,
ServerName: "servername",
}, nil)
},
skipVerify: true,
expected: &tls.Config{
InsecureSkipVerify: true,
MinVersion: tls.VersionTLS12,
ServerName: "servername",
},
},
{
name: "agent tls, skip verify, with server name override",
conf: func() (*Configurator, error) {
return NewConfigurator(Config{
TLSMinVersion: "tls12",
EnableAgentTLSForChecks: true,
ServerName: "servername",
}, nil)
},
skipVerify: true,
serverName: "override",
expected: &tls.Config{
InsecureSkipVerify: true,
MinVersion: tls.VersionTLS12,
ServerName: "override",
},
},
}
c.base.EnableAgentTLSForChecks = true for _, tc := range testCases {
c.base.ServerName = "servername" t.Run(tc.name, func(t *testing.T) {
tlsConf = c.OutgoingTLSConfigForCheck(true, "") run(t, tc)
require.Equal(t, true, tlsConf.InsecureSkipVerify) })
require.Equal(t, tlsLookup[c.base.TLSMinVersion], tlsConf.MinVersion) }
require.Equal(t, c.base.ServerName, tlsConf.ServerName) }
tlsConf = c.OutgoingTLSConfigForCheck(true, "servername2") func assertDeepEqual(t *testing.T, x, y interface{}, opts ...cmp.Option) {
require.Equal(t, true, tlsConf.InsecureSkipVerify) t.Helper()
require.Equal(t, tlsLookup[c.base.TLSMinVersion], tlsConf.MinVersion) if diff := cmp.Diff(x, y, opts...); diff != "" {
require.Equal(t, "servername2", tlsConf.ServerName) t.Fatalf("assertion failed: values are not equal\n--- expected\n+++ actual\n%v", diff)
}
} }
func TestConfigurator_OutgoingRPCConfig(t *testing.T) { func TestConfigurator_OutgoingRPCConfig(t *testing.T) {