mirror of
https://github.com/Team254/cheesy-arena-lite.git
synced 2026-03-09 13:46:44 -04:00
Allow teams not appearing in the team list to be loaded into driver stations.
This commit is contained in:
4
arena.go
4
arena.go
@@ -174,13 +174,13 @@ func (arena *Arena) AssignTeam(teamId int, station string) error {
|
||||
return nil
|
||||
}
|
||||
|
||||
// Load the team model. Raise an error if a team doesn't exist.
|
||||
// Load the team model. If it doesn't exist, enable anonymous operation.
|
||||
team, err := db.GetTeamById(teamId)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
if team == nil {
|
||||
return fmt.Errorf("Invalid team number '%d'.", teamId)
|
||||
team = &Team{Id: teamId}
|
||||
}
|
||||
|
||||
arena.AllianceStations[station].team = team
|
||||
|
||||
@@ -48,12 +48,6 @@ func TestAssignTeam(t *testing.T) {
|
||||
err = dsConn.conn.Close()
|
||||
assert.NotNil(t, err) // Connection should have already been closed.
|
||||
|
||||
// Check assigning an unknown team.
|
||||
err = mainArena.AssignTeam(1503, "R1")
|
||||
if assert.NotNil(t, err) {
|
||||
assert.Contains(t, err.Error(), "Invalid team number")
|
||||
}
|
||||
|
||||
// Check assigning zero as the team number.
|
||||
err = mainArena.AssignTeam(0, "R2")
|
||||
assert.Nil(t, err)
|
||||
@@ -451,10 +445,6 @@ func TestSubstituteTeam(t *testing.T) {
|
||||
assert.Nil(t, err)
|
||||
assert.Equal(t, 101, mainArena.currentMatch.Blue1)
|
||||
assert.Equal(t, 101, mainArena.AllianceStations["B1"].team.Id)
|
||||
err = mainArena.SubstituteTeam(1503, "R1")
|
||||
if assert.NotNil(t, err) {
|
||||
assert.Contains(t, err.Error(), "Invalid team number")
|
||||
}
|
||||
err = mainArena.AssignTeam(104, "R4")
|
||||
if assert.NotNil(t, err) {
|
||||
assert.Contains(t, err.Error(), "Invalid alliance station")
|
||||
|
||||
Reference in New Issue
Block a user