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) { 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") } }