tests fix

This commit is contained in:
AndrewChubatiuk 2021-02-09 14:46:13 +02:00
parent 3d0aa2ef56
commit af7a16d774

View file

@ -21,13 +21,19 @@ var (
{Label: "connect-proxy-redis", Value: 3000, To: 3000}, {Label: "connect-proxy-redis", Value: 3000, To: 3000},
}, },
}} }}
testConnectPorts = structs.AllocatedPorts{{
Label = "connect-proxy-redis"
Value = 3000,
To = 3000,
HostIP = "192.168.30.1",
}}
) )
func TestConnect_newConnect(t *testing.T) { func TestConnect_newConnect(t *testing.T) {
t.Parallel() t.Parallel()
t.Run("nil", func(t *testing.T) { t.Run("nil", func(t *testing.T) {
asr, err := newConnect("", "", nil, nil) asr, err := newConnect("", "", nil, nil, nil)
require.NoError(t, err) require.NoError(t, err)
require.Nil(t, asr) require.Nil(t, asr)
}) })
@ -35,7 +41,7 @@ func TestConnect_newConnect(t *testing.T) {
t.Run("native", func(t *testing.T) { t.Run("native", func(t *testing.T) {
asr, err := newConnect("", "", &structs.ConsulConnect{ asr, err := newConnect("", "", &structs.ConsulConnect{
Native: true, Native: true,
}, nil) }, nil, nil)
require.NoError(t, err) require.NoError(t, err)
require.True(t, asr.Native) require.True(t, asr.Native)
require.Nil(t, asr.SidecarService) require.Nil(t, asr.SidecarService)
@ -48,7 +54,7 @@ func TestConnect_newConnect(t *testing.T) {
Tags: []string{"foo", "bar"}, Tags: []string{"foo", "bar"},
Port: "sidecarPort", Port: "sidecarPort",
}, },
}, testConnectNetwork) }, testConnectNetwork, testConnectPorts)
require.NoError(t, err) require.NoError(t, err)
require.Equal(t, &api.AgentServiceRegistration{ require.Equal(t, &api.AgentServiceRegistration{
Tags: []string{"foo", "bar"}, Tags: []string{"foo", "bar"},
@ -79,7 +85,7 @@ func TestConnect_connectSidecarRegistration(t *testing.T) {
t.Parallel() t.Parallel()
t.Run("nil", func(t *testing.T) { t.Run("nil", func(t *testing.T) {
sidecarReg, err := connectSidecarRegistration("", "", nil, testConnectNetwork) sidecarReg, err := connectSidecarRegistration("", "", nil, testConnectNetwork, testConnectPorts)
require.NoError(t, err) require.NoError(t, err)
require.Nil(t, sidecarReg) require.Nil(t, sidecarReg)
}) })
@ -87,7 +93,7 @@ func TestConnect_connectSidecarRegistration(t *testing.T) {
t.Run("no service port", func(t *testing.T) { t.Run("no service port", func(t *testing.T) {
_, err := connectSidecarRegistration("unknown-id", "unknown", &structs.ConsulSidecarService{ _, err := connectSidecarRegistration("unknown-id", "unknown", &structs.ConsulSidecarService{
// irrelevant // irrelevant
}, testConnectNetwork) }, testConnectNetwork, testConnectPorts)
require.EqualError(t, err, `No Connect port defined for service "unknown"`) require.EqualError(t, err, `No Connect port defined for service "unknown"`)
}) })
@ -100,7 +106,7 @@ func TestConnect_connectSidecarRegistration(t *testing.T) {
}}, }},
}, },
}, },
}, testConnectNetwork) }, testConnectNetwork, testConnectPorts)
require.EqualError(t, err, `No port of label "badPort" defined`) require.EqualError(t, err, `No port of label "badPort" defined`)
}) })
@ -108,7 +114,7 @@ func TestConnect_connectSidecarRegistration(t *testing.T) {
proxy, err := connectSidecarRegistration("redis-service-id", "redis", &structs.ConsulSidecarService{ proxy, err := connectSidecarRegistration("redis-service-id", "redis", &structs.ConsulSidecarService{
Tags: []string{"foo", "bar"}, Tags: []string{"foo", "bar"},
Port: "sidecarPort", Port: "sidecarPort",
}, testConnectNetwork) }, testConnectNetwork, testConnectPorts)
require.NoError(t, err) require.NoError(t, err)
require.Equal(t, &api.AgentServiceRegistration{ require.Equal(t, &api.AgentServiceRegistration{
Tags: []string{"foo", "bar"}, Tags: []string{"foo", "bar"},