space-ace-go/db_test.go

package main

import (
	"fmt"
	"os"
	"testing"
	"time"
)

const DEBUG_DB bool = false

func TestLock(t *testing.T) {
	const testdb = ".space-settings.db"

	db := DBData{}
	// Start with empty database
	os.Remove(testdb)

	db.Open(testdb)
	defer db.Close()
	if !DEBUG_DB {
		defer os.Remove(testdb)
	}
	db.User = LOCK_USERNAME
	when := time.Now().Add(-time.Hour).Unix()
	var oldLock string = fmt.Sprintf("%d,%d", os.Getpid(), when)
	db.SetSetting(LOCK_SETTING, oldLock)
	b := db.Lock(1)
	if !b {
		t.Error("Failed to get lock (time on lock expired).")
	}
}

func TestSettings(t *testing.T) {
	const testdb = ".space-settings.db"

	db := DBData{}
	// Start with empty database
	os.Remove(testdb)

	db.Open(testdb)
	defer db.Close()
	if !DEBUG_DB {
		defer os.Remove(testdb)
	}

	users := []string{"test", "bob", "fish"}
	settings := map[string]string{
		"LastPlayed": "20200101",
		"Deck":       "green",
		"Rabbit":     "carrot",
	}

	for _, user := range users {
		db.User = user
		for key, value := range settings {
			db.SetSetting(key, value)
		}
	}

	for _, user := range users {
		db.User = user
		for key, value := range settings {
			found := db.GetSetting(key, "?")
			if found != value {
				t.Errorf("Settings %s/%s: got %s, expected %s\n",
					user, key, found, value)
			}
		}
		found := db.GetSetting("missing", "ok")
		if found != "ok" {
			t.Errorf("Default %s: expected ok, got %s\n", user, found)
		}
	}
}

// Test the db.ExpireScores
func TestExpireScores(t *testing.T) {
	const testdb = ".space-test.db"

	db := DBData{}
	// Start with empty database
	os.Remove(testdb)

	db.Open(testdb)
	defer db.Close()
	if !DEBUG_DB {
		defer os.Remove(testdb)
	}

	db.Unlock()

	db.User = "Testing"

	// stuff score data into database
	// SaveScore(when int64, date int64, hand int, won int, best int, score int)
	type TData struct {
		when  YMD
		date  YMD
		hand  int
		won   int
		run   int
		score int
	}

	type TScore struct {
		Score int
		Run   int
		Won   int
	}

	var scoredata map[YMD]TScore = make(map[YMD]TScore)

	var data []TData = []TData{
		{20230101, 20230101, 1, 0, 8, 750},
		{20230101, 20230101, 2, 0, 8, 780},
		// {20230101, 20230101, 3, 1, 5, 980},
		{20230101, 20230102, 1, 1, 9, 945},
		{20230101, 20230102, 2, 1, 6, 985},
		{20230101, 20230102, 3, 0, 10, 700},
		{20230101, 20230115, 1, 0, 7, 600},
		{20230101, 20230115, 2, 0, 7, 730},
		{20230101, 20230115, 3, 1, 5, 1000}}

	var score, best, won int

	for idx := range data {
		db.SaveScore(data[idx].when, data[idx].date, data[idx].hand,
			data[idx].won, data[idx].run, data[idx].score)

		// Calculate the score, best, and won
		score += data[idx].score
		if data[idx].run > best {
			best = data[idx].run
		}
		won += data[idx].won

		w := data[idx].date
		_, has := scoredata[w]
		if !has {
			scoredata[w] = TScore{}
		}
		sd, has := scoredata[w]

		sd.Score += data[idx].score
		sd.Won += data[idx].won
		if sd.Run < data[idx].run {
			sd.Run = data[idx].run
		}

		scoredata[w] = sd
	}

	played := db.WhenPlayed()
	playdata := map[YMD]int{20230101: 2,
		20230102: 3, 20230115: 3}

	for k, v := range played {
		if v != playdata[k] {
			t.Errorf("WhenPlayed %d: expected %d, got %d\n",
				k, playdata[k], v)
		}
	}

	scores := db.GetScores(5)
	// Verify this has the expected scores (pre-expire)

	// t.Errorf("ScoreData: %#v\n", scoredata)
	// t.Errorf("GetScores: %#v\n", scores)

	for _, s := range scores {
		sd, _ := scoredata[s.Date]
		if sd.Score != s.Score {
			t.Errorf("GetScores %d: Got score %d, expected %d\n",
				s.Date, s.Score, sd.Score)
		}
		if sd.Won != s.Won {
			t.Errorf("GetScores %d: Got Won %d, expected %d\n",
				s.Date, s.Won, sd.Won)
		}
		if sd.Run != s.Run {
			t.Errorf("GetScores %d: Got Run %d, expected %d\n",
				s.Date, s.Run, sd.Run)
		}
	}

	// Data loaded .. call Expire!
	var next_month YMD = 20230202 //time.Time = time.Unix(1645929274, 0)
	next_month.SetDay(1)
	db.ExpireScores(next_month)

	// 1. Verify the scores table is empty.
	scores = db.GetScores(5)

	if len(scores) != 0 {
		t.Errorf("Scores not empty after ExpireScores got %d expected 0.\n", len(scores))
	}
	// 2. Verify the months table is correct.
	monthly := db.GetMonthlyScores(5)

	/*
		type MonthlyData struct {
			Date      int64
			User      string
			Days      int
			Hands_Won int
			Best      int
			Score     int
		}
	*/

	if len(monthly) != 1 {
		t.Errorf("Monthly Scores: got %d, expected 1.\n", len(monthly))
	} else {
		var md MonthlyData = MonthlyData{202301, "Testing", 3, won, best, score}

		if monthly[0].Date != md.Date {
			t.Errorf("Date %d, expected %d.\n", monthly[0].Date, md.Date)
		}
		if monthly[0].User != md.User {
			t.Errorf("User %s, expected %s.\n", monthly[0].User, md.User)
		}
		if monthly[0].Days != md.Days {
			t.Errorf("Days %d, expected %d.\n", monthly[0].Days, md.Days)
		}
		if monthly[0].Hands_Won != md.Hands_Won {
			t.Errorf("Hands %d, expected %d.\n", monthly[0].Hands_Won, md.Hands_Won)
		}
		if monthly[0].Best_Run != md.Best_Run {
			t.Errorf("Run %d, expected %d.\n", monthly[0].Best_Run, md.Best_Run)
		}
		if monthly[0].Score != md.Score {
			t.Errorf("Score %d, expected %d.\n", monthly[0].Score, md.Score)
		}
	}

}