From 129690e90184caaf4e711b880b9518f552e6be09 Mon Sep 17 00:00:00 2001 From: sergystepanov Date: Sat, 22 Nov 2025 22:21:05 +0300 Subject: [PATCH] Fix map test --- pkg/worker/room/router_test.go | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/pkg/worker/room/router_test.go b/pkg/worker/room/router_test.go index f404073c..013f1e27 100644 --- a/pkg/worker/room/router_test.go +++ b/pkg/worker/room/router_test.go @@ -25,7 +25,9 @@ type lookMap struct { func (l *lookMap) Reset() { l.prev = com.NewNetMap[string, *tSession]() - l.Map.ForEach(func(s *tSession) { l.prev.Add(s) }) + for s := range l.Map.Values() { + l.prev.Add(s) + } l.NetMap.Reset() } @@ -59,7 +61,9 @@ func TestRouterReset(t *testing.T) { router.Reset() disconnected := true - u.prev.ForEach(func(u *tSession) { disconnected = disconnected && !u.connected }) + for u := range u.prev.Values() { + disconnected = disconnected && !u.connected + } if !disconnected { t.Errorf("not all users were disconnected, but should") }