diff options
Diffstat (limited to 'internal/game/harness_test.go')
| -rw-r--r-- | internal/game/harness_test.go | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/internal/game/harness_test.go b/internal/game/harness_test.go index c80f437..1d3df10 100644 --- a/internal/game/harness_test.go +++ b/internal/game/harness_test.go @@ -196,7 +196,8 @@ func (h *harness) connect(name string) (*net.Session, *chanConn) { t.Fatalf("connect %s: %v", name, err) } conn := newChanConn() - sess := &net.Session{Conn: conn, Account: acc, Player: p, State: net.StateGame} + sess := &net.Session{Conn: conn, Account: acc, Player: p} + sess.SetState(net.StateGame) sess.StartWriter() h.hub.Add(sess) h.hub.EnterRoom(sess, p.RoomID) @@ -216,7 +217,8 @@ func (h *harness) connectMenu(accountName string) *net.Session { h.t.Fatalf("connectMenu %s: %v", accountName, err) } conn := newChanConn() - sess := &net.Session{Conn: conn, Account: acc, State: net.StateMenu} + sess := &net.Session{Conn: conn, Account: acc} + sess.SetState(net.StateMenu) sess.StartWriter() h.hub.Add(sess) h.mu.Lock() |