mirror of
https://github.com/woodpecker-ci/woodpecker.git
synced 2024-12-28 19:30:30 +00:00
102 lines
2.1 KiB
Go
102 lines
2.1 KiB
Go
package meddler
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func TestLoad(t *testing.T) {
|
|
once.Do(setup)
|
|
insertAliceBob(t)
|
|
|
|
elt := new(Person)
|
|
elt.Age = 50
|
|
elt.Closed = time.Now()
|
|
if err := Load(db, "person", elt, 2); err != nil {
|
|
t.Errorf("Load error on Bob: %v", err)
|
|
return
|
|
}
|
|
bob.ID = 2
|
|
personEqual(t, elt, bob)
|
|
db.Exec("delete from person")
|
|
}
|
|
|
|
func TestLoadUint(t *testing.T) {
|
|
once.Do(setup)
|
|
insertAliceBob(t)
|
|
|
|
elt := new(UintPerson)
|
|
elt.Age = 50
|
|
elt.Closed = time.Now()
|
|
if err := Load(db, "person", elt, 2); err != nil {
|
|
t.Errorf("Load error on Bob: %v", err)
|
|
return
|
|
}
|
|
bob.ID = 2
|
|
db.Exec("delete from person")
|
|
}
|
|
|
|
func TestSave(t *testing.T) {
|
|
once.Do(setup)
|
|
insertAliceBob(t)
|
|
|
|
h := 73
|
|
chris := &Person{
|
|
ID: 0,
|
|
Name: "Chris",
|
|
Email: "chris@chris.com",
|
|
Ephemeral: 19,
|
|
Age: 23,
|
|
Opened: when.Local(),
|
|
Closed: when,
|
|
Updated: nil,
|
|
Height: &h,
|
|
}
|
|
|
|
tx, err := db.Begin()
|
|
if err != nil {
|
|
t.Errorf("DB error on begin: %v", err)
|
|
}
|
|
if err = Save(tx, "person", chris); err != nil {
|
|
t.Errorf("DB error on Save: %v", err)
|
|
}
|
|
|
|
id := chris.ID
|
|
if id != 3 {
|
|
t.Errorf("DB error on Save: expected ID of 3 but got %d", id)
|
|
}
|
|
|
|
chris.Email = "chris@chrischris.com"
|
|
chris.Age = 27
|
|
|
|
if err = Save(tx, "person", chris); err != nil {
|
|
t.Errorf("DB error on Save: %v", err)
|
|
}
|
|
if chris.ID != id {
|
|
t.Errorf("ID mismatch: found %d when %d expected", chris.ID, id)
|
|
}
|
|
if err = tx.Commit(); err != nil {
|
|
t.Errorf("Commit error: %v", err)
|
|
}
|
|
|
|
// now test if the data looks right
|
|
rows, err := db.Query("select * from person where id = ?", id)
|
|
if err != nil {
|
|
t.Errorf("DB error on query: %v", err)
|
|
return
|
|
}
|
|
|
|
p := new(Person)
|
|
if err = Default.ScanRow(rows, p); err != nil {
|
|
t.Errorf("ScanRow error on Chris: %v", err)
|
|
return
|
|
}
|
|
|
|
personEqual(t, p, &Person{3, "Chris", 0, "chris@chrischris.com", 0, 27, when, when, nil, &h})
|
|
|
|
// delete this record so we don't confuse other tests
|
|
if _, err = db.Exec("delete from person where id = ?", id); err != nil {
|
|
t.Errorf("DB error on delete: %v", err)
|
|
}
|
|
db.Exec("delete from person")
|
|
}
|