Feat: card_mark repo
This commit is contained in:
39
repos/card_marks.go
Normal file
39
repos/card_marks.go
Normal file
@ -0,0 +1,39 @@
|
||||
package repos
|
||||
|
||||
import (
|
||||
"context"
|
||||
"gralias/models"
|
||||
|
||||
"github.com/jmoiron/sqlx"
|
||||
)
|
||||
|
||||
type CardMarksRepo interface {
|
||||
CardMarksByCardID(ctx context.Context, cardID string) ([]models.CardMark, error)
|
||||
CardMarksAdd(ctx context.Context, cm *models.CardMark) error
|
||||
CardMarksRemove(ctx context.Context, cardID, username string) error
|
||||
}
|
||||
|
||||
type cardMarksRepo struct {
|
||||
db *sqlx.DB
|
||||
}
|
||||
|
||||
func NewCardMarksRepo(db *sqlx.DB) CardMarksRepo {
|
||||
return &cardMarksRepo{db: db}
|
||||
}
|
||||
|
||||
func (r *cardMarksRepo) CardMarksByCardID(ctx context.Context, cardID string) ([]models.CardMark, error) {
|
||||
var cardMarks []models.CardMark
|
||||
err := sqlx.SelectContext(ctx, getDB(ctx, r.db), &cardMarks, "SELECT * FROM card_marks WHERE card_id = ?", cardID)
|
||||
return cardMarks, err
|
||||
}
|
||||
|
||||
func (r *cardMarksRepo) CardMarksAdd(ctx context.Context, cm *models.CardMark) error {
|
||||
_, err := getDB(ctx, r.db).ExecContext(ctx, "INSERT INTO card_marks (card_id, username, mark_type) VALUES (?, ?, ?)", cm.CardID, cm.Username, cm.MarkType)
|
||||
return err
|
||||
}
|
||||
|
||||
func (r *cardMarksRepo) CardMarksRemove(ctx context.Context, cardID, username string) error {
|
||||
db := getDB(ctx, r.db)
|
||||
_, err := db.ExecContext(ctx, "DELETE FROM card_marks WHERE card_id = ? AND username = ?", cardID, username)
|
||||
return err
|
||||
}
|
@ -18,6 +18,7 @@ type AllRepos interface {
|
||||
SessionsRepo
|
||||
WordCardsRepo
|
||||
SettingsRepo
|
||||
CardMarksRepo
|
||||
InitTx(ctx context.Context) (context.Context, *sqlx.Tx, error)
|
||||
}
|
||||
|
||||
@ -25,6 +26,7 @@ type RepoProvider struct {
|
||||
DB *sqlx.DB
|
||||
mu sync.RWMutex
|
||||
pathToDB string
|
||||
CardMarksRepo
|
||||
}
|
||||
|
||||
func NewRepoProvider(pathToDB string) *RepoProvider {
|
||||
@ -42,6 +44,7 @@ func NewRepoProvider(pathToDB string) *RepoProvider {
|
||||
rp := &RepoProvider{
|
||||
DB: db,
|
||||
pathToDB: pathToDB,
|
||||
CardMarksRepo: NewCardMarksRepo(db),
|
||||
}
|
||||
|
||||
go rp.pingLoop()
|
||||
|
@ -7,8 +7,8 @@ import (
|
||||
"time"
|
||||
|
||||
"github.com/jmoiron/sqlx"
|
||||
"github.com/stretchr/testify/assert"
|
||||
_ "github.com/mattn/go-sqlite3"
|
||||
"github.com/stretchr/testify/assert"
|
||||
)
|
||||
|
||||
func setupTestDB(t *testing.T) (*sqlx.DB, func()) {
|
||||
@ -181,9 +181,9 @@ func TestRoomsRepo_GetRoomByID(t *testing.T) {
|
||||
}
|
||||
|
||||
// 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)
|
||||
_, 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)
|
||||
_, 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)
|
||||
@ -288,7 +288,7 @@ func TestRoomsRepo_DeleteRoomByID(t *testing.T) {
|
||||
|
||||
_, 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, room.Settings.TurnSecondsLeft)
|
||||
_, 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
|
||||
@ -347,21 +347,19 @@ func TestRoomsRepo_UpdateRoom(t *testing.T) {
|
||||
},
|
||||
}
|
||||
|
||||
_, 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)
|
||||
var err error
|
||||
_, 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, room.Settings.TurnSecondsLeft)
|
||||
_, 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)
|
||||
@ -371,6 +369,6 @@ func TestRoomsRepo_UpdateRoom(t *testing.T) {
|
||||
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)
|
||||
assert.Equal(t, uint32(30), updatedSettings.TurnSecondsLeft)
|
||||
|
||||
}
|
||||
|
||||
|
Reference in New Issue
Block a user