Test cluster service with TLS Mux

cluster/service_mux_test.go

@@ -3,9 +3,11 @@ package cluster
 import (
 	"fmt"
 	"net"
+	"os"
 	"testing"
 
 	"github.com/rqlite/rqlite/tcp"
+	"github.com/rqlite/rqlite/testdata/x509"
 )
 
 func Test_NewServiceSetGetNodeAPIAddrMuxed(t *testing.T) {
@@ -40,6 +42,38 @@ func Test_NewServiceSetGetNodeAPIAddrMuxed(t *testing.T) {
 	}
 }
 
+func Test_NewServiceSetGetNodeAPIAddrMuxedTLS(t *testing.T) {
+	ln, mux := mustNewTLSMux()
+	go mux.Serve()
+	tn := mux.Listen(1) // Could be any byte value.
+
+	s := NewService(tn)
+	if s == nil {
+		t.Fatalf("failed to create cluster service")
+	}
+
+	if err := s.Open(); err != nil {
+		t.Fatalf("failed to open cluster service")
+	}
+
+	s.SetAPIAddr("foo")
+
+	addr, err := s.GetNodeAPIAddr(s.Addr())
+	if err != nil {
+		t.Fatalf("failed to get node API address: %s", err)
+	}
+	if addr != "http://foo" {
+		t.Fatalf("failed to get correct node API address")
+	}
+
+	if err := ln.Close(); err != nil {
+		t.Fatalf("failed to close Mux's listener: %s", err)
+	}
+	if err := s.Close(); err != nil {
+		t.Fatalf("failed to close cluster service")
+	}
+}
+
 func mustNewMux() (net.Listener, *tcp.Mux) {
 	ln, err := net.Listen("tcp", "localhost:0")
 	if err != nil {
@@ -53,3 +87,23 @@ func mustNewMux() (net.Listener, *tcp.Mux) {
 
 	return ln, mux
 }
+
+func mustNewTLSMux() (net.Listener, *tcp.Mux) {
+	ln, err := net.Listen("tcp", "localhost:0")
+	if err != nil {
+		panic("failed to create mock listener")
+	}
+
+	cert := x509.CertFile("")
+	defer os.Remove(cert)
+	key := x509.KeyFile("")
+	defer os.Remove(key)
+
+	mux, err := tcp.NewTLSMux(ln, nil, cert, key, "")
+	if err != nil {
+		panic(fmt.Sprintf("failed to create TLS mux: %s", err))
+	}
+	mux.InsecureSkipVerify = true
+
+	return ln, mux
+}