gogrlx/nats-server
1
0
Fork 0
mirror of https://github.com/gogrlx/nats-server.git synced 2026-04-02 03:38:42 -07:00
Code Issues Packages Projects Releases Wiki Activity

Read INFO used bufio.Reader

Browse Source 
Signed-off-by: Derek Collison <derek@nats.io>
This commit is contained in:
Derek Collison
2020-06-01 11:46:18 -07:00
parent 2bd7553c71
commit 5f130128ce
1 changed files with 9 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
server/client_test.go
View File

@@ -1998,20 +1998,24 @@ func TestNoClientLeakOnSlowConsumer(t *testing.T) {
}
defer c.Close()
var buf [4 * 1024]byte
cr := bufio.NewReader(c)
// Wait for INFO...
n, _ := c.Read(buf[:])
line, _, _ := cr.ReadLine()
var info serverInfo
if err = json.Unmarshal([]byte(buf[5:n]), &info); err != nil {
if err = json.Unmarshal(line[5:], &info); err != nil {
t.Fatalf("Could not parse INFO json: %v\n", err)
}
// Send our connect
if _, err := c.Write([]byte("CONNECT {}\r\nSUB foo 1\r\nPING\r\n")); err != nil {
if _, err := c.Write([]byte("CONNECT {\"verbose\": false}\r\nSUB foo 1\r\nPING\r\n")); err != nil {
t.Fatalf("Error sending CONNECT and SUB: %v", err)
}
// Wait for PONG
c.Read(buf[:])
line, _, _ = cr.ReadLine()
if string(line) != "PONG" {
t.Fatalf("Expected 'PONG' but got %q", line)
}
// Get the client from server map
cli := s.GetClient(info.CID)