mirror of
https://github.com/Team254/cheesy-arena-lite.git
synced 2026-03-09 13:46:44 -04:00
102 lines
4.1 KiB
Go
102 lines
4.1 KiB
Go
// Copyright 2018 Team 254. All Rights Reserved.
|
|
// Author: pat@patfairbank.com (Patrick Fairbank)
|
|
|
|
package web
|
|
|
|
import (
|
|
"github.com/Team254/cheesy-arena-lite/field"
|
|
"github.com/Team254/cheesy-arena-lite/websocket"
|
|
gorillawebsocket "github.com/gorilla/websocket"
|
|
"github.com/mitchellh/mapstructure"
|
|
"github.com/stretchr/testify/assert"
|
|
"testing"
|
|
)
|
|
|
|
func TestSetupDisplays(t *testing.T) {
|
|
web := setupTestWeb(t)
|
|
|
|
recorder := web.getHttpResponse("/setup/displays")
|
|
assert.Equal(t, 200, recorder.Code)
|
|
assert.Contains(t, recorder.Body.String(), "Display Configuration - Untitled Event - Cheesy Arena")
|
|
}
|
|
|
|
func TestSetupDisplaysWebsocket(t *testing.T) {
|
|
web := setupTestWeb(t)
|
|
|
|
server, wsUrl := web.startTestServer()
|
|
defer server.Close()
|
|
conn, _, err := gorillawebsocket.DefaultDialer.Dial(wsUrl+"/setup/displays/websocket", nil)
|
|
assert.Nil(t, err)
|
|
defer conn.Close()
|
|
ws := websocket.NewTestWebsocket(conn)
|
|
|
|
// Should get a few status updates right after connection.
|
|
message := readDisplayConfiguration(t, ws)
|
|
assert.Empty(t, message)
|
|
|
|
// Connect a couple of displays and verify the resulting configuration messages.
|
|
displayConn1, _, _ := gorillawebsocket.DefaultDialer.Dial(wsUrl+"/display/websocket?displayId=1", nil)
|
|
defer displayConn1.Close()
|
|
displayWs1 := websocket.NewTestWebsocket(displayConn1)
|
|
assert.Equal(t, "/display?displayId=1", readWebsocketType(t, displayWs1, "displayConfiguration"))
|
|
readDisplayConfiguration(t, ws)
|
|
displayConn2, _, _ := gorillawebsocket.DefaultDialer.Dial(wsUrl+
|
|
"/displays/alliance_station/websocket?displayId=2&station=R2", nil)
|
|
defer displayConn2.Close()
|
|
message = readDisplayConfiguration(t, ws)
|
|
if assert.Equal(t, 2, len(message)) {
|
|
assert.Equal(t, field.DisplayConfiguration{"1", "", field.PlaceholderDisplay, map[string]string{}},
|
|
message["1"].DisplayConfiguration)
|
|
assert.Equal(t, 1, message["1"].ConnectionCount)
|
|
assert.Equal(t, "127.0.0.1", message["1"].IpAddress)
|
|
assert.Equal(t, field.DisplayConfiguration{"2", "", field.AllianceStationDisplay,
|
|
map[string]string{"station": "R2"}}, message["2"].DisplayConfiguration)
|
|
assert.Equal(t, 1, message["2"].ConnectionCount)
|
|
assert.Equal(t, "127.0.0.1", message["2"].IpAddress)
|
|
}
|
|
|
|
// Reconfigure a display and verify the result.
|
|
displayConfig := field.DisplayConfiguration{Id: "1", Nickname: "Audience Display", Type: field.AudienceDisplay,
|
|
Configuration: map[string]string{"background": "#00f", "reversed": "true"}}
|
|
ws.Write("configureDisplay", displayConfig)
|
|
message = readDisplayConfiguration(t, ws)
|
|
assert.Equal(t, displayConfig, message["1"].DisplayConfiguration)
|
|
assert.Equal(t, "/displays/audience?displayId=1&nickname=Audience+Display&background=%2300f&reversed=true",
|
|
readWebsocketType(t, displayWs1, "displayConfiguration"))
|
|
}
|
|
|
|
func TestSetupDisplaysWebsocketReloadDisplays(t *testing.T) {
|
|
web := setupTestWeb(t)
|
|
|
|
server, wsUrl := web.startTestServer()
|
|
defer server.Close()
|
|
conn, _, err := gorillawebsocket.DefaultDialer.Dial(wsUrl+"/setup/displays/websocket", nil)
|
|
assert.Nil(t, err)
|
|
defer conn.Close()
|
|
ws := websocket.NewTestWebsocket(conn)
|
|
|
|
// Should get a few status updates right after connection.
|
|
readDisplayConfiguration(t, ws)
|
|
|
|
// Connect a display and verify the resulting configuration messages.
|
|
displayConn, _, _ := gorillawebsocket.DefaultDialer.Dial(wsUrl+"/display/websocket?displayId=1", nil)
|
|
defer displayConn.Close()
|
|
displayWs := websocket.NewTestWebsocket(displayConn)
|
|
assert.Equal(t, "/display?displayId=1", readWebsocketType(t, displayWs, "displayConfiguration"))
|
|
readDisplayConfiguration(t, ws)
|
|
|
|
// Reset a display selectively and verify the resulting message.
|
|
ws.Write("reloadDisplay", "1")
|
|
assert.Equal(t, "1", readWebsocketType(t, displayWs, "reload"))
|
|
ws.Write("reloadAllDisplays", nil)
|
|
assert.Equal(t, nil, readWebsocketType(t, displayWs, "reload"))
|
|
}
|
|
|
|
func readDisplayConfiguration(t *testing.T, ws *websocket.Websocket) map[string]field.Display {
|
|
message := readWebsocketType(t, ws, "displayConfiguration")
|
|
var displayConfigurationMessage map[string]field.Display
|
|
err := mapstructure.Decode(message, &displayConfigurationMessage)
|
|
assert.Nil(t, err)
|
|
return displayConfigurationMessage
|
|
}
|