Fix: unittests
This commit is contained in:
@ -3,65 +3,72 @@ package repos
|
||||
import (
|
||||
"context"
|
||||
"gralias/models"
|
||||
|
||||
"github.com/jmoiron/sqlx"
|
||||
)
|
||||
|
||||
type PlayersRepo interface {
|
||||
PlayerGetByName(username string) (*models.Player, error)
|
||||
PlayerAdd(player *models.Player) error
|
||||
PlayerUpdate(player *models.Player) error
|
||||
PlayerDelete(roomID, username string) error
|
||||
PlayerSetRoomID(username, roomID string) error
|
||||
PlayerExitRoom(username string) error
|
||||
PlayerListNames() ([]string, error)
|
||||
PlayerList(isBot bool) ([]models.Player, error)
|
||||
PlayerGetByName(ctx context.Context, username string) (*models.Player, error)
|
||||
PlayerAdd(ctx context.Context, player *models.Player) error
|
||||
PlayerUpdate(ctx context.Context, player *models.Player) error
|
||||
PlayerDelete(ctx context.Context, roomID, username string) error
|
||||
PlayerSetRoomID(ctx context.Context, username, roomID string) error
|
||||
PlayerExitRoom(ctx context.Context, username string) error
|
||||
PlayerListNames(ctx context.Context) ([]string, error)
|
||||
PlayerList(ctx context.Context, isBot bool) ([]models.Player, error)
|
||||
}
|
||||
|
||||
func (p *RepoProvider) PlayerListNames() ([]string, error) {
|
||||
func (p *RepoProvider) PlayerListNames(ctx context.Context) ([]string, error) {
|
||||
var names []string
|
||||
err := p.DB.SelectContext(context.Background(), &names, "SELECT username FROM players;")
|
||||
err := sqlx.SelectContext(ctx, p.DB, &names, "SELECT username FROM players;")
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return names, nil
|
||||
}
|
||||
|
||||
func (p *RepoProvider) PlayerGetByName(username string) (*models.Player, error) {
|
||||
func (p *RepoProvider) PlayerGetByName(ctx context.Context, username string) (*models.Player, error) {
|
||||
var player models.Player
|
||||
err := p.DB.GetContext(context.Background(), &player, "SELECT id, room_id, username, team, role, is_bot FROM players WHERE username = ?", username)
|
||||
err := sqlx.GetContext(ctx, p.DB, &player, "SELECT id, room_id, username, team, role, is_bot FROM players WHERE username = ?", username)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return &player, nil
|
||||
}
|
||||
|
||||
func (p *RepoProvider) PlayerAdd(player *models.Player) error {
|
||||
_, err := p.DB.ExecContext(context.Background(), "INSERT INTO players (room_id, username, team, role, is_bot) VALUES (?, ?, ?, ?, ?)",
|
||||
func (p *RepoProvider) PlayerAdd(ctx context.Context, player *models.Player) error {
|
||||
db := getDB(ctx, p.DB)
|
||||
_, err := db.ExecContext(ctx, "INSERT INTO players (room_id, username, team, role, is_bot) VALUES (?, ?, ?, ?, ?)",
|
||||
player.RoomID, player.Username, player.Team, player.Role, player.IsBot)
|
||||
return err
|
||||
}
|
||||
|
||||
func (p *RepoProvider) PlayerUpdate(player *models.Player) error {
|
||||
_, err := p.DB.ExecContext(context.Background(), "UPDATE players SET room_id = ?, username = ?, team = ?, role = ?, is_bot = ? WHERE id = ?",
|
||||
func (p *RepoProvider) PlayerUpdate(ctx context.Context, player *models.Player) error {
|
||||
db := getDB(ctx, p.DB)
|
||||
_, err := db.ExecContext(ctx, "UPDATE players SET room_id = ?, username = ?, team = ?, role = ?, is_bot = ? WHERE id = ?",
|
||||
player.RoomID, player.Username, player.Team, player.Role, player.IsBot, player.ID)
|
||||
return err
|
||||
}
|
||||
|
||||
func (p *RepoProvider) PlayerDelete(roomID, username string) error {
|
||||
_, err := p.DB.ExecContext(context.Background(), "DELETE FROM players WHERE room_id = ? AND username = ?", roomID, username)
|
||||
func (p *RepoProvider) PlayerDelete(ctx context.Context, roomID, username string) error {
|
||||
db := getDB(ctx, p.DB)
|
||||
_, err := db.ExecContext(ctx, "DELETE FROM players WHERE room_id = ? AND username = ?", roomID, username)
|
||||
return err
|
||||
}
|
||||
|
||||
func (p *RepoProvider) PlayerSetRoomID(username, roomID string) error {
|
||||
_, err := p.DB.ExecContext(context.Background(), "UPDATE players SET room_id = ? WHERE username = ?", roomID, username)
|
||||
func (p *RepoProvider) PlayerSetRoomID(ctx context.Context, username, roomID string) error {
|
||||
db := getDB(ctx, p.DB)
|
||||
_, err := db.ExecContext(ctx, "UPDATE players SET room_id = ? WHERE username = ?", roomID, username)
|
||||
return err
|
||||
}
|
||||
|
||||
func (p *RepoProvider) PlayerExitRoom(username string) error {
|
||||
_, err := p.DB.ExecContext(context.Background(), "UPDATE players SET room_id = null WHERE username = ?", username)
|
||||
func (p *RepoProvider) PlayerExitRoom(ctx context.Context, username string) error {
|
||||
db := getDB(ctx, p.DB)
|
||||
_, err := db.ExecContext(ctx, "UPDATE players SET room_id = null WHERE username = ?", username)
|
||||
return err
|
||||
}
|
||||
|
||||
func (p *RepoProvider) PlayerList(isBot bool) ([]models.Player, error) {
|
||||
func (p *RepoProvider) PlayerList(ctx context.Context, isBot bool) ([]models.Player, error) {
|
||||
var players []models.Player
|
||||
var query string
|
||||
if isBot {
|
||||
@ -69,7 +76,7 @@ func (p *RepoProvider) PlayerList(isBot bool) ([]models.Player, error) {
|
||||
} else {
|
||||
query = "SELECT id, room_id, username, team, role, is_bot FROM players WHERE is_bot = false;"
|
||||
}
|
||||
err := p.DB.SelectContext(context.Background(), &players, query)
|
||||
err := sqlx.SelectContext(ctx, p.DB, &players, query)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
Reference in New Issue
Block a user