Fix tests after last change.

This commit is contained in:
Patrick Fairbank
2019-04-12 17:32:40 -07:00
parent 74c1063a2d
commit 0f74de7ad8
7 changed files with 37 additions and 6 deletions

View File

@@ -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

View File

@@ -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)
}

View File

@@ -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")
}

View File

@@ -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) {

View File

@@ -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)

View File

@@ -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")

View File

@@ -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)
}