func TestAuthServers(t *testing.T) {
var plainServers = []string{
"nats://*****:*****@localhost:1224",
}
aservers := strings.Join(authServers, ",")
nc, err = nats.Connect(aservers, nats.DontRandomize(), nats.Timeout(5*time.Second))
if err != nil {
t.Fatalf("Expected to connect properly: %v\n", err)
}
defer nc.Close()
if nc.ConnectedUrl() != authServers[1] {
t.Fatalf("Does not report correct connection: %s\n",
nc.ConnectedUrl())
}
}
func TestMoreErrOnConnect(t *testing.T) {
l, e := net.Listen("tcp", "127.0.0.1:0")
if e != nil {
t.Fatal("Could not listen on an ephemeral port")
}
tl := l.(*net.TCPListener)
defer tl.Close()
addr := tl.Addr().(*net.TCPAddr)
done := make(chan bool)
case1 := make(chan bool)
case2 := make(chan bool)
case3 := make(chan bool)
case4 := make(chan bool)
go func() {
for i := 0; i < 5; i++ {
conn, err := l.Accept()
if err != nil {
t.Fatalf("Error accepting client connection: %v\n", err)
}
switch i {
case 0:
// Send back a partial INFO and close the connection.
conn.Write([]byte("INFO"))
case 1:
// Send just INFO
conn.Write([]byte("INFO\r\n"))
// Stick around a bit
<-case1
case 2:
info := fmt.Sprintf("INFO {\"server_id\":\"foobar\",\"version\":\"0.7.3\",\"go\":\"go1.5.1\",\"host\":\"%s\",\"port\":%d,\"auth_required\":false,\"ssl_required\":false,\"max_payload\":1048576}\r\n", addr.IP, addr.Port)
// Send complete INFO
conn.Write([]byte(info))
// Read connect and ping commands sent from the client
br := bufio.NewReaderSize(conn, 1024)
if _, err := br.ReadString('\n'); err != nil {
t.Fatalf("Expected CONNECT from client, got: %s", err)
}
if _, err := br.ReadString('\n'); err != nil {
t.Fatalf("Expected PING from client, got: %s", err)
}
// Client expect +OK, send it but then something else than PONG
conn.Write([]byte("+OK\r\n"))
// Stick around a bit
<-case2
case 3:
info := fmt.Sprintf("INFO {\"server_id\":\"foobar\",\"version\":\"0.7.3\",\"go\":\"go1.5.1\",\"host\":\"%s\",\"port\":%d,\"auth_required\":false,\"ssl_required\":false,\"max_payload\":1048576}\r\n", addr.IP, addr.Port)
// Send complete INFO
conn.Write([]byte(info))
// Read connect and ping commands sent from the client
br := bufio.NewReaderSize(conn, 1024)
if _, err := br.ReadString('\n'); err != nil {
t.Fatalf("Expected CONNECT from client, got: %s", err)
}
if _, err := br.ReadString('\n'); err != nil {
t.Fatalf("Expected PING from client, got: %s", err)
}
// Client expect +OK, send it but then something else than PONG
conn.Write([]byte("+OK\r\nXXX\r\n"))
// Stick around a bit
<-case3
case 4:
info := fmt.Sprintf("INFO {'x'}\r\n")
// Send INFO with JSON marshall error
conn.Write([]byte(info))
// Stick around a bit
<-case4
}
conn.Close()
}
// Hang around until asked to quit
<-done
}()
natsURL := fmt.Sprintf("nats://localhost:%d", addr.Port)
if nc, err := nats.Connect(natsURL, nats.Timeout(20*time.Millisecond)); err == nil {
nc.Close()
t.Fatal("Expected error, got none")
}
if nc, err := nats.Connect(natsURL, nats.Timeout(20*time.Millisecond)); err == nil {
close(case1)
nc.Close()
t.Fatal("Expected error, got none")
}
close(case1)
opts := nats.DefaultOptions
opts.Servers = []string{natsURL}
opts.Timeout = 20 * time.Millisecond
opts.Verbose = true
if nc, err := opts.Connect(); err == nil {
close(case2)
nc.Close()
t.Fatal("Expected error, got none")
}
close(case2)
if nc, err := opts.Connect(); err == nil {
close(case3)
nc.Close()
t.Fatal("Expected error, got none")
}
close(case3)
if nc, err := opts.Connect(); err == nil {
close(case4)
nc.Close()
t.Fatal("Expected error, got none")
}
close(case4)
close(done)
}