david
0977d3a58b
- Add SQLite and UUID dependencies (go-sqlite3, google/uuid) - Create database connection with automatic migrations - Implement UserRepository with full CRUD operations - Implement HabitRepository with JSON serialization for arrays - Implement HabitEntryRepository with date range queries - Add comprehensive test coverage for all repositories - Fix User entity to default timezone to UTC when empty - All tests passing with TDD approach (Red-Green-Refactor)
88 lines
1.6 KiB
Go
88 lines
1.6 KiB
Go
package sqlite
|
|
|
|
import (
|
|
"os"
|
|
"testing"
|
|
)
|
|
|
|
func TestNewDatabase(t *testing.T) {
|
|
dbPath := "./test_db.sqlite"
|
|
defer os.Remove(dbPath)
|
|
|
|
db, err := NewDatabase(dbPath)
|
|
if err != nil {
|
|
t.Fatalf("NewDatabase failed: %v", err)
|
|
}
|
|
defer db.Close()
|
|
|
|
if db == nil {
|
|
t.Fatal("Expected database instance, got nil")
|
|
}
|
|
}
|
|
|
|
func TestNewDatabaseCreatesDataDirectory(t *testing.T) {
|
|
dbPath := "./test_data/nested/db.sqlite"
|
|
defer os.RemoveAll("./test_data")
|
|
|
|
db, err := NewDatabase(dbPath)
|
|
if err != nil {
|
|
t.Fatalf("NewDatabase failed: %v", err)
|
|
}
|
|
defer db.Close()
|
|
|
|
if _, err := os.Stat("./test_data/nested"); os.IsNotExist(err) {
|
|
t.Fatal("Data directory was not created")
|
|
}
|
|
}
|
|
|
|
func TestNewDatabaseRunsMigrations(t *testing.T) {
|
|
dbPath := ":memory:"
|
|
|
|
db, err := NewDatabase(dbPath)
|
|
if err != nil {
|
|
t.Fatalf("NewDatabase failed: %v", err)
|
|
}
|
|
defer db.Close()
|
|
|
|
conn := db.Conn()
|
|
var name string
|
|
err = conn.QueryRow("SELECT name FROM sqlite_master WHERE type='table' AND name='users'").Scan(&name)
|
|
if err != nil {
|
|
t.Fatal("Migrations were not run, users table does not exist")
|
|
}
|
|
}
|
|
|
|
func TestDatabaseClose(t *testing.T) {
|
|
dbPath := ":memory:"
|
|
|
|
db, err := NewDatabase(dbPath)
|
|
if err != nil {
|
|
t.Fatalf("NewDatabase failed: %v", err)
|
|
}
|
|
|
|
err = db.Close()
|
|
if err != nil {
|
|
t.Errorf("Close failed: %v", err)
|
|
}
|
|
}
|
|
|
|
func TestDatabaseConn(t *testing.T) {
|
|
dbPath := ":memory:"
|
|
|
|
db, err := NewDatabase(dbPath)
|
|
if err != nil {
|
|
t.Fatalf("NewDatabase failed: %v", err)
|
|
}
|
|
defer db.Close()
|
|
|
|
conn := db.Conn()
|
|
if conn == nil {
|
|
t.Fatal("Expected sql.DB connection, got nil")
|
|
}
|
|
|
|
err = conn.Ping()
|
|
if err != nil {
|
|
t.Errorf("Ping failed: %v", err)
|
|
}
|
|
}
|