371 lines
13 KiB
Go
371 lines
13 KiB
Go
package repos
|
|
|
|
import (
|
|
"context"
|
|
"gralias/models"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/jmoiron/sqlx"
|
|
"github.com/stretchr/testify/assert"
|
|
_ "github.com/mattn/go-sqlite3"
|
|
)
|
|
|
|
func setupTestDB(t *testing.T) (*sqlx.DB, func()) {
|
|
db, err := sqlx.Connect("sqlite3", ":memory:")
|
|
assert.NoError(t, err)
|
|
|
|
// Enable foreign key constraints for SQLite
|
|
_, err = db.Exec("PRAGMA foreign_keys = ON;")
|
|
assert.NoError(t, err)
|
|
|
|
schema := `
|
|
CREATE TABLE IF NOT EXISTS rooms (
|
|
id TEXT PRIMARY KEY,
|
|
created_at TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP,
|
|
creator_name TEXT NOT NULL,
|
|
team_turn TEXT NOT NULL DEFAULT '',
|
|
this_turn_limit INTEGER NOT NULL DEFAULT 0,
|
|
opened_this_turn INTEGER NOT NULL DEFAULT 0,
|
|
blue_counter INTEGER NOT NULL DEFAULT 0,
|
|
red_counter INTEGER NOT NULL DEFAULT 0,
|
|
red_turn BOOLEAN NOT NULL DEFAULT FALSE,
|
|
mime_done BOOLEAN NOT NULL DEFAULT FALSE,
|
|
is_running BOOLEAN NOT NULL DEFAULT FALSE,
|
|
is_over BOOLEAN NOT NULL DEFAULT FALSE,
|
|
team_won TEXT NOT NULL DEFAULT '',
|
|
room_link TEXT NOT NULL DEFAULT ''
|
|
);
|
|
|
|
CREATE TABLE IF NOT EXISTS players (
|
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|
room_id TEXT,
|
|
username TEXT NOT NULL UNIQUE,
|
|
team TEXT NOT NULL DEFAULT '',
|
|
role TEXT NOT NULL DEFAULT '',
|
|
is_bot BOOLEAN NOT NULL DEFAULT FALSE,
|
|
FOREIGN KEY (room_id) REFERENCES rooms(id) ON DELETE CASCADE
|
|
);
|
|
|
|
CREATE TABLE IF NOT EXISTS word_cards (
|
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|
room_id TEXT NOT NULL,
|
|
word TEXT NOT NULL,
|
|
color TEXT NOT NULL DEFAULT '',
|
|
revealed BOOLEAN NOT NULL DEFAULT FALSE,
|
|
mime_view BOOLEAN NOT NULL DEFAULT FALSE,
|
|
FOREIGN KEY (room_id) REFERENCES rooms(id) ON DELETE CASCADE
|
|
);
|
|
|
|
CREATE TABLE IF NOT EXISTS card_marks (
|
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|
card_id INTEGER NOT NULL,
|
|
username TEXT NOT NULL,
|
|
active BOOLEAN NOT NULL DEFAULT TRUE,
|
|
FOREIGN KEY (card_id) REFERENCES word_cards(id) ON DELETE CASCADE
|
|
);
|
|
|
|
CREATE TABLE IF NOT EXISTS actions (
|
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|
room_id TEXT NOT NULL,
|
|
actor TEXT NOT NULL,
|
|
actor_color TEXT NOT NULL DEFAULT '',
|
|
action_type TEXT NOT NULL,
|
|
word TEXT NOT NULL DEFAULT '',
|
|
word_color TEXT NOT NULL DEFAULT '',
|
|
number_associated TEXT NOT NULL DEFAULT '',
|
|
created_at TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP,
|
|
FOREIGN KEY (room_id) REFERENCES rooms(id) ON DELETE CASCADE
|
|
);
|
|
|
|
CREATE TABLE IF NOT EXISTS settings (
|
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|
room_id TEXT NOT NULL,
|
|
language TEXT NOT NULL DEFAULT 'en',
|
|
room_pass TEXT NOT NULL DEFAULT '',
|
|
turn_time INTEGER NOT NULL DEFAULT 60,
|
|
created_at TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP,
|
|
FOREIGN KEY (room_id) REFERENCES rooms(id) ON DELETE CASCADE
|
|
);
|
|
|
|
CREATE TABLE IF NOT EXISTS sessions(
|
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|
updated_at TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP,
|
|
lifetime INTEGER NOT NULL DEFAULT 3600,
|
|
token_key TEXT NOT NULL DEFAULT '' UNIQUE,
|
|
username TEXT NOT NULL,
|
|
FOREIGN KEY (username) REFERENCES players(username) ON DELETE CASCADE
|
|
);
|
|
`
|
|
_, err = db.Exec(schema)
|
|
assert.NoError(t, err)
|
|
|
|
return db, func() {
|
|
db.Close()
|
|
}
|
|
}
|
|
|
|
func TestRoomsRepo_CreateRoom(t *testing.T) {
|
|
db, teardown := setupTestDB(t)
|
|
defer teardown()
|
|
|
|
repo := &RepoProvider{DB: db}
|
|
|
|
room := &models.Room{
|
|
ID: "test_room_1",
|
|
CreatedAt: time.Now(),
|
|
CreatorName: "test_creator",
|
|
TeamTurn: "blue",
|
|
ThisTurnLimit: 5,
|
|
OpenedThisTurn: 0,
|
|
BlueCounter: 0,
|
|
RedCounter: 0,
|
|
RedTurn: false,
|
|
MimeDone: false,
|
|
IsRunning: false,
|
|
IsOver: false,
|
|
TeamWon: "",
|
|
RoomLink: "",
|
|
Settings: models.GameSettings{
|
|
Language: "en",
|
|
RoundTime: 60,
|
|
RoomPass: "",
|
|
},
|
|
}
|
|
|
|
err := repo.RoomCreate(context.Background(), room)
|
|
assert.NoError(t, err)
|
|
|
|
// Verify the room was created
|
|
var retrievedRoom models.Room
|
|
err = db.Get(&retrievedRoom, "SELECT * FROM rooms WHERE id = ?", room.ID)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, room.ID, retrievedRoom.ID)
|
|
assert.Equal(t, room.CreatorName, retrievedRoom.CreatorName)
|
|
|
|
var retrievedSettings models.GameSettings
|
|
err = db.Get(&retrievedSettings, "SELECT id, language, room_pass, turn_time FROM settings WHERE room_id = ?", room.ID)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, room.Settings.Language, retrievedSettings.Language)
|
|
assert.Equal(t, room.Settings.RoundTime, retrievedSettings.RoundTime)
|
|
assert.Equal(t, room.Settings.RoomPass, retrievedSettings.RoomPass)
|
|
}
|
|
|
|
func TestRoomsRepo_GetRoomByID(t *testing.T) {
|
|
db, teardown := setupTestDB(t)
|
|
defer teardown()
|
|
|
|
repo := &RepoProvider{DB: db}
|
|
|
|
room := &models.Room{
|
|
ID: "test_room_2",
|
|
CreatedAt: time.Now(),
|
|
CreatorName: "test_creator_2",
|
|
TeamTurn: "red",
|
|
ThisTurnLimit: 5,
|
|
OpenedThisTurn: 0,
|
|
BlueCounter: 0,
|
|
RedCounter: 0,
|
|
RedTurn: true,
|
|
MimeDone: false,
|
|
IsRunning: false,
|
|
IsOver: false,
|
|
TeamWon: "",
|
|
RoomLink: "",
|
|
Settings: models.GameSettings{
|
|
Language: "en",
|
|
RoundTime: 60,
|
|
RoomPass: "",
|
|
},
|
|
}
|
|
|
|
// Insert a room directly into the database for testing GetRoomByID
|
|
_, err := db.Exec(`INSERT INTO rooms (id, created_at, creator_name, team_turn, this_turn_limit, opened_this_turn, blue_counter, red_counter, red_turn, mime_done, is_running, is_over, team_won, room_link) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`, room.ID, room.CreatedAt, room.CreatorName, room.TeamTurn, room.ThisTurnLimit, room.OpenedThisTurn, room.BlueCounter, room.RedCounter, room.RedTurn, room.MimeDone, room.IsRunning, room.IsOver, room.TeamWon, room.RoomLink)
|
|
assert.NoError(t, err)
|
|
_, err = db.Exec(`INSERT INTO settings (room_id, language, room_pass, turn_time) VALUES (?, ?, ?, ?)`, room.ID, room.Settings.Language, room.Settings.RoomPass, room.Settings.RoundTime)
|
|
assert.NoError(t, err)
|
|
|
|
retrievedRoom, err := repo.RoomGetByID(context.Background(), room.ID)
|
|
assert.NoError(t, err)
|
|
assert.NotNil(t, retrievedRoom)
|
|
assert.Equal(t, room.ID, retrievedRoom.ID)
|
|
assert.Equal(t, room.CreatorName, retrievedRoom.CreatorName)
|
|
assert.Equal(t, room.Settings.Language, retrievedRoom.Settings.Language)
|
|
}
|
|
|
|
func TestRoomsRepo_ListRooms(t *testing.T) {
|
|
db, teardown := setupTestDB(t)
|
|
defer teardown()
|
|
|
|
repo := &RepoProvider{DB: db}
|
|
|
|
room1 := &models.Room{
|
|
ID: "list_room_1",
|
|
CreatedAt: time.Now(),
|
|
CreatorName: "list_creator_1",
|
|
TeamTurn: "blue",
|
|
ThisTurnLimit: 5,
|
|
OpenedThisTurn: 0,
|
|
BlueCounter: 0,
|
|
RedCounter: 0,
|
|
RedTurn: false,
|
|
MimeDone: false,
|
|
IsRunning: false,
|
|
IsOver: false,
|
|
TeamWon: "",
|
|
RoomLink: "",
|
|
Settings: models.GameSettings{
|
|
Language: "en",
|
|
RoundTime: 60,
|
|
RoomPass: "",
|
|
},
|
|
}
|
|
room2 := &models.Room{
|
|
ID: "list_room_2",
|
|
CreatedAt: time.Now(),
|
|
CreatorName: "list_creator_2",
|
|
TeamTurn: "red",
|
|
ThisTurnLimit: 5,
|
|
OpenedThisTurn: 0,
|
|
BlueCounter: 0,
|
|
RedCounter: 0,
|
|
RedTurn: true,
|
|
MimeDone: false,
|
|
IsRunning: false,
|
|
IsOver: false,
|
|
TeamWon: "",
|
|
RoomLink: "",
|
|
Settings: models.GameSettings{
|
|
Language: "en",
|
|
RoundTime: 60,
|
|
RoomPass: "",
|
|
},
|
|
}
|
|
|
|
_, err := db.Exec(`INSERT INTO rooms (id, created_at, creator_name, team_turn, this_turn_limit, opened_this_turn, blue_counter, red_counter, red_turn, mime_done, is_running, is_over, team_won, room_link) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`, room1.ID, room1.CreatedAt, room1.CreatorName, room1.TeamTurn, room1.ThisTurnLimit, room1.OpenedThisTurn, room1.BlueCounter, room1.RedCounter, room1.RedTurn, room1.MimeDone, room1.IsRunning, room1.IsOver, room1.TeamWon, room1.RoomLink)
|
|
assert.NoError(t, err)
|
|
_, err = db.Exec(`INSERT INTO settings (room_id, language, room_pass, turn_time) VALUES (?, ?, ?, ?)`, room1.ID, room1.Settings.Language, room1.Settings.RoomPass, room1.Settings.RoundTime)
|
|
assert.NoError(t, err)
|
|
|
|
_, err = db.Exec(`INSERT INTO rooms (id, created_at, creator_name, team_turn, this_turn_limit, opened_this_turn, blue_counter, red_counter, red_turn, mime_done, is_running, is_over, team_won, room_link) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`, room2.ID, room2.CreatedAt, room2.CreatorName, room2.TeamTurn, room2.ThisTurnLimit, room2.OpenedThisTurn, room2.BlueCounter, room2.RedCounter, room2.RedTurn, room2.MimeDone, room2.IsRunning, room2.IsOver, room2.TeamWon, room2.RoomLink)
|
|
assert.NoError(t, err)
|
|
_, err = db.Exec(`INSERT INTO settings (room_id, language, room_pass, turn_time) VALUES (?, ?, ?, ?)`, room2.ID, room2.Settings.Language, room2.Settings.RoomPass, room2.Settings.RoundTime)
|
|
assert.NoError(t, err)
|
|
|
|
rooms, err := repo.RoomList(context.Background())
|
|
assert.NoError(t, err)
|
|
assert.Len(t, rooms, 2)
|
|
}
|
|
|
|
func TestRoomsRepo_DeleteRoomByID(t *testing.T) {
|
|
db, teardown := setupTestDB(t)
|
|
defer teardown()
|
|
|
|
repo := &RepoProvider{DB: db}
|
|
|
|
room := &models.Room{
|
|
ID: "delete_room_1",
|
|
CreatedAt: time.Now(),
|
|
CreatorName: "delete_creator_1",
|
|
TeamTurn: "blue",
|
|
ThisTurnLimit: 5,
|
|
OpenedThisTurn: 0,
|
|
BlueCounter: 0,
|
|
RedCounter: 0,
|
|
RedTurn: false,
|
|
MimeDone: false,
|
|
IsRunning: false,
|
|
IsOver: false,
|
|
TeamWon: "",
|
|
RoomLink: "",
|
|
Settings: models.GameSettings{
|
|
Language: "en",
|
|
RoundTime: 60,
|
|
RoomPass: "",
|
|
},
|
|
}
|
|
|
|
_, err := db.Exec(`INSERT INTO rooms (id, created_at, creator_name, team_turn, this_turn_limit, opened_this_turn, blue_counter, red_counter, red_turn, mime_done, is_running, is_over, team_won, room_link) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`, room.ID, room.CreatedAt, room.CreatorName, room.TeamTurn, room.ThisTurnLimit, room.OpenedThisTurn, room.BlueCounter, room.RedCounter, room.RedTurn, room.MimeDone, room.IsRunning, room.IsOver, room.TeamWon, room.RoomLink)
|
|
assert.NoError(t, err)
|
|
_, err = db.Exec(`INSERT INTO settings (room_id, language, room_pass, turn_time) VALUES (?, ?, ?, ?)`, room.ID, room.Settings.Language, room.Settings.RoomPass, room.Settings.RoundTime)
|
|
assert.NoError(t, err)
|
|
|
|
// Insert a word card for the room
|
|
wordCard := &models.WordCard{
|
|
RoomID: room.ID,
|
|
Word: "test_word",
|
|
Color: models.WordColorBlue,
|
|
Revealed: false,
|
|
Mime: false,
|
|
}
|
|
_, err = db.Exec(`INSERT INTO word_cards (room_id, word, color, revealed, mime_view) VALUES (?, ?, ?, ?, ?)`, wordCard.RoomID, wordCard.Word, wordCard.Color, wordCard.Revealed, wordCard.Mime)
|
|
assert.NoError(t, err)
|
|
|
|
err = repo.RoomDeleteByID(context.Background(), room.ID)
|
|
assert.NoError(t, err)
|
|
|
|
var count int
|
|
err = db.Get(&count, "SELECT COUNT(*) FROM rooms WHERE id = ?", room.ID)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, 0, count)
|
|
|
|
err = db.Get(&count, "SELECT COUNT(*) FROM settings WHERE room_id = ?", room.ID)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, 0, count)
|
|
|
|
err = db.Get(&count, "SELECT COUNT(*) FROM word_cards WHERE room_id = ?", room.ID)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, 0, count)
|
|
}
|
|
|
|
func TestRoomsRepo_UpdateRoom(t *testing.T) {
|
|
db, teardown := setupTestDB(t)
|
|
defer teardown()
|
|
|
|
repo := &RepoProvider{DB: db}
|
|
|
|
room := &models.Room{
|
|
ID: "update_room_1",
|
|
CreatedAt: time.Now(),
|
|
CreatorName: "update_creator_1",
|
|
TeamTurn: "blue",
|
|
ThisTurnLimit: 5,
|
|
OpenedThisTurn: 0,
|
|
BlueCounter: 0,
|
|
RedCounter: 0,
|
|
RedTurn: false,
|
|
MimeDone: false,
|
|
IsRunning: false,
|
|
IsOver: false,
|
|
TeamWon: "",
|
|
RoomLink: "",
|
|
Settings: models.GameSettings{
|
|
Language: "en",
|
|
RoundTime: 60,
|
|
RoomPass: "",
|
|
},
|
|
}
|
|
|
|
_, err := db.Exec(`INSERT INTO rooms (id, created_at, creator_name, team_turn, this_turn_limit, opened_this_turn, blue_counter, red_counter, red_turn, mime_done, is_running, is_over, team_won, room_link) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`, room.ID, room.CreatedAt, room.CreatorName, room.TeamTurn, room.ThisTurnLimit, room.OpenedThisTurn, room.BlueCounter, room.RedCounter, room.RedTurn, room.MimeDone, room.IsRunning, room.IsOver, room.TeamWon, room.RoomLink)
|
|
assert.NoError(t, err)
|
|
_, err = db.Exec(`INSERT INTO settings (room_id, language, room_pass, turn_time) VALUES (?, ?, ?, ?)`, room.ID, room.Settings.Language, room.Settings.RoomPass, room.Settings.RoundTime)
|
|
assert.NoError(t, err)
|
|
|
|
room.TeamTurn = "red"
|
|
room.BlueCounter = 10
|
|
room.Settings.RoundTime = 120
|
|
|
|
err = repo.RoomUpdate(context.Background(), room)
|
|
assert.NoError(t, err)
|
|
|
|
var updatedRoom models.Room
|
|
err = db.Get(&updatedRoom, "SELECT * FROM rooms WHERE id = ?", room.ID)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, models.UserTeam("red"), updatedRoom.TeamTurn)
|
|
assert.Equal(t, uint8(10), updatedRoom.BlueCounter)
|
|
|
|
var updatedSettings models.GameSettings
|
|
err = db.Get(&updatedSettings, "SELECT * FROM settings WHERE room_id = ?", room.ID)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, uint32(120), updatedSettings.RoundTime)
|
|
} |