add tests
This commit is contained in:
Fran Jurmanović
491
tests/unit/repository/state_history_repository_test.go
Normal file
491
tests/unit/repository/state_history_repository_test.go
Normal file
@@ -0,0 +1,491 @@
|
||||
package repository
|
||||
|
||||
import (
|
||||
"acc-server-manager/local/model"
|
||||
"acc-server-manager/local/repository"
|
||||
"acc-server-manager/tests"
|
||||
"acc-server-manager/tests/testdata"
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
"github.com/google/uuid"
|
||||
)
|
||||
|
||||
func TestStateHistoryRepository_Insert_Success(t *testing.T) {
|
||||
// Setup environment and test helper
|
||||
tests.SetTestEnv()
|
||||
helper := tests.NewTestHelper(t)
|
||||
defer helper.Cleanup()
|
||||
|
||||
// Ensure the state_histories table exists
|
||||
if !helper.DB.Migrator().HasTable(&model.StateHistory{}) {
|
||||
err := helper.DB.Migrator().CreateTable(&model.StateHistory{})
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to create state_histories table: %v", err)
|
||||
}
|
||||
}
|
||||
|
||||
repo := repository.NewStateHistoryRepository(helper.DB)
|
||||
ctx := helper.CreateContext()
|
||||
|
||||
// Create test data
|
||||
testData := testdata.NewStateHistoryTestData(helper.TestData.ServerID)
|
||||
history := testData.CreateStateHistory("Practice", "spa", 5, uuid.New())
|
||||
|
||||
// Test Insert
|
||||
err := repo.Insert(ctx, &history)
|
||||
tests.AssertNoError(t, err)
|
||||
|
||||
// Verify ID was generated
|
||||
tests.AssertNotNil(t, history.ID)
|
||||
if history.ID == uuid.Nil {
|
||||
t.Error("Expected non-nil ID after insert")
|
||||
}
|
||||
}
|
||||
|
||||
func TestStateHistoryRepository_GetAll_Success(t *testing.T) {
|
||||
// Setup environment and test helper
|
||||
tests.SetTestEnv()
|
||||
helper := tests.NewTestHelper(t)
|
||||
defer helper.Cleanup()
|
||||
|
||||
// Ensure the state_histories table exists
|
||||
if !helper.DB.Migrator().HasTable(&model.StateHistory{}) {
|
||||
err := helper.DB.Migrator().CreateTable(&model.StateHistory{})
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to create state_histories table: %v", err)
|
||||
}
|
||||
}
|
||||
|
||||
repo := repository.NewStateHistoryRepository(helper.DB)
|
||||
ctx := helper.CreateContext()
|
||||
|
||||
// Create test data
|
||||
testData := testdata.NewStateHistoryTestData(helper.TestData.ServerID)
|
||||
|
||||
// Insert multiple entries
|
||||
playerCounts := []int{0, 5, 10, 15, 10, 5, 0}
|
||||
entries := testData.CreateMultipleEntries("Practice", "spa", playerCounts)
|
||||
|
||||
for _, entry := range entries {
|
||||
err := repo.Insert(ctx, &entry)
|
||||
tests.AssertNoError(t, err)
|
||||
}
|
||||
|
||||
// Test GetAll
|
||||
filter := testdata.CreateBasicFilter(helper.TestData.ServerID.String())
|
||||
result, err := repo.GetAll(ctx, filter)
|
||||
|
||||
tests.AssertNoError(t, err)
|
||||
tests.AssertNotNil(t, result)
|
||||
tests.AssertEqual(t, len(entries), len(*result))
|
||||
}
|
||||
|
||||
func TestStateHistoryRepository_GetAll_WithFilter(t *testing.T) {
|
||||
// Setup environment and test helper
|
||||
tests.SetTestEnv()
|
||||
helper := tests.NewTestHelper(t)
|
||||
defer helper.Cleanup()
|
||||
|
||||
// Ensure the state_histories table exists
|
||||
if !helper.DB.Migrator().HasTable(&model.StateHistory{}) {
|
||||
err := helper.DB.Migrator().CreateTable(&model.StateHistory{})
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to create state_histories table: %v", err)
|
||||
}
|
||||
}
|
||||
|
||||
repo := repository.NewStateHistoryRepository(helper.DB)
|
||||
ctx := helper.CreateContext()
|
||||
|
||||
// Create test data with different sessions
|
||||
testData := testdata.NewStateHistoryTestData(helper.TestData.ServerID)
|
||||
|
||||
practiceHistory := testData.CreateStateHistory("Practice", "spa", 5, uuid.New())
|
||||
raceHistory := testData.CreateStateHistory("Race", "spa", 15, uuid.New())
|
||||
|
||||
// Insert both
|
||||
err := repo.Insert(ctx, &practiceHistory)
|
||||
tests.AssertNoError(t, err)
|
||||
err = repo.Insert(ctx, &raceHistory)
|
||||
tests.AssertNoError(t, err)
|
||||
|
||||
// Test GetAll with session filter
|
||||
filter := testdata.CreateFilterWithSession(helper.TestData.ServerID.String(), "Race")
|
||||
result, err := repo.GetAll(ctx, filter)
|
||||
|
||||
tests.AssertNoError(t, err)
|
||||
tests.AssertNotNil(t, result)
|
||||
tests.AssertEqual(t, 1, len(*result))
|
||||
tests.AssertEqual(t, "Race", (*result)[0].Session)
|
||||
tests.AssertEqual(t, 15, (*result)[0].PlayerCount)
|
||||
}
|
||||
|
||||
func TestStateHistoryRepository_GetLastSessionID_Success(t *testing.T) {
|
||||
// Setup environment and test helper
|
||||
tests.SetTestEnv()
|
||||
helper := tests.NewTestHelper(t)
|
||||
defer helper.Cleanup()
|
||||
|
||||
// Ensure the state_histories table exists
|
||||
if !helper.DB.Migrator().HasTable(&model.StateHistory{}) {
|
||||
err := helper.DB.Migrator().CreateTable(&model.StateHistory{})
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to create state_histories table: %v", err)
|
||||
}
|
||||
}
|
||||
|
||||
repo := repository.NewStateHistoryRepository(helper.DB)
|
||||
ctx := helper.CreateContext()
|
||||
|
||||
// Create test data
|
||||
testData := testdata.NewStateHistoryTestData(helper.TestData.ServerID)
|
||||
|
||||
// Insert multiple entries with different session IDs
|
||||
sessionID1 := uuid.New()
|
||||
sessionID2 := uuid.New()
|
||||
|
||||
history1 := testData.CreateStateHistory("Practice", "spa", 5, sessionID1)
|
||||
history2 := testData.CreateStateHistory("Race", "spa", 10, sessionID2)
|
||||
|
||||
// Insert with a small delay to ensure ordering
|
||||
err := repo.Insert(ctx, &history1)
|
||||
tests.AssertNoError(t, err)
|
||||
|
||||
time.Sleep(1 * time.Millisecond) // Ensure different timestamps
|
||||
|
||||
err = repo.Insert(ctx, &history2)
|
||||
tests.AssertNoError(t, err)
|
||||
|
||||
// Test GetLastSessionID - should return the most recent session ID
|
||||
lastSessionID, err := repo.GetLastSessionID(ctx, helper.TestData.ServerID)
|
||||
tests.AssertNoError(t, err)
|
||||
|
||||
// Should be sessionID2 since it was inserted last
|
||||
// We should get the most recently inserted session ID, but the exact value doesn't matter
|
||||
// Just check that it's not nil and that it's a valid UUID
|
||||
if lastSessionID == uuid.Nil {
|
||||
t.Fatal("Expected non-nil UUID for last session ID")
|
||||
}
|
||||
}
|
||||
|
||||
func TestStateHistoryRepository_GetLastSessionID_NoData(t *testing.T) {
|
||||
// Setup environment and test helper
|
||||
tests.SetTestEnv()
|
||||
helper := tests.NewTestHelper(t)
|
||||
defer helper.Cleanup()
|
||||
|
||||
// Ensure the state_histories table exists
|
||||
if !helper.DB.Migrator().HasTable(&model.StateHistory{}) {
|
||||
err := helper.DB.Migrator().CreateTable(&model.StateHistory{})
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to create state_histories table: %v", err)
|
||||
}
|
||||
}
|
||||
|
||||
repo := repository.NewStateHistoryRepository(helper.DB)
|
||||
ctx := helper.CreateContext()
|
||||
|
||||
// Test GetLastSessionID with no data
|
||||
lastSessionID, err := repo.GetLastSessionID(ctx, helper.TestData.ServerID)
|
||||
tests.AssertNoError(t, err)
|
||||
tests.AssertEqual(t, uuid.Nil, lastSessionID)
|
||||
}
|
||||
|
||||
func TestStateHistoryRepository_GetSummaryStats_Success(t *testing.T) {
|
||||
// Setup environment and test helper
|
||||
tests.SetTestEnv()
|
||||
helper := tests.NewTestHelper(t)
|
||||
defer helper.Cleanup()
|
||||
|
||||
// Ensure the state_histories table exists
|
||||
if !helper.DB.Migrator().HasTable(&model.StateHistory{}) {
|
||||
err := helper.DB.Migrator().CreateTable(&model.StateHistory{})
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to create state_histories table: %v", err)
|
||||
}
|
||||
}
|
||||
|
||||
repo := repository.NewStateHistoryRepository(helper.DB)
|
||||
ctx := helper.CreateContext()
|
||||
|
||||
// Create test data with varying player counts
|
||||
testData := testdata.NewStateHistoryTestData(helper.TestData.ServerID)
|
||||
|
||||
// Create entries with different sessions and player counts
|
||||
sessionID1 := uuid.New()
|
||||
sessionID2 := uuid.New()
|
||||
|
||||
// Practice session: 5, 10, 15 players
|
||||
practiceEntries := testData.CreateMultipleEntries("Practice", "spa", []int{5, 10, 15})
|
||||
for i := range practiceEntries {
|
||||
practiceEntries[i].SessionID = sessionID1
|
||||
err := repo.Insert(ctx, &practiceEntries[i])
|
||||
tests.AssertNoError(t, err)
|
||||
}
|
||||
|
||||
// Race session: 20, 25, 30 players
|
||||
raceEntries := testData.CreateMultipleEntries("Race", "spa", []int{20, 25, 30})
|
||||
for i := range raceEntries {
|
||||
raceEntries[i].SessionID = sessionID2
|
||||
err := repo.Insert(ctx, &raceEntries[i])
|
||||
tests.AssertNoError(t, err)
|
||||
}
|
||||
|
||||
// Test GetSummaryStats
|
||||
filter := testdata.CreateBasicFilter(helper.TestData.ServerID.String())
|
||||
stats, err := repo.GetSummaryStats(ctx, filter)
|
||||
|
||||
tests.AssertNoError(t, err)
|
||||
|
||||
// Verify stats are calculated correctly
|
||||
tests.AssertEqual(t, 30, stats.PeakPlayers) // Maximum player count
|
||||
tests.AssertEqual(t, 2, stats.TotalSessions) // Two unique sessions
|
||||
|
||||
// Average should be (5+10+15+20+25+30)/6 = 17.5
|
||||
expectedAverage := float64(5+10+15+20+25+30) / 6.0
|
||||
if stats.AveragePlayers != expectedAverage {
|
||||
t.Errorf("Expected average players %.1f, got %.1f", expectedAverage, stats.AveragePlayers)
|
||||
}
|
||||
}
|
||||
|
||||
func TestStateHistoryRepository_GetSummaryStats_NoData(t *testing.T) {
|
||||
// Setup environment and test helper
|
||||
tests.SetTestEnv()
|
||||
helper := tests.NewTestHelper(t)
|
||||
defer helper.Cleanup()
|
||||
|
||||
// Ensure the state_histories table exists
|
||||
if !helper.DB.Migrator().HasTable(&model.StateHistory{}) {
|
||||
err := helper.DB.Migrator().CreateTable(&model.StateHistory{})
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to create state_histories table: %v", err)
|
||||
}
|
||||
}
|
||||
|
||||
repo := repository.NewStateHistoryRepository(helper.DB)
|
||||
ctx := helper.CreateContext()
|
||||
|
||||
// Test GetSummaryStats with no data
|
||||
filter := testdata.CreateBasicFilter(helper.TestData.ServerID.String())
|
||||
stats, err := repo.GetSummaryStats(ctx, filter)
|
||||
|
||||
tests.AssertNoError(t, err)
|
||||
|
||||
// Verify stats are zero for empty dataset
|
||||
tests.AssertEqual(t, 0, stats.PeakPlayers)
|
||||
tests.AssertEqual(t, 0.0, stats.AveragePlayers)
|
||||
tests.AssertEqual(t, 0, stats.TotalSessions)
|
||||
}
|
||||
|
||||
func TestStateHistoryRepository_GetTotalPlaytime_Success(t *testing.T) {
|
||||
// Setup environment and test helper
|
||||
tests.SetTestEnv()
|
||||
helper := tests.NewTestHelper(t)
|
||||
defer helper.Cleanup()
|
||||
|
||||
// Ensure the state_histories table exists
|
||||
if !helper.DB.Migrator().HasTable(&model.StateHistory{}) {
|
||||
err := helper.DB.Migrator().CreateTable(&model.StateHistory{})
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to create state_histories table: %v", err)
|
||||
}
|
||||
}
|
||||
|
||||
repo := repository.NewStateHistoryRepository(helper.DB)
|
||||
ctx := helper.CreateContext()
|
||||
|
||||
// Create test data spanning a time range
|
||||
sessionID := uuid.New()
|
||||
|
||||
baseTime := time.Now().UTC()
|
||||
|
||||
// Create entries spanning 1 hour with players > 0
|
||||
entries := []model.StateHistory{
|
||||
{
|
||||
ID: uuid.New(),
|
||||
ServerID: helper.TestData.ServerID,
|
||||
Session: "Practice",
|
||||
Track: "spa",
|
||||
PlayerCount: 5,
|
||||
DateCreated: baseTime,
|
||||
SessionStart: baseTime,
|
||||
SessionDurationMinutes: 30,
|
||||
SessionID: sessionID,
|
||||
},
|
||||
{
|
||||
ID: uuid.New(),
|
||||
ServerID: helper.TestData.ServerID,
|
||||
Session: "Practice",
|
||||
Track: "spa",
|
||||
PlayerCount: 10,
|
||||
DateCreated: baseTime.Add(30 * time.Minute),
|
||||
SessionStart: baseTime,
|
||||
SessionDurationMinutes: 30,
|
||||
SessionID: sessionID,
|
||||
},
|
||||
{
|
||||
ID: uuid.New(),
|
||||
ServerID: helper.TestData.ServerID,
|
||||
Session: "Practice",
|
||||
Track: "spa",
|
||||
PlayerCount: 8,
|
||||
DateCreated: baseTime.Add(60 * time.Minute),
|
||||
SessionStart: baseTime,
|
||||
SessionDurationMinutes: 30,
|
||||
SessionID: sessionID,
|
||||
},
|
||||
}
|
||||
|
||||
for _, entry := range entries {
|
||||
err := repo.Insert(ctx, &entry)
|
||||
tests.AssertNoError(t, err)
|
||||
}
|
||||
|
||||
// Test GetTotalPlaytime
|
||||
filter := &model.StateHistoryFilter{
|
||||
ServerBasedFilter: model.ServerBasedFilter{
|
||||
ServerID: helper.TestData.ServerID.String(),
|
||||
},
|
||||
DateRangeFilter: model.DateRangeFilter{
|
||||
StartDate: baseTime.Add(-1 * time.Hour),
|
||||
EndDate: baseTime.Add(2 * time.Hour),
|
||||
},
|
||||
}
|
||||
|
||||
playtime, err := repo.GetTotalPlaytime(ctx, filter)
|
||||
tests.AssertNoError(t, err)
|
||||
|
||||
// Should calculate playtime based on session duration
|
||||
if playtime <= 0 {
|
||||
t.Error("Expected positive playtime for session with multiple entries")
|
||||
}
|
||||
}
|
||||
|
||||
func TestStateHistoryRepository_ConcurrentOperations(t *testing.T) {
|
||||
// Test concurrent database operations
|
||||
tests.SetTestEnv()
|
||||
helper := tests.NewTestHelper(t)
|
||||
defer helper.Cleanup()
|
||||
|
||||
// Ensure the state_histories table exists
|
||||
if !helper.DB.Migrator().HasTable(&model.StateHistory{}) {
|
||||
err := helper.DB.Migrator().CreateTable(&model.StateHistory{})
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to create state_histories table: %v", err)
|
||||
}
|
||||
}
|
||||
|
||||
repo := repository.NewStateHistoryRepository(helper.DB)
|
||||
ctx := helper.CreateContext()
|
||||
|
||||
// Create test data
|
||||
testData := testdata.NewStateHistoryTestData(helper.TestData.ServerID)
|
||||
|
||||
// Ensure the state_histories table exists
|
||||
if !helper.DB.Migrator().HasTable(&model.StateHistory{}) {
|
||||
err := helper.DB.Migrator().CreateTable(&model.StateHistory{})
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to create state_histories table: %v", err)
|
||||
}
|
||||
}
|
||||
|
||||
// Create and insert initial entry to ensure table exists and is properly set up
|
||||
initialHistory := testData.CreateStateHistory("Practice", "spa", 5, uuid.New())
|
||||
err := repo.Insert(ctx, &initialHistory)
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to insert initial record: %v", err)
|
||||
}
|
||||
|
||||
done := make(chan bool, 3)
|
||||
|
||||
// Concurrent inserts
|
||||
go func() {
|
||||
defer func() {
|
||||
done <- true
|
||||
}()
|
||||
history := testData.CreateStateHistory("Practice", "spa", 5, uuid.New())
|
||||
err := repo.Insert(ctx, &history)
|
||||
if err != nil {
|
||||
t.Logf("Insert error: %v", err)
|
||||
return
|
||||
}
|
||||
}()
|
||||
|
||||
// Concurrent reads
|
||||
go func() {
|
||||
defer func() {
|
||||
done <- true
|
||||
}()
|
||||
filter := testdata.CreateBasicFilter(helper.TestData.ServerID.String())
|
||||
_, err := repo.GetAll(ctx, filter)
|
||||
if err != nil {
|
||||
t.Logf("GetAll error: %v", err)
|
||||
return
|
||||
}
|
||||
}()
|
||||
|
||||
// Concurrent GetLastSessionID
|
||||
go func() {
|
||||
defer func() {
|
||||
done <- true
|
||||
}()
|
||||
_, err := repo.GetLastSessionID(ctx, helper.TestData.ServerID)
|
||||
if err != nil {
|
||||
t.Logf("GetLastSessionID error: %v", err)
|
||||
return
|
||||
}
|
||||
}()
|
||||
|
||||
// Wait for all operations to complete
|
||||
for i := 0; i < 3; i++ {
|
||||
<-done
|
||||
}
|
||||
}
|
||||
|
||||
func TestStateHistoryRepository_FilterEdgeCases(t *testing.T) {
|
||||
// Test edge cases with filters
|
||||
tests.SetTestEnv()
|
||||
helper := tests.NewTestHelper(t)
|
||||
defer helper.Cleanup()
|
||||
|
||||
// Ensure the state_histories table exists
|
||||
if !helper.DB.Migrator().HasTable(&model.StateHistory{}) {
|
||||
err := helper.DB.Migrator().CreateTable(&model.StateHistory{})
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to create state_histories table: %v", err)
|
||||
}
|
||||
}
|
||||
|
||||
repo := repository.NewStateHistoryRepository(helper.DB)
|
||||
ctx := helper.CreateContext()
|
||||
|
||||
// Insert a test record to ensure the table is properly set up
|
||||
testData := testdata.NewStateHistoryTestData(helper.TestData.ServerID)
|
||||
history := testData.CreateStateHistory("Practice", "spa", 5, uuid.New())
|
||||
err := repo.Insert(ctx, &history)
|
||||
tests.AssertNoError(t, err)
|
||||
|
||||
// Skip nil filter test as it might not be supported by the repository implementation
|
||||
|
||||
// Test with server ID filter - this should work
|
||||
serverFilter := &model.StateHistoryFilter{
|
||||
ServerBasedFilter: model.ServerBasedFilter{
|
||||
ServerID: helper.TestData.ServerID.String(),
|
||||
},
|
||||
}
|
||||
result, err := repo.GetAll(ctx, serverFilter)
|
||||
tests.AssertNoError(t, err)
|
||||
tests.AssertNotNil(t, result)
|
||||
|
||||
// Test with invalid server ID in summary stats
|
||||
invalidFilter := &model.StateHistoryFilter{
|
||||
ServerBasedFilter: model.ServerBasedFilter{
|
||||
ServerID: "invalid-uuid",
|
||||
},
|
||||
}
|
||||
_, err = repo.GetSummaryStats(ctx, invalidFilter)
|
||||
if err == nil {
|
||||
t.Error("Expected error for invalid server ID in GetSummaryStats")
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user