111 lines
2.7 KiB
Go
111 lines
2.7 KiB
Go
package agent
|
|
|
|
import (
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/hashicorp/consul/sdk/testutil"
|
|
"github.com/hashicorp/consul/testrpc"
|
|
)
|
|
|
|
func testGRPCStreamingWorking(t *testing.T, config string) {
|
|
if testing.Short() {
|
|
t.Skip("too slow for testing.Short")
|
|
}
|
|
|
|
a := NewTestAgent(t, config)
|
|
defer a.Shutdown()
|
|
|
|
testrpc.WaitForLeader(t, a.RPC, "dc1")
|
|
|
|
req, _ := http.NewRequest("GET", "/v1/health/service/consul?index=3", nil)
|
|
resp := httptest.NewRecorder()
|
|
_, err := a.srv.HealthServiceNodes(resp, req)
|
|
if err != nil {
|
|
t.Fatalf("err: %v", err)
|
|
}
|
|
|
|
assertIndex(t, resp)
|
|
require.NotEmpty(t, resp.Header().Get("X-Consul-Index"))
|
|
require.Equal(t, "streaming", resp.Header().Get("X-Consul-Query-Backend"))
|
|
}
|
|
|
|
func TestGRPCWithTLSConfigs(t *testing.T) {
|
|
// if this test is failing because of expired certificates
|
|
// use the procedure in test/CA-GENERATION.md
|
|
t.Parallel()
|
|
testCases := []struct {
|
|
name string
|
|
config string
|
|
}{
|
|
{
|
|
name: "no-tls",
|
|
config: "",
|
|
},
|
|
{
|
|
name: "tls-all-enabled",
|
|
config: `
|
|
# tls
|
|
ca_file = "../test/hostname/CertAuth.crt"
|
|
cert_file = "../test/hostname/Bob.crt"
|
|
key_file = "../test/hostname/Bob.key"
|
|
verify_incoming = true
|
|
verify_outgoing = true
|
|
verify_server_hostname = true
|
|
`,
|
|
},
|
|
{
|
|
name: "tls ready no verify incoming",
|
|
config: `
|
|
# tls
|
|
ca_file = "../test/hostname/CertAuth.crt"
|
|
cert_file = "../test/hostname/Bob.crt"
|
|
key_file = "../test/hostname/Bob.key"
|
|
verify_incoming = false
|
|
verify_outgoing = true
|
|
verify_server_hostname = false
|
|
`,
|
|
},
|
|
{
|
|
name: "tls ready no verify outgoing and incoming",
|
|
config: `
|
|
# tls
|
|
ca_file = "../test/hostname/CertAuth.crt"
|
|
cert_file = "../test/hostname/Bob.crt"
|
|
key_file = "../test/hostname/Bob.key"
|
|
verify_incoming = false
|
|
verify_outgoing = false
|
|
verify_server_hostname = false
|
|
`,
|
|
},
|
|
{
|
|
name: "tls ready, all defaults",
|
|
config: `
|
|
# tls
|
|
ca_file = "../test/hostname/CertAuth.crt"
|
|
cert_file = "../test/hostname/Bob.crt"
|
|
key_file = "../test/hostname/Bob.key"
|
|
`,
|
|
},
|
|
}
|
|
for _, tt := range testCases {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
dataDir := testutil.TempDir(t, "agent") // we manage the data dir
|
|
cfg := `data_dir = "` + dataDir + `"
|
|
domain = "consul"
|
|
node_name = "my-fancy-server"
|
|
datacenter = "dc1"
|
|
primary_datacenter = "dc1"
|
|
rpc {
|
|
enable_streaming = true
|
|
}
|
|
use_streaming_backend = true
|
|
` + tt.config
|
|
testGRPCStreamingWorking(t, cfg)
|
|
})
|
|
}
|
|
}
|