mirror of
https://github.com/Team254/cheesy-arena-lite.git
synced 2026-03-09 13:46:44 -04:00
Fix tests after last change.
This commit is contained in:
@@ -442,7 +442,7 @@ func (arena *Arena) Update() {
|
||||
arena.AllianceStationDisplayMode = "match"
|
||||
arena.AllianceStationDisplayModeNotifier.Notify()
|
||||
//arena.sendGameSpecificDataPacket()
|
||||
if (game.MatchTiming.WarmupDurationSec > 0) {
|
||||
if game.MatchTiming.WarmupDurationSec > 0 {
|
||||
arena.MatchState = WarmupPeriod
|
||||
enabled = false
|
||||
sendDsPacket = false
|
||||
|
||||
@@ -495,18 +495,33 @@ func TestAstop(t *testing.T) {
|
||||
assert.Equal(t, true, arena.AllianceStations["R1"].DsConn.Enabled)
|
||||
|
||||
arena.handleEstop("R1", true)
|
||||
arena.handleEstop("R2", false)
|
||||
assert.Equal(t, true, arena.AllianceStations["R1"].Astop)
|
||||
assert.Equal(t, false, arena.AllianceStations["R1"].Estop)
|
||||
assert.Equal(t, false, arena.AllianceStations["R2"].Astop)
|
||||
assert.Equal(t, false, arena.AllianceStations["R2"].Estop)
|
||||
arena.lastDsPacketTime = time.Unix(0, 0) // Force a DS packet.
|
||||
arena.Update()
|
||||
assert.Equal(t, false, arena.AllianceStations["R1"].DsConn.Enabled)
|
||||
assert.Equal(t, true, arena.AllianceStations["R2"].DsConn.Enabled)
|
||||
|
||||
arena.handleEstop("R1", true)
|
||||
arena.handleEstop("R2", true)
|
||||
assert.Equal(t, true, arena.AllianceStations["R1"].Astop)
|
||||
assert.Equal(t, false, arena.AllianceStations["R1"].Estop)
|
||||
assert.Equal(t, true, arena.AllianceStations["R2"].Astop)
|
||||
assert.Equal(t, false, arena.AllianceStations["R2"].Estop)
|
||||
arena.lastDsPacketTime = time.Unix(0, 0) // Force a DS packet.
|
||||
arena.Update()
|
||||
assert.Equal(t, false, arena.AllianceStations["R1"].DsConn.Enabled)
|
||||
assert.Equal(t, false, arena.AllianceStations["R2"].DsConn.Enabled)
|
||||
|
||||
arena.handleEstop("R1", false)
|
||||
arena.handleEstop("R2", true)
|
||||
assert.Equal(t, true, arena.AllianceStations["R1"].Astop)
|
||||
assert.Equal(t, false, arena.AllianceStations["R1"].Estop)
|
||||
assert.Equal(t, true, arena.AllianceStations["R2"].Astop)
|
||||
assert.Equal(t, false, arena.AllianceStations["R2"].Estop)
|
||||
arena.lastDsPacketTime = time.Unix(0, 0) // Force a DS packet.
|
||||
arena.Update()
|
||||
assert.Equal(t, false, arena.AllianceStations["R1"].DsConn.Enabled)
|
||||
@@ -518,16 +533,27 @@ func TestAstop(t *testing.T) {
|
||||
assert.Equal(t, PausePeriod, arena.MatchState)
|
||||
arena.MatchStartTime = time.Now().Add(-time.Duration(game.MatchTiming.WarmupDurationSec+
|
||||
game.MatchTiming.AutoDurationSec+game.MatchTiming.PauseDurationSec) * time.Second)
|
||||
arena.handleEstop("R1", false)
|
||||
arena.handleEstop("R2", true)
|
||||
assert.Equal(t, false, arena.AllianceStations["R1"].Astop)
|
||||
assert.Equal(t, false, arena.AllianceStations["R1"].Estop)
|
||||
assert.Equal(t, false, arena.AllianceStations["R2"].Astop)
|
||||
assert.Equal(t, true, arena.AllianceStations["R2"].Estop)
|
||||
arena.lastDsPacketTime = time.Unix(0, 0) // Force a DS packet.
|
||||
arena.Update()
|
||||
assert.Equal(t, TeleopPeriod, arena.MatchState)
|
||||
assert.Equal(t, true, arena.AllianceStations["R1"].DsConn.Enabled)
|
||||
assert.Equal(t, false, arena.AllianceStations["R2"].DsConn.Enabled)
|
||||
|
||||
arena.handleEstop("R1", true)
|
||||
arena.handleEstop("R2", false)
|
||||
assert.Equal(t, false, arena.AllianceStations["R1"].Astop)
|
||||
assert.Equal(t, true, arena.AllianceStations["R1"].Estop)
|
||||
assert.Equal(t, false, arena.AllianceStations["R2"].Astop)
|
||||
assert.Equal(t, true, arena.AllianceStations["R2"].Estop)
|
||||
arena.lastDsPacketTime = time.Unix(0, 0) // Force a DS packet.
|
||||
arena.Update()
|
||||
assert.Equal(t, true, arena.AllianceStations["R1"].DsConn.Enabled)
|
||||
assert.Equal(t, false, arena.AllianceStations["R1"].DsConn.Enabled)
|
||||
assert.Equal(t, false, arena.AllianceStations["R2"].DsConn.Enabled)
|
||||
}
|
||||
|
||||
|
||||
@@ -7,6 +7,7 @@ package field
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"github.com/Team254/cheesy-arena/game"
|
||||
"github.com/Team254/cheesy-arena/model"
|
||||
"github.com/stretchr/testify/assert"
|
||||
"math/rand"
|
||||
@@ -26,5 +27,7 @@ func SetupTestArena(t *testing.T, uniqueName string) *Arena {
|
||||
}
|
||||
|
||||
func setupTestArena(t *testing.T) *Arena {
|
||||
game.MatchTiming.WarmupDurationSec = 3
|
||||
game.MatchTiming.PauseDurationSec = 2
|
||||
return SetupTestArena(t, "field")
|
||||
}
|
||||
|
||||
@@ -65,7 +65,7 @@ func TestSecondCounting(t *testing.T) {
|
||||
assertRedSeconds(t, 13.8, 0.0, redSwitch, timeAfterStart(17), timeAfterEnd(-10))
|
||||
redSwitch.UpdateState([2]bool{true, false}, timeAfterEnd(-5.1))
|
||||
assertRedSeconds(t, 18.9, 0.0, redSwitch, timeAfterStart(17), timeAfterEnd(0))
|
||||
assertBlueSeconds(t, 111.9, 0.0, redSwitch, timeAfterStart(17), timeAfterEnd(0))
|
||||
assertBlueSeconds(t, 109.9, 0.0, redSwitch, timeAfterStart(17), timeAfterEnd(0))
|
||||
}
|
||||
|
||||
func TestForce(t *testing.T) {
|
||||
|
||||
@@ -60,7 +60,7 @@ func TestAllianceStationDisplayWebsocket(t *testing.T) {
|
||||
web.arena.AllianceStations["B3"].Bypass = true
|
||||
web.arena.StartMatch()
|
||||
web.arena.Update()
|
||||
messages := readWebsocketMultiple(t, ws, 2)
|
||||
messages := readWebsocketMultiple(t, ws, 3)
|
||||
_, ok := messages["matchTime"]
|
||||
assert.True(t, ok)
|
||||
web.arena.MatchStartTime = time.Now().Add(-time.Duration(game.MatchTiming.WarmupDurationSec) * time.Second)
|
||||
|
||||
@@ -350,7 +350,7 @@ func TestMatchPlayWebsocketNotifications(t *testing.T) {
|
||||
web.arena.AllianceStations["B3"].Bypass = true
|
||||
assert.Nil(t, web.arena.StartMatch())
|
||||
web.arena.Update()
|
||||
messages := readWebsocketMultiple(t, ws, 3)
|
||||
messages := readWebsocketMultiple(t, ws, 4)
|
||||
_, ok := messages["matchTime"]
|
||||
assert.True(t, ok)
|
||||
_, ok = messages["audienceDisplayMode"]
|
||||
@@ -364,7 +364,6 @@ func TestMatchPlayWebsocketNotifications(t *testing.T) {
|
||||
assert.Equal(t, true, statusReceived)
|
||||
assert.Equal(t, 3, matchTime.MatchState)
|
||||
assert.Equal(t, 3, matchTime.MatchTimeSec)
|
||||
assert.True(t, ok)
|
||||
web.arena.ScoringStatusNotifier.Notify()
|
||||
readWebsocketType(t, ws, "scoringStatus")
|
||||
|
||||
|
||||
@@ -5,6 +5,7 @@ package web
|
||||
|
||||
import (
|
||||
"github.com/Team254/cheesy-arena/field"
|
||||
"github.com/Team254/cheesy-arena/game"
|
||||
"github.com/Team254/cheesy-arena/websocket"
|
||||
"github.com/stretchr/testify/assert"
|
||||
"net/http"
|
||||
@@ -83,6 +84,8 @@ func readWebsocketMultiple(t *testing.T, ws *websocket.Websocket, count int) map
|
||||
}
|
||||
|
||||
func setupTestWeb(t *testing.T) *Web {
|
||||
game.MatchTiming.WarmupDurationSec = 3
|
||||
game.MatchTiming.PauseDurationSec = 2
|
||||
arena := field.SetupTestArena(t, "web")
|
||||
return NewWeb(arena)
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user