Fix: unittests
This commit is contained in:
Grail Finder
@ -18,22 +18,29 @@ func setupTestDB(t *testing.T) (*sqlx.DB, func()) {
|
||||
schema := `
|
||||
CREATE TABLE IF NOT EXISTS rooms (
|
||||
id TEXT PRIMARY KEY,
|
||||
created_at DATETIME,
|
||||
creator_name TEXT,
|
||||
team_turn TEXT,
|
||||
this_turn_limit INTEGER,
|
||||
opened_this_turn INTEGER,
|
||||
blue_counter INTEGER,
|
||||
red_counter INTEGER,
|
||||
red_turn BOOLEAN,
|
||||
mime_done BOOLEAN,
|
||||
is_public BOOLEAN,
|
||||
is_running BOOLEAN,
|
||||
language TEXT,
|
||||
round_time INTEGER,
|
||||
is_over BOOLEAN,
|
||||
team_won TEXT,
|
||||
room_pass TEXT
|
||||
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 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)
|
||||
);
|
||||
`
|
||||
_, err = db.Exec(schema)
|
||||
@ -61,16 +68,18 @@ func TestRoomsRepo_CreateRoom(t *testing.T) {
|
||||
RedCounter: 0,
|
||||
RedTurn: false,
|
||||
MimeDone: false,
|
||||
IsPublic: true,
|
||||
IsRunning: false,
|
||||
Language: "en",
|
||||
RoundTime: 60,
|
||||
IsOver: false,
|
||||
TeamWon: "",
|
||||
RoomPass: "",
|
||||
RoomLink: "",
|
||||
Settings: models.GameSettings{
|
||||
Language: "en",
|
||||
RoundTime: 60,
|
||||
RoomPass: "",
|
||||
},
|
||||
}
|
||||
|
||||
err := repo.CreateRoom(context.Background(), room)
|
||||
err := repo.RoomCreate(context.Background(), room)
|
||||
assert.NoError(t, err)
|
||||
|
||||
// Verify the room was created
|
||||
@ -79,6 +88,13 @@ func TestRoomsRepo_CreateRoom(t *testing.T) {
|
||||
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) {
|
||||
@ -98,24 +114,29 @@ func TestRoomsRepo_GetRoomByID(t *testing.T) {
|
||||
RedCounter: 0,
|
||||
RedTurn: true,
|
||||
MimeDone: false,
|
||||
IsPublic: true,
|
||||
IsRunning: false,
|
||||
Language: "en",
|
||||
RoundTime: 60,
|
||||
IsOver: false,
|
||||
TeamWon: "",
|
||||
RoomPass: "",
|
||||
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_public, is_running, language, round_time, is_over, team_won, room_pass) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`, room.ID, room.CreatedAt, room.CreatorName, room.TeamTurn, room.ThisTurnLimit, room.OpenedThisTurn, room.BlueCounter, room.RedCounter, room.RedTurn, room.MimeDone, room.IsPublic, room.IsRunning, room.Language, room.RoundTime, room.IsOver, room.TeamWon, room.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)
|
||||
|
||||
retrievedRoom, err := repo.GetRoomByID(context.Background(), room.ID)
|
||||
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) {
|
||||
@ -135,13 +156,15 @@ func TestRoomsRepo_ListRooms(t *testing.T) {
|
||||
RedCounter: 0,
|
||||
RedTurn: false,
|
||||
MimeDone: false,
|
||||
IsPublic: true,
|
||||
IsRunning: false,
|
||||
Language: "en",
|
||||
RoundTime: 60,
|
||||
IsOver: false,
|
||||
TeamWon: "",
|
||||
RoomPass: "",
|
||||
RoomLink: "",
|
||||
Settings: models.GameSettings{
|
||||
Language: "en",
|
||||
RoundTime: 60,
|
||||
RoomPass: "",
|
||||
},
|
||||
}
|
||||
room2 := &models.Room{
|
||||
ID: "list_room_2",
|
||||
@ -154,21 +177,28 @@ func TestRoomsRepo_ListRooms(t *testing.T) {
|
||||
RedCounter: 0,
|
||||
RedTurn: true,
|
||||
MimeDone: false,
|
||||
IsPublic: true,
|
||||
IsRunning: false,
|
||||
Language: "en",
|
||||
RoundTime: 60,
|
||||
IsOver: false,
|
||||
TeamWon: "",
|
||||
RoomPass: "",
|
||||
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_public, is_running, language, round_time, is_over, team_won, room_pass) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`, room1.ID, room1.CreatedAt, room1.CreatorName, room1.TeamTurn, room1.ThisTurnLimit, room1.OpenedThisTurn, room1.BlueCounter, room1.RedCounter, room1.RedTurn, room1.MimeDone, room1.IsPublic, room1.IsRunning, room1.Language, room1.RoundTime, room1.IsOver, room1.TeamWon, room1.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 rooms (id, created_at, creator_name, team_turn, this_turn_limit, opened_this_turn, blue_counter, red_counter, red_turn, mime_done, is_public, is_running, language, round_time, is_over, team_won, room_pass) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`, room2.ID, room2.CreatedAt, room2.CreatorName, room2.TeamTurn, room2.ThisTurnLimit, room2.OpenedThisTurn, room2.BlueCounter, room2.RedCounter, room2.RedTurn, room2.MimeDone, room2.IsPublic, room2.IsRunning, room2.Language, room2.RoundTime, room2.IsOver, room2.TeamWon, room2.RoomPass)
|
||||
_, 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)
|
||||
|
||||
rooms, err := repo.ListRooms(context.Background())
|
||||
_, 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)
|
||||
}
|
||||
@ -190,25 +220,33 @@ func TestRoomsRepo_DeleteRoomByID(t *testing.T) {
|
||||
RedCounter: 0,
|
||||
RedTurn: false,
|
||||
MimeDone: false,
|
||||
IsPublic: true,
|
||||
IsRunning: false,
|
||||
Language: "en",
|
||||
RoundTime: 60,
|
||||
IsOver: false,
|
||||
TeamWon: "",
|
||||
RoomPass: "",
|
||||
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_public, is_running, language, round_time, is_over, team_won, room_pass) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`, room.ID, room.CreatedAt, room.CreatorName, room.TeamTurn, room.ThisTurnLimit, room.OpenedThisTurn, room.BlueCounter, room.RedCounter, room.RedTurn, room.MimeDone, room.IsPublic, room.IsRunning, room.Language, room.RoundTime, room.IsOver, room.TeamWon, room.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)
|
||||
|
||||
err = repo.DeleteRoomByID(context.Background(), room.ID)
|
||||
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)
|
||||
}
|
||||
|
||||
func TestRoomsRepo_UpdateRoom(t *testing.T) {
|
||||
@ -228,27 +266,37 @@ func TestRoomsRepo_UpdateRoom(t *testing.T) {
|
||||
RedCounter: 0,
|
||||
RedTurn: false,
|
||||
MimeDone: false,
|
||||
IsPublic: true,
|
||||
IsRunning: false,
|
||||
Language: "en",
|
||||
RoundTime: 60,
|
||||
IsOver: false,
|
||||
TeamWon: "",
|
||||
RoomPass: "",
|
||||
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_public, is_running, language, round_time, is_over, team_won, room_pass) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`, room.ID, room.CreatedAt, room.CreatorName, room.TeamTurn, room.ThisTurnLimit, room.OpenedThisTurn, room.BlueCounter, room.RedCounter, room.RedTurn, room.MimeDone, room.IsPublic, room.IsRunning, room.Language, room.RoundTime, room.IsOver, room.TeamWon, room.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.UpdateRoom(context.Background(), room)
|
||||
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, 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)
|
||||
}
|
||||
Reference in New Issue
Block a user