mirror of
https://github.com/Team254/cheesy-arena-lite.git
synced 2026-03-09 13:46:44 -04:00
125 lines
3.3 KiB
Go
125 lines
3.3 KiB
Go
// Copyright 2014 Team 254. All Rights Reserved.
|
|
// Author: pat@patfairbank.com (Patrick Fairbank)
|
|
|
|
package main
|
|
|
|
import (
|
|
"github.com/stretchr/testify/assert"
|
|
"testing"
|
|
)
|
|
|
|
func TestGetNonexistentAlliance(t *testing.T) {
|
|
clearDb()
|
|
defer clearDb()
|
|
db, err := OpenDatabase(testDbPath)
|
|
assert.Nil(t, err)
|
|
defer db.Close()
|
|
|
|
allianceTeams, err := db.GetTeamsByAlliance(1114)
|
|
assert.Nil(t, err)
|
|
assert.Empty(t, allianceTeams)
|
|
}
|
|
|
|
func TestAllianceTeamCrud(t *testing.T) {
|
|
clearDb()
|
|
defer clearDb()
|
|
db, err := OpenDatabase(testDbPath)
|
|
assert.Nil(t, err)
|
|
defer db.Close()
|
|
|
|
allianceTeam := AllianceTeam{0, 1, 0, 254}
|
|
db.CreateAllianceTeam(&allianceTeam)
|
|
allianceTeams, err := db.GetTeamsByAlliance(1)
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, 1, len(allianceTeams))
|
|
assert.Equal(t, allianceTeam, allianceTeams[0])
|
|
|
|
allianceTeam.TeamId = 1114
|
|
db.SaveAllianceTeam(&allianceTeam)
|
|
allianceTeams, err = db.GetTeamsByAlliance(1)
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, 1, len(allianceTeams))
|
|
assert.Equal(t, allianceTeam.TeamId, allianceTeams[0].TeamId)
|
|
|
|
db.DeleteAllianceTeam(&allianceTeam)
|
|
allianceTeams, err = db.GetTeamsByAlliance(1)
|
|
assert.Nil(t, err)
|
|
assert.Empty(t, allianceTeams)
|
|
}
|
|
|
|
func TestGetTeamsByAlliance(t *testing.T) {
|
|
clearDb()
|
|
defer clearDb()
|
|
db, err := OpenDatabase(testDbPath)
|
|
assert.Nil(t, err)
|
|
defer db.Close()
|
|
|
|
buildTestAlliances(db)
|
|
allianceTeams, err := db.GetTeamsByAlliance(1)
|
|
assert.Nil(t, err)
|
|
if assert.Equal(t, 4, len(allianceTeams)) {
|
|
assert.Equal(t, 254, allianceTeams[0].TeamId)
|
|
assert.Equal(t, 469, allianceTeams[1].TeamId)
|
|
assert.Equal(t, 2848, allianceTeams[2].TeamId)
|
|
assert.Equal(t, 74, allianceTeams[3].TeamId)
|
|
}
|
|
allianceTeams, err = db.GetTeamsByAlliance(2)
|
|
assert.Nil(t, err)
|
|
if assert.Equal(t, 2, len(allianceTeams)) {
|
|
assert.Equal(t, 1718, allianceTeams[0].TeamId)
|
|
assert.Equal(t, 2451, allianceTeams[1].TeamId)
|
|
}
|
|
}
|
|
|
|
func TestTruncateAllianceTeams(t *testing.T) {
|
|
clearDb()
|
|
defer clearDb()
|
|
db, err := OpenDatabase(testDbPath)
|
|
assert.Nil(t, err)
|
|
defer db.Close()
|
|
|
|
allianceTeam := AllianceTeam{0, 1, 0, 254}
|
|
db.CreateAllianceTeam(&allianceTeam)
|
|
db.TruncateAllianceTeams()
|
|
allianceTeams, err := db.GetTeamsByAlliance(1)
|
|
assert.Nil(t, err)
|
|
assert.Empty(t, allianceTeams)
|
|
}
|
|
|
|
func TestGetAllAlliances(t *testing.T) {
|
|
clearDb()
|
|
defer clearDb()
|
|
db, err := OpenDatabase(testDbPath)
|
|
assert.Nil(t, err)
|
|
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)) {
|
|
if assert.Equal(t, 4, len(alliances[0])) {
|
|
assert.Equal(t, 254, alliances[0][0].TeamId)
|
|
assert.Equal(t, 469, alliances[0][1].TeamId)
|
|
assert.Equal(t, 2848, alliances[0][2].TeamId)
|
|
assert.Equal(t, 74, alliances[0][3].TeamId)
|
|
}
|
|
if assert.Equal(t, 2, len(alliances[1])) {
|
|
assert.Equal(t, 1718, alliances[1][0].TeamId)
|
|
assert.Equal(t, 2451, alliances[1][1].TeamId)
|
|
}
|
|
}
|
|
}
|
|
|
|
func buildTestAlliances(db *Database) {
|
|
db.CreateAllianceTeam(&AllianceTeam{0, 2, 0, 1718})
|
|
db.CreateAllianceTeam(&AllianceTeam{0, 1, 3, 74})
|
|
db.CreateAllianceTeam(&AllianceTeam{0, 1, 1, 469})
|
|
db.CreateAllianceTeam(&AllianceTeam{0, 1, 0, 254})
|
|
db.CreateAllianceTeam(&AllianceTeam{0, 1, 2, 2848})
|
|
db.CreateAllianceTeam(&AllianceTeam{0, 2, 1, 2451})
|
|
}
|