mirror of
https://github.com/Team254/cheesy-arena-lite.git
synced 2026-03-09 13:46:44 -04:00
101 lines
2.6 KiB
Go
101 lines
2.6 KiB
Go
// Copyright 2014 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()
|
|
|
|
allianceTeams, err := db.GetTeamsByAlliance(1114)
|
|
assert.Nil(t, err)
|
|
assert.Empty(t, allianceTeams)
|
|
}
|
|
|
|
func TestAllianceTeamCrud(t *testing.T) {
|
|
db := setupTestDb(t)
|
|
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.UpdateAllianceTeam(&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.Id)
|
|
allianceTeams, err = db.GetTeamsByAlliance(1)
|
|
assert.Nil(t, err)
|
|
assert.Empty(t, allianceTeams)
|
|
}
|
|
|
|
func TestGetTeamsByAlliance(t *testing.T) {
|
|
db := setupTestDb(t)
|
|
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) {
|
|
db := setupTestDb(t)
|
|
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) {
|
|
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)) {
|
|
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)
|
|
}
|
|
}
|
|
}
|