func TestGetSpectatingIds(t *testing.T) {
testhelpers.CleanupDB()
player, _ := models.NewPlayer("asdf")
database.DB.Save(player)
specIds, specErr := player.GetSpectatingIds()
assert.Nil(t, specErr)
assert.Equal(t, len(specIds), 0)
//assert.Equal(t, []uint{lobby.ID, lobby2.ID}, specIds)
lobby1 := models.NewLobby("cp_badlands", models.LobbyTypeSixes, "ugc", models.ServerRecord{}, 1, false)
database.DB.Save(lobby1)
lobby1.AddSpectator(player)
specIds, specErr = player.GetSpectatingIds()
assert.Nil(t, specErr)
assert.Equal(t, specIds[0], lobby1.ID)
lobby2 := models.NewLobby("cp_badlands", models.LobbyTypeSixes, "ugc", models.ServerRecord{}, 1, false)
database.DB.Save(lobby2)
lobby2.AddSpectator(player)
specIds, specErr = player.GetSpectatingIds()
assert.Nil(t, specErr)
assert.Equal(t, []uint{lobby1.ID, lobby2.ID}, specIds)
}
func TestIsSpectating(t *testing.T) {
testhelpers.CleanupDB()
lobby := models.NewLobby("cp_badlands", models.LobbyTypeSixes, "ugc", models.ServerRecord{}, 1, false)
database.DB.Save(lobby)
lobby2 := models.NewLobby("cp_badlands", models.LobbyTypeSixes, "ugc", models.ServerRecord{}, 1, false)
database.DB.Save(lobby2)
player, _ := models.NewPlayer("asdf")
database.DB.Save(player)
isSpectating := player.IsSpectatingId(lobby.ID)
assert.False(t, isSpectating)
lobby.AddSpectator(player)
isSpectating = player.IsSpectatingId(lobby.ID)
assert.True(t, isSpectating)
lobby2.AddSpectator(player)
isSpectating2 := player.IsSpectatingId(lobby2.ID)
assert.True(t, isSpectating2)
specIds, specErr := player.GetSpectatingIds()
assert.Nil(t, specErr)
assert.Equal(t, []uint{lobby.ID, lobby2.ID}, specIds)
lobby.RemoveSpectator(player, false)
isSpectating = player.IsSpectatingId(lobby.ID)
assert.False(t, isSpectating)
}
func TestPlayerInfoFetching(t *testing.T) {
testhelpers.CleanupDB()
if config.Constants.SteamDevApiKey == "your steam dev api key" {
return
}
// disable mock mode because we're actually testing it
config.Constants.SteamApiMockUp = false
player, playErr := models.NewPlayer("76561197999073985")
assert.Nil(t, playErr)
assert.Equal(t, "http://steamcommunity.com/id/nonagono/", player.Profileurl)
assert.Regexp(t, "(.*)steamcommunity/public/images/avatars/(.*).jpg", player.Avatar)
assert.True(t, player.GameHours >= 3000)
player.Stats.PlayedCountSet(models.LobbyTypeSixes, 3)
player.Stats.PlayedCountSet(models.LobbyTypeHighlander, 7)
player.Stats.PlayedCountIncrease(models.LobbyTypeSixes) // sixes: 3 -> 4
assert.Equal(t, 4, player.Stats.PlayedCountGet(models.LobbyTypeSixes))
assert.Equal(t, 7, player.Stats.PlayedCountGet(models.LobbyTypeHighlander))
database.DB.Save(player)
player2, err := models.GetPlayerWithStats(player.SteamId)
assert.Nil(t, err)
assert.Equal(t, 4, player2.Stats.PlayedCountGet(models.LobbyTypeSixes))
assert.Equal(t, 7, player2.Stats.PlayedCountGet(models.LobbyTypeHighlander))
assert.Equal(t, "http://steamcommunity.com/id/nonagono/", player2.Profileurl)
}
func TestPlayerSettings(t *testing.T) {
testhelpers.CleanupDB()
player, _ := models.NewPlayer("76561197999073985")
settings, err := player.GetSettings()
assert.Nil(t, err)
assert.Equal(t, 0, len(settings))
err = player.SetSetting("foo", "bar")
assert.Nil(t, err)
settings, err = player.GetSettings()
assert.Nil(t, err)
assert.Equal(t, "foo", settings[0].Key)
assert.Equal(t, "bar", settings[0].Value)
setting, err := player.GetSetting("foo")
assert.Nil(t, err)
assert.Equal(t, "bar", setting.Value)
err = player.SetSetting("hello", "world")
assert.Nil(t, err)
settings, err = player.GetSettings()
assert.Nil(t, err)
assert.Equal(t, 2, len(settings))
}
func TestNewSub(t *testing.T) {
testhelpers.CleanupDB()
lobby := testhelpers.CreateLobby()
lobby.Save()
player := testhelpers.CreatePlayer()
player.Save()
tperr := lobby.AddPlayer(player, 0, "red", "scout1")
assert.Nil(t, tperr)
sub, err := models.NewSub(lobby.ID, player.SteamId)
assert.Nil(t, err)
db.DB.Save(sub)
subs, err := models.GetPlayerSubs(player.SteamId)
assert.Nil(t, err)
assert.Equal(t, len(subs), 1)
assert.Equal(t, subs[0].LobbyID, lobby.ID)
player2 := testhelpers.CreatePlayer()
player2.Save()
tperr = lobby.AddPlayer(player2, 0, "red", "scout1")
assert.Nil(t, tperr)
err = db.DB.Where("lobby_id = ? AND steam_id = ?", lobby.ID, player.SteamId).First(sub).Error
assert.Nil(t, err)
assert.True(t, sub.Filled)
}
func TestLobbyRemove(t *testing.T) {
testhelpers.CleanupDB()
lobby := models.NewLobby("cp_badlands", models.LobbyTypeSixes, "", models.ServerRecord{0, "", "", ""}, 0, false)
lobby.Save()
player, playErr := models.NewPlayer("1235")
assert.Nil(t, playErr)
player.Save()
// add player
err := lobby.AddPlayer(player, 0)
assert.Nil(t, err)
// remove player
err = lobby.RemovePlayer(player)
assert.Nil(t, err)
// this should be empty now
_, err2 := lobby.GetPlayerIdBySlot(0)
assert.NotNil(t, err2)
// can add player again
err = lobby.AddPlayer(player, 0)
assert.Nil(t, err)
}
func TestLobbyAdd(t *testing.T) {
testhelpers.CleanupDB()
lobby := models.NewLobby("cp_badlands", models.LobbyTypeSixes, "ugc", models.ServerRecord{0, "", "", ""}, 0, false)
lobby.Save()
var players []*models.Player
for i := 0; i < 12; i++ {
player, playErr := models.NewPlayer("p" + fmt.Sprint(i))
assert.Nil(t, playErr)
player.Save()
players = append(players, player)
}
// add player
err := lobby.AddPlayer(players[0], 0)
assert.Nil(t, err)
slot, err2 := lobby.GetPlayerSlot(players[0])
assert.Equal(t, slot, 0)
assert.Nil(t, err2)
id, err3 := lobby.GetPlayerIdBySlot(0)
assert.Equal(t, id, players[0].ID)
assert.Nil(t, err3)
// try to switch slots
err = lobby.AddPlayer(players[0], 1)
assert.Nil(t, err)
slot, err2 = lobby.GetPlayerSlot(players[0])
assert.Equal(t, slot, 1)
assert.Nil(t, err2)
// this should be empty now
id, err3 = lobby.GetPlayerIdBySlot(0)
assert.NotNil(t, err3)
// try to add a second player to the same slot
err = lobby.AddPlayer(players[1], 1)
assert.NotNil(t, err)
// try to add a player to a wrong slot slot
err = lobby.AddPlayer(players[2], 55)
assert.NotNil(t, err)
lobby2 := models.NewLobby("cp_granary", models.LobbyTypeSixes, "", models.ServerRecord{0, "", "", ""}, 0, false)
lobby2.Save()
// try to add a player while they're in another lobby
err = lobby.AddPlayer(players[0], 1)
assert.NotNil(t, err)
}
func TestPlayerBanning(t *testing.T) {
testhelpers.CleanupDB()
player, _ := models.NewPlayer("76561197999073985")
player.Save()
assert.False(t, player.IsBanned(models.PlayerBanJoin))
assert.False(t, player.IsBanned(models.PlayerBanCreate))
assert.False(t, player.IsBanned(models.PlayerBanChat))
assert.False(t, player.IsBanned(models.PlayerBanFull))
past := time.Now().Add(time.Second * -10)
player.BanUntil(past, models.PlayerBanJoin, "they suck")
assert.False(t, player.IsBanned(models.PlayerBanJoin))
future := time.Now().Add(time.Second * 10)
player.BanUntil(future, models.PlayerBanJoin, "they suck")
player.BanUntil(future, models.PlayerBanFull, "they suck")
player2, _ := models.GetPlayerBySteamId(player.SteamId)
assert.False(t, player2.IsBanned(models.PlayerBanCreate))
assert.False(t, player2.IsBanned(models.PlayerBanChat))
isBannedFull, untilFull := player2.IsBannedWithTime(models.PlayerBanFull)
assert.True(t, isBannedFull)
assert.True(t, future.Sub(untilFull) < time.Second)
assert.True(t, untilFull.Sub(future) < time.Second)
log.Println(future.Sub(untilFull))
isBannedJoin, untilJoin := player2.IsBannedWithTime(models.PlayerBanJoin)
assert.True(t, isBannedJoin)
assert.True(t, future.Sub(untilJoin) < time.Second)
assert.True(t, untilJoin.Sub(future) < time.Second)
future2 := time.Now().Add(time.Second * 20)
player2.BanUntil(future2, models.PlayerBanJoin, "they suck")
isBannedJoin2, untilJoin2 := player2.IsBannedWithTime(models.PlayerBanJoin)
assert.True(t, isBannedJoin2)
assert.True(t, future2.Sub(untilJoin2) < time.Second)
assert.True(t, untilJoin.Sub(future2) < time.Second)
bans, err := player2.GetActiveBans()
assert.Nil(t, err)
assert.Equal(t, 3, len(bans))
player2.Unban(models.PlayerBanJoin)
player2.Unban(models.PlayerBanFull)
assert.False(t, player2.IsBanned(models.PlayerBanJoin))
assert.False(t, player2.IsBanned(models.PlayerBanCreate))
assert.False(t, player2.IsBanned(models.PlayerBanChat))
assert.False(t, player2.IsBanned(models.PlayerBanFull))
}
func TestLogCreation(t *testing.T) {
testhelpers.CleanupDB()
var obj = models.AdminLogEntry{}
count := 5
database.DB.Model(obj).Count(&count)
assert.Equal(t, 0, count)
models.LogAdminAction(1, helpers.ActionBanPlayer, 2)
models.LogCustomAdminAction(2, "test", 4)
database.DB.Model(obj).Count(&count)
assert.Equal(t, 2, count)
}
func TestSpectators(t *testing.T) {
testhelpers.CleanupDB()
player, playErr := models.NewPlayer("apple")
assert.Nil(t, playErr)
player.Save()
player2, playErr2 := models.NewPlayer("testing1")
assert.Nil(t, playErr2)
player2.Save()
lobby := models.NewLobby("cp_badlands", models.LobbyTypeSixes, "", models.ServerRecord{0, "", "", ""}, 0, false)
lobby.Save()
err := lobby.AddSpectator(player)
assert.Nil(t, err)
var specs []models.Player
db.DB.Model(lobby).Association("Spectators").Find(&specs)
assert.Equal(t, 1, len(specs))
err = lobby.AddSpectator(player2)
assert.Nil(t, err)
specs = nil
db.DB.Model(lobby).Association("Spectators").Find(&specs)
assert.Equal(t, 2, len(specs))
assert.Equal(t, true, specs[0].IsSpectatingId(lobby.ID))
err = lobby.RemoveSpectator(player, false)
assert.Nil(t, err)
specs = nil
db.DB.Model(lobby).Association("Spectators").Find(&specs)
assert.Equal(t, 1, len(specs))
// adding the same player again should not increase the count
err = lobby.AddSpectator(player2)
specs = nil
db.DB.Model(lobby).Association("Spectators").Find(&specs)
assert.Equal(t, 1, len(specs))
// adding a player should remove them from spectators
lobby.AddPlayer(player2, 11)
specs = nil
db.DB.Model(lobby).Association("Spectators").Find(&specs)
assert.Equal(t, 0, len(specs))
}
func TestNewChatMessage(t *testing.T) {
testhelpers.CleanupDB()
lobby := testhelpers.CreateLobby()
lobby.Save()
player := testhelpers.CreatePlayer()
player.Save()
for i := 0; i < 3; i++ {
time.Sleep(time.Second)
message := models.NewChatMessage(strconv.Itoa(i), 0, player)
assert.NotNil(t, message)
err := db.DB.Save(message).Error
assert.Nil(t, err)
}
messages, err := models.GetMessages(player, 0)
assert.Nil(t, err)
assert.Equal(t, len(messages), 3)
for i := 1; i < len(messages); i++ {
assert.True(t, messages[i].CreatedAt.Unix() > messages[i-1].CreatedAt.Unix())
}
for i := 0; i < 3; i++ {
time.Sleep(time.Second)
message := models.NewChatMessage(strconv.Itoa(i), 1, player)
assert.NotNil(t, message)
err := db.DB.Save(message).Error
assert.Nil(t, err)
}
messages, err = models.GetMessages(player, 1)
assert.Nil(t, err)
assert.Equal(t, len(messages), 3)
for i := 1; i < len(messages); i++ {
assert.True(t, messages[i].CreatedAt.Unix() > messages[i-1].CreatedAt.Unix())
}
messages, err = models.GetAllMessages(player)
assert.Nil(t, err)
assert.Equal(t, len(messages), 6)
}
func TestLobbiesPlayed(t *testing.T) {
testhelpers.CleanupDB()
stats1 := &models.PlayerStats{}
stats1.PlayedCountIncrease(models.LobbyTypeSixes) // sixes: 0 -> 1
database.DB.Save(stats1)
// can load the record
var stats2 models.PlayerStats
err := database.DB.First(&stats2, stats1.ID).Error
assert.Nil(t, err)
assert.Equal(t, 1, stats2.PlayedSixesCount)
}
func TestUnreadyPlayer(t *testing.T) {
testhelpers.CleanupDB()
player, playErr := models.NewPlayer("testing")
assert.Nil(t, playErr)
player.Save()
lobby := models.NewLobby("cp_badlands", models.LobbyTypeSixes, "", models.ServerRecord{0, "", "", ""}, 0, false)
lobby.Save()
lobby.AddPlayer(player, 0)
lobby.ReadyPlayer(player)
lobby.UnreadyPlayer(player)
ready, err := lobby.IsPlayerReady(player)
assert.Equal(t, ready, false)
assert.Nil(t, err)
}
func TestLobbyCreation(t *testing.T) {
testhelpers.CleanupDB()
lobby := models.NewLobby("cp_badlands", models.LobbyTypeSixes, "ugc", models.ServerRecord{0, "testip", "", ""}, 0, false)
lobby.Save()
lobby2, _ := models.GetLobbyById(lobby.ID)
assert.Equal(t, lobby.ID, lobby2.ID)
assert.Equal(t, lobby.ServerInfo.Host, lobby2.ServerInfo.Host)
assert.Equal(t, lobby.ServerInfo.ID, lobby2.ServerInfo.ID)
lobby.MapName = "cp_granary"
lobby.Save()
db.DB.First(lobby2)
assert.Equal(t, "cp_granary", lobby2.MapName)
}
func TestUnreadyAllPlayers(t *testing.T) {
testhelpers.CleanupDB()
lobby := models.NewLobby("cp_badlands", models.LobbyTypeSixes, "", models.ServerRecord{0, "", "", ""}, 0, false)
lobby.Save()
for i := 0; i < 12; i++ {
player, playErr := models.NewPlayer(strconv.Itoa(i))
assert.Nil(t, playErr)
player.Save()
lobby.AddPlayer(player, i)
lobby.ReadyPlayer(player)
}
err := lobby.UnreadyAllPlayers()
assert.Nil(t, err)
ready := lobby.IsEveryoneReady()
assert.Equal(t, ready, false)
}
func TestLobbiesPlayed(t *testing.T) {
testhelpers.CleanupDB()
stats1 := &models.PlayerStats{}
stats1.PlayedCountSet(models.LobbyTypeSixes, 5)
stats1.PlayedCountSet(models.LobbyTypeHighlander, 8)
stats1.PlayedCountIncrease(models.LobbyTypeSixes) // sixes: 5 -> 6
assert.Equal(t, 6, stats1.PlayedCountGet(models.LobbyTypeSixes))
assert.Equal(t, 8, stats1.PlayedCountGet(models.LobbyTypeHighlander))
database.DB.Save(stats1)
// can load the record
var stats2 models.PlayerStats
err := database.DB.First(&stats2, stats1.ID).Error
assert.Nil(t, err)
assert.Equal(t, 6, stats2.PlayedCountGet(models.LobbyTypeSixes))
assert.Equal(t, 8, stats2.PlayedCountGet(models.LobbyTypeHighlander))
}
func TestRemoveUnreadyPlayers(t *testing.T) {
testhelpers.CleanupDB()
lobby := models.NewLobby("cp_badlands", models.LobbyTypeSixes, "", models.ServerRecord{0, "", "", ""}, 0, false)
lobby.Save()
for i := 0; i < 12; i++ {
player, playErr := models.NewPlayer(strconv.Itoa(i))
assert.Nil(t, playErr)
player.Save()
lobby.AddPlayer(player, i)
}
err := lobby.RemoveUnreadyPlayers()
assert.Nil(t, err)
for i := 0; i < 12; i++ {
_, err := lobby.GetPlayerIdBySlot(i)
assert.NotNil(t, err)
}
}
func TestIsEveryoneReady(t *testing.T) {
testhelpers.CleanupDB()
player, playErr := models.NewPlayer("0")
assert.Nil(t, playErr)
player.Save()
lobby := models.NewLobby("cp_badlands", models.LobbyTypeSixes, "", models.ServerRecord{0, "", "", ""}, 0, false)
lobby.Save()
lobby.AddPlayer(player, 0)
lobby.ReadyPlayer(player)
assert.Equal(t, lobby.IsEveryoneReady(), false)
for i := 1; i < 12; i++ {
player, playErr = models.NewPlayer(strconv.Itoa(i))
assert.Nil(t, playErr)
player.Save()
lobby.AddPlayer(player, i)
lobby.ReadyPlayer(player)
}
assert.Equal(t, lobby.IsEveryoneReady(), true)
}
func TestLobbyBan(t *testing.T) {
testhelpers.CleanupDB()
lobby := models.NewLobby("cp_badlands", models.LobbyTypeSixes, "", models.ServerRecord{0, "", "", ""}, 0, false)
lobby.Save()
player, playErr := models.NewPlayer("1235")
assert.Nil(t, playErr)
player.Save()
// add player
err := lobby.AddPlayer(player, 0)
assert.Nil(t, err)
// ban player
err = lobby.RemovePlayer(player)
lobby.BanPlayer(player)
assert.Nil(t, err)
// should not be able to add again
err = lobby.AddPlayer(player, 5)
assert.NotNil(t, err)
}
