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

test: fixes for TestLeafNodeSlowConsumer (#4388)

Browse Source 
It would fail sometimes locally otherwise...
```
=== RUN   TestLeafNodeSlowConsumer
    leafnode_test.go:7069: got: 0, expected: 1
--- FAIL: TestLeafNodeSlowConsumer (0.29s)
=== RUN   TestLeafNodeSlowConsumer
    leafnode_test.go:7069: got: 0, expected: 1
--- FAIL: TestLeafNodeSlowConsumer (0.28s)
=== RUN   TestLeafNodeSlowConsumer
--- PASS: TestLeafNodeSlowConsumer (0.28s)
=== RUN   TestLeafNodeSlowConsumer
    leafnode_test.go:7069: got: 0, expected: 1
--- FAIL: TestLeafNodeSlowConsumer (0.28s)
=== RUN   TestLeafNodeSlowConsumer
--- PASS: TestLeafNodeSlowConsumer (0.28s)
```
This commit is contained in:
Waldemar Quevedo
2023-08-10 01:12:21 -07:00
committed by GitHub
parent 9e1e92e325 7c9ea91296
commit 37d3220dfb
1 changed files with 28 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

46
server/leafnode_test.go
View File

@@ -7029,6 +7029,7 @@ func TestLeafNodeSlowConsumer(t *testing.T) {
ao := DefaultOptions()
ao.LeafNode.Host = "127.0.0.1"
ao.LeafNode.Port = -1
ao.WriteDeadline = 1 * time.Millisecond
a := RunServer(ao)
defer a.Shutdown()
@@ -7036,21 +7037,9 @@ func TestLeafNodeSlowConsumer(t *testing.T) {
if err != nil {
t.Fatalf("Error connecting: %v", err)
}
// Only leafnode slow consumers that made it past connect are tracked
// in the slow consumers counter.
if _, err := c.Write([]byte("CONNECT {}\r\n")); err != nil {
t.Fatalf("Error writing connect: %v", err)
}
if _, err := c.Write([]byte("PING\r\n")); err != nil {
t.Fatalf("Unexpected error writing PING: %v", err)
}
defer c.Close()
// Read info
br := bufio.NewReader(c)
br.ReadLine()
time.Sleep(5 * time.Millisecond)
a.mu.Lock()
checkFor(t, time.Second, 15*time.Millisecond, func() error {
checkFor(t, 2*time.Second, 15*time.Millisecond, func() error {
a.grMu.Lock()
defer a.grMu.Unlock()
for _, cli := range a.grTmpClients {
@@ -7060,12 +7049,33 @@ func TestLeafNodeSlowConsumer(t *testing.T) {
return nil
})
a.mu.Unlock()
<-time.After(250 * time.Millisecond)
// Only leafnode slow consumers that made it past connect are tracked
// in the slow consumers counter.
if _, err := c.Write([]byte("CONNECT {}\r\n")); err != nil {
t.Fatalf("Error writing connect: %v", err)
}
// Read info
br := bufio.NewReader(c)
br.ReadLine()
for i := 0; i < 10; i++ {
if _, err := c.Write([]byte("PING\r\n")); err != nil {
t.Fatalf("Unexpected error writing PING: %v", err)
}
}
defer c.Close()
timeout := time.Now().Add(time.Second)
var (
got = a.NumSlowConsumersLeafs()
got uint64
expected uint64 = 1
)
if got != expected {
t.Errorf("got: %d, expected: %d", got, expected)
for time.Now().Before(timeout) {
got = a.NumSlowConsumersLeafs()
if got == expected {
return
}
time.Sleep(1 * time.Millisecond)
}
t.Fatalf("Timed out waiting for slow consumer leafnodes, got: %v, expected: %v", got, expected)
}