mirror of https://github.com/hashicorp/consul
199 lines
4.4 KiB
Go
199 lines
4.4 KiB
Go
package proxy
|
|
|
|
import (
|
|
"strings"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/hashicorp/consul/agent"
|
|
"github.com/hashicorp/consul/connect/proxy"
|
|
"github.com/mitchellh/cli"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestCommandConfigWatcher(t *testing.T) {
|
|
if testing.Short() {
|
|
t.Skip("too slow for testing.Short")
|
|
}
|
|
|
|
t.Parallel()
|
|
|
|
cases := []struct {
|
|
Name string
|
|
Flags []string
|
|
Test func(*testing.T, *proxy.Config)
|
|
WantErr string
|
|
}{
|
|
{
|
|
Name: "-service flag only",
|
|
Flags: []string{"-service", "web"},
|
|
Test: func(t *testing.T, cfg *proxy.Config) {
|
|
require.Equal(t, 0, cfg.PublicListener.BindPort)
|
|
require.Len(t, cfg.Upstreams, 0)
|
|
},
|
|
},
|
|
|
|
{
|
|
Name: "-service flag with upstreams",
|
|
Flags: []string{
|
|
"-service", "web",
|
|
"-upstream", "db:1234",
|
|
"-upstream", "db2:2345",
|
|
},
|
|
Test: func(t *testing.T, cfg *proxy.Config) {
|
|
require.Equal(t, 0, cfg.PublicListener.BindPort)
|
|
require.Len(t, cfg.Upstreams, 2)
|
|
require.Equal(t, 1234, cfg.Upstreams[0].LocalBindPort)
|
|
require.Equal(t, 2345, cfg.Upstreams[1].LocalBindPort)
|
|
},
|
|
},
|
|
|
|
{
|
|
Name: "-service flag with -service-addr",
|
|
Flags: []string{"-service", "web"},
|
|
Test: func(t *testing.T, cfg *proxy.Config) {
|
|
// -service-addr has no affect since -listen isn't set
|
|
require.Equal(t, 0, cfg.PublicListener.BindPort)
|
|
require.Len(t, cfg.Upstreams, 0)
|
|
},
|
|
},
|
|
|
|
{
|
|
Name: "-service, -service-addr, -listen",
|
|
Flags: []string{
|
|
"-service", "web",
|
|
"-service-addr", "127.0.0.1:1234",
|
|
"-listen", ":4567",
|
|
},
|
|
Test: func(t *testing.T, cfg *proxy.Config) {
|
|
require.Len(t, cfg.Upstreams, 0)
|
|
|
|
require.Equal(t, "", cfg.PublicListener.BindAddress)
|
|
require.Equal(t, 4567, cfg.PublicListener.BindPort)
|
|
require.Equal(t, "127.0.0.1:1234", cfg.PublicListener.LocalServiceAddress)
|
|
},
|
|
},
|
|
|
|
{
|
|
Name: "-sidecar-for, no sidecar",
|
|
Flags: []string{
|
|
"-sidecar-for", "no-sidecar",
|
|
},
|
|
WantErr: "No sidecar proxy registered",
|
|
},
|
|
|
|
{
|
|
Name: "-sidecar-for, multiple sidecars",
|
|
Flags: []string{
|
|
"-sidecar-for", "two-sidecars",
|
|
},
|
|
// Order is non-deterministic so don't assert the list of proxy IDs here
|
|
WantErr: `More than one sidecar proxy registered for two-sidecars.
|
|
Start proxy with -proxy-id and one of the following IDs: `,
|
|
},
|
|
|
|
{
|
|
Name: "-sidecar-for, non-existent",
|
|
Flags: []string{
|
|
"-sidecar-for", "foo",
|
|
},
|
|
WantErr: "No sidecar proxy registered",
|
|
},
|
|
|
|
{
|
|
Name: "-sidecar-for, one sidecar",
|
|
Flags: []string{
|
|
"-sidecar-for", "one-sidecar",
|
|
},
|
|
Test: func(t *testing.T, cfg *proxy.Config) {
|
|
// Sanity check we got the right instance.
|
|
require.Equal(t, 9999, cfg.PublicListener.BindPort)
|
|
},
|
|
},
|
|
}
|
|
|
|
for _, tc := range cases {
|
|
t.Run(tc.Name, func(t *testing.T) {
|
|
require := require.New(t)
|
|
|
|
// Register a few services with 0, 1 and 2 sidecars
|
|
a := agent.NewTestAgent(t, `
|
|
services {
|
|
name = "no-sidecar"
|
|
port = 1111
|
|
}
|
|
services {
|
|
name = "one-sidecar"
|
|
port = 2222
|
|
connect {
|
|
sidecar_service {
|
|
port = 9999
|
|
}
|
|
}
|
|
}
|
|
services {
|
|
name = "two-sidecars"
|
|
port = 3333
|
|
connect {
|
|
sidecar_service {}
|
|
}
|
|
}
|
|
services {
|
|
kind = "connect-proxy"
|
|
name = "other-sidecar-for-two-sidecars"
|
|
port = 4444
|
|
proxy {
|
|
destination_service_id = "two-sidecars"
|
|
destination_service_name = "two-sidecars"
|
|
}
|
|
}
|
|
`)
|
|
defer a.Shutdown()
|
|
client := a.Client()
|
|
|
|
ui := cli.NewMockUi()
|
|
c := New(ui, make(chan struct{}))
|
|
c.testNoStart = true
|
|
|
|
// Run the command
|
|
code := c.Run(append([]string{
|
|
"-http-addr=" + a.HTTPAddr(),
|
|
}, tc.Flags...))
|
|
if tc.WantErr == "" {
|
|
require.Equal(0, code, ui.ErrorWriter.String())
|
|
} else {
|
|
require.Equal(1, code, ui.ErrorWriter.String())
|
|
require.Contains(ui.ErrorWriter.String(), tc.WantErr)
|
|
return
|
|
}
|
|
|
|
// Get the configuration watcher
|
|
cw, err := c.configWatcher(client)
|
|
require.NoError(err)
|
|
if tc.Test != nil {
|
|
tc.Test(t, testConfig(t, cw))
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func testConfig(t *testing.T, cw proxy.ConfigWatcher) *proxy.Config {
|
|
t.Helper()
|
|
|
|
select {
|
|
case cfg := <-cw.Watch():
|
|
return cfg
|
|
|
|
case <-time.After(1 * time.Second):
|
|
t.Fatal("no configuration loaded")
|
|
return nil // satisfy compiler
|
|
}
|
|
}
|
|
|
|
func TestCatalogCommand_noTabs(t *testing.T) {
|
|
t.Parallel()
|
|
if strings.ContainsRune(New(nil, nil).Help(), '\t') {
|
|
t.Fatal("help has tabs")
|
|
}
|
|
}
|