76 lines
1.8 KiB
Go
76 lines
1.8 KiB
Go
package db_test
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"apigo.cc/go/db"
|
|
"apigo.cc/go/file"
|
|
"apigo.cc/go/log"
|
|
_ "modernc.org/sqlite"
|
|
)
|
|
|
|
func TestSmartDelete(t *testing.T) {
|
|
db.ResetAllForTest()
|
|
|
|
dbPath := "./test_smart_delete.db"
|
|
dbName := "test_delete"
|
|
file.Remove(dbPath)
|
|
|
|
db.SetConfigForTest(dbName, &db.Config{
|
|
Type: "sqlite",
|
|
Host: dbPath,
|
|
})
|
|
|
|
dbInst := db.GetDB(dbName, log.DefaultLogger)
|
|
if dbInst == nil {
|
|
t.Fatal("dbInst should not be nil")
|
|
}
|
|
defer func() {
|
|
dbInst.Destroy()
|
|
file.Remove(dbPath)
|
|
}()
|
|
|
|
// Create table and shadow table
|
|
dbInst.Exec("CREATE TABLE orders (id INTEGER PRIMARY KEY, item TEXT)")
|
|
dbInst.Exec("CREATE TABLE orders_deleted (id INTEGER PRIMARY KEY, item TEXT)")
|
|
|
|
t.Run("ShadowDelete", func(t *testing.T) {
|
|
dbInst.Exec("INSERT INTO orders (id, item) VALUES (1, 'Phone')")
|
|
|
|
res := dbInst.Delete("orders", "id = 1")
|
|
if res.Error != nil {
|
|
t.Fatalf("Delete failed: %v", res.Error)
|
|
}
|
|
if res.Changes() != 1 {
|
|
t.Errorf("Expected 1 change, got %d", res.Changes())
|
|
}
|
|
|
|
// Verify it's gone from main table
|
|
qr := dbInst.Query("SELECT COUNT(*) FROM orders WHERE id = 1")
|
|
count, _ := db.To[int](qr)
|
|
if count != 0 {
|
|
t.Errorf("Expected 0 records in main table, got %d", count)
|
|
}
|
|
|
|
// Verify it's in shadow table
|
|
qr2 := dbInst.Query("SELECT COUNT(*) FROM orders_deleted WHERE id = 1")
|
|
countDeleted, _ := db.To[int](qr2)
|
|
if countDeleted != 1 {
|
|
t.Errorf("Expected 1 record in shadow table, got %d", countDeleted)
|
|
}
|
|
})
|
|
|
|
t.Run("PhysicalDelete", func(t *testing.T) {
|
|
dbInst.Exec("CREATE TABLE logs (id INTEGER PRIMARY KEY, msg TEXT)")
|
|
dbInst.Exec("INSERT INTO logs (id, msg) VALUES (1, 'Login')")
|
|
|
|
dbInst.Delete("logs", "id = 1")
|
|
|
|
qr := dbInst.Query("SELECT COUNT(*) FROM logs WHERE id = 1")
|
|
count, _ := db.To[int](qr)
|
|
if count != 0 {
|
|
t.Errorf("Expected 0 records in logs, got %d", count)
|
|
}
|
|
})
|
|
}
|