Fix static call to session GetNumberOfConnectedClients (#431)

Co-authored-by: Rodrigo Ueda <rodrigo.ueda@wildlifestudios.com>
topfreegames · Dec 19, 2024 · de0a7d0 · de0a7d0
1 parent cec30f5
commit de0a7d0
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 1 deletion.
diff --git a/session/static.go b/session/static.go
@@ -39,5 +39,5 @@ func CloseAll() {
 
 // GetNumberOfConnectedClients returns the number of connected clients
 func GetNumberOfConnectedClients() int64 {
-	return DefaultSessionPool.GetSessionCount()
+	return DefaultSessionPool.GetNumberOfConnectedClients()
 }
diff --git a/session/test/static_test.go b/session/test/static_test.go
@@ -1,6 +1,7 @@
 package test
 
 import (
+	"math"
 	"testing"
 
 	"github.com/golang/mock/gomock"
@@ -115,3 +116,15 @@ func TestStaticCloseAll(t *testing.T) {
 	session.DefaultSessionPool = sessionPool
 	session.CloseAll()
 }
+
+func TestGetNumberOfConnectedClients(t *testing.T) {
+	ctrl := gomock.NewController(t)
+
+	expected := int64(math.MaxInt64)
+	sessionPool := mocks.NewMockSessionPool(ctrl)
+	sessionPool.EXPECT().GetNumberOfConnectedClients().Return(expected)
+
+	session.DefaultSessionPool = sessionPool
+	numberOfConnections := session.GetNumberOfConnectedClients()
+	require.Equal(t, expected, numberOfConnections)
+}