diff --git a/src/device/input/parse/v01/server_test.go b/src/device/input/parse/v01/server_test.go index 35d4ddb..b564749 100644 --- a/src/device/input/parse/v01/server_test.go +++ b/src/device/input/parse/v01/server_test.go @@ -238,7 +238,15 @@ func TestServeGM(t *testing.T) { usersAssigned int players int }{ - "empty": {}, + "empty": {}, + "just users": {users: 2}, + "just players": {players: 2}, + "2 unassigned users and 2 players": {users: 2, players: 2}, + "2 users and 2 players": {users: 2, usersAssigned: 2, players: 2}, + "1 users and 2 players": {users: 1, usersAssigned: 1, players: 2}, + "1 unassigned users and 2 players": {users: 1, players: 2}, + "4 players for 7 users 0 assigned": {users: 7, players: 4}, + "4 players for 7 users 4 assigned": {users: 7, players: 4, usersAssigned: 4}, } for name, d := range cases { @@ -276,7 +284,11 @@ func TestServeGM(t *testing.T) { assignments[v.Player] = struct{}{} } } - if len(assignments) != c.usersAssigned { + lesser := c.users + if c.players < lesser { + lesser = c.players + } + if len(assignments) != lesser { t.Error(assignments) } })