mirror of
https://github.com/Team254/cheesy-arena-lite.git
synced 2026-03-09 21:56:50 -04:00
123 lines
3.2 KiB
Go
123 lines
3.2 KiB
Go
// Copyright 2022 Team 254. All Rights Reserved.
|
|
// Author: pat@patfairbank.com (Patrick Fairbank)
|
|
|
|
package model
|
|
|
|
import (
|
|
"github.com/stretchr/testify/assert"
|
|
"testing"
|
|
)
|
|
|
|
func TestGetNonexistentAlliance(t *testing.T) {
|
|
db := setupTestDb(t)
|
|
defer db.Close()
|
|
|
|
alliance, err := db.GetAllianceById(1114)
|
|
assert.Nil(t, err)
|
|
assert.Nil(t, alliance)
|
|
}
|
|
|
|
func TestAllianceCrud(t *testing.T) {
|
|
db := setupTestDb(t)
|
|
defer db.Close()
|
|
|
|
alliance := Alliance{Id: 3, TeamIds: []int{254, 1114, 296, 1503}, Lineup: [3]int{1114, 254, 296}}
|
|
assert.Nil(t, db.CreateAlliance(&alliance))
|
|
alliance2, err := db.GetAllianceById(3)
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, alliance, *alliance2)
|
|
|
|
alliance.TeamIds = append(alliance.TeamIds, 296)
|
|
assert.Nil(t, db.UpdateAlliance(&alliance))
|
|
alliance2, err = db.GetAllianceById(3)
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, alliance, *alliance2)
|
|
|
|
assert.Nil(t, db.DeleteAlliance(alliance.Id))
|
|
alliance2, err = db.GetAllianceById(3)
|
|
assert.Nil(t, err)
|
|
assert.Nil(t, alliance2)
|
|
}
|
|
|
|
func TestTruncateAllianceTeams(t *testing.T) {
|
|
db := setupTestDb(t)
|
|
defer db.Close()
|
|
|
|
alliance := Alliance{Id: 1, TeamIds: []int{148, 118, 125}, Lineup: [3]int{118, 148, 125}}
|
|
assert.Nil(t, db.CreateAlliance(&alliance))
|
|
assert.Nil(t, db.TruncateAlliances())
|
|
alliance2, err := db.GetAllianceById(1)
|
|
assert.Nil(t, err)
|
|
assert.Nil(t, alliance2)
|
|
}
|
|
|
|
func TestGetAllAlliances(t *testing.T) {
|
|
db := setupTestDb(t)
|
|
defer db.Close()
|
|
|
|
alliances, err := db.GetAllAlliances()
|
|
assert.Nil(t, err)
|
|
assert.Empty(t, alliances)
|
|
|
|
BuildTestAlliances(db)
|
|
alliances, err = db.GetAllAlliances()
|
|
assert.Nil(t, err)
|
|
if assert.Equal(t, 2, len(alliances)) {
|
|
assert.Equal(t, 1, alliances[0].Id)
|
|
assert.Equal(t, []int{254, 469, 2848, 74, 3175}, alliances[0].TeamIds)
|
|
assert.Equal(t, 2, alliances[1].Id)
|
|
assert.Equal(t, []int{1718, 2451, 1619}, alliances[1].TeamIds)
|
|
}
|
|
}
|
|
|
|
func TestGetOffFieldTeamIds(t *testing.T) {
|
|
db := setupTestDb(t)
|
|
defer db.Close()
|
|
BuildTestAlliances(db)
|
|
|
|
match := &Match{
|
|
ElimRedAlliance: 1,
|
|
ElimBlueAlliance: 2,
|
|
Red1: 469,
|
|
Red2: 254,
|
|
Red3: 2848,
|
|
Blue1: 1619,
|
|
Blue2: 1718,
|
|
Blue3: 2451,
|
|
}
|
|
|
|
redOffFieldTeams, blueOffFieldTeams, err := db.GetOffFieldTeamIds(match)
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, []int{74, 3175}, redOffFieldTeams)
|
|
assert.Equal(t, []int{}, blueOffFieldTeams)
|
|
|
|
match.Red1 = 74
|
|
match.Red2 = 3175
|
|
redOffFieldTeams, blueOffFieldTeams, err = db.GetOffFieldTeamIds(match)
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, []int{254, 469}, redOffFieldTeams)
|
|
assert.Equal(t, []int{}, blueOffFieldTeams)
|
|
|
|
match.ElimRedAlliance = 0
|
|
match.ElimBlueAlliance = 0
|
|
redOffFieldTeams, blueOffFieldTeams, err = db.GetOffFieldTeamIds(match)
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, []int{}, redOffFieldTeams)
|
|
assert.Equal(t, []int{}, blueOffFieldTeams)
|
|
|
|
match = &Match{
|
|
ElimRedAlliance: 2,
|
|
ElimBlueAlliance: 1,
|
|
Red1: 1718,
|
|
Red2: 2451,
|
|
Red3: 1619,
|
|
Blue1: 3175,
|
|
Blue2: 74,
|
|
Blue3: 2848,
|
|
}
|
|
redOffFieldTeams, blueOffFieldTeams, err = db.GetOffFieldTeamIds(match)
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, []int{}, redOffFieldTeams)
|
|
assert.Equal(t, []int{254, 469}, blueOffFieldTeams)
|
|
}
|