chore: add test for Engine.CountConnections

panjf2000 · May 18, 2023 · b493107 · b493107
1 parent 222a395
commit b493107
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/gnet_test.go b/gnet_test.go
@@ -243,7 +243,11 @@ func (s *testServer) OnBoot(eng Engine) (action Action) {
 
 func (s *testServer) OnOpen(c Conn) (out []byte, action Action) {
 	c.SetContext(c)
-	atomic.AddInt32(&s.connected, 1)
+	nclients := atomic.AddInt32(&s.connected, 1)
+	if int(nclients) == s.nclients {
+		require.EqualValuesf(s.tester, s.nclients, s.eng.CountConnections(), "expected connected clients: %d, but got: %d",
+			nclients, s.nclients)
+	}
 	out = []byte("sweetness\r\n")
 	require.NotNil(s.tester, c.LocalAddr(), "nil local addr")
 	require.NotNil(s.tester, c.RemoteAddr(), "nil remote addr")