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) }